kgdata_core 4.0.3

Library to process dumps of knowledge graphs (Wikipedia, DBpedia, Wikidata)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use hashbrown::HashSet;
use pyo3::{
    prelude::*,
    types::{PyList, PySet},
};

/// An zero-cost abstraction for automatically receiving HashSet<String> from Python.
pub struct PySetString(pub HashSet<String>);

impl<'t> FromPyObject<'t> for PySetString {
    fn extract(obj: &'t PyAny) -> PyResult<Self> {
        if let Ok(lst) = obj.downcast::<PyList>() {
            return Ok(PySetString(
                lst.into_iter()
                    .map(|x| x.extract())
                    .collect::<PyResult<HashSet<String>>>()?,
            ));
        }
        if let Ok(obj) = obj.downcast::<PySet>() {
            return Ok(PySetString(
                obj.into_iter()
                    .map(|x| x.extract())
                    .collect::<PyResult<HashSet<String>>>()?,
            ));
        }
        return Err(PyErr::new::<pyo3::exceptions::PyTypeError, _>(
            "Expected a list or set of strings",
        ));
    }
}