Fork of https://github.com/oxigraph/oxigraph.git for the purpose of NextGraph project
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
oxigraph/python/src/lib.rs

133 lines
3.6 KiB

#![cfg_attr(
feature = "cargo-clippy",
allow(zero_ptr, transmute_ptr_to_ptr)
)]
#[macro_use]
extern crate cpython;
extern crate rudf;
use cpython::exc::ValueError;
use cpython::CompareOp;
use cpython::PyErr;
use cpython::PyResult;
use cpython::Python;
use cpython::PythonObject;
use cpython::ToPyObject;
use rudf::model;
use rudf::Error;
use std::collections::hash_map::DefaultHasher;
use std::hash::Hash;
use std::hash::Hasher;
use std::str::FromStr;
py_module_initializer!(rudf, initrudf, PyInit_rudf, |py, m| {
try!(m.add(py, "__doc__", "Rudf Python bindings"));
try!(m.add_class::<NamedNode>(py));
try!(m.add_class::<BlankNode>(py));
try!(m.add_class::<Literal>(py));
Ok(())
});
fn new_value_error(py: Python, error: &Error) -> PyErr {
PyErr::new::<ValueError, _>(py, error.to_string())
}
fn eq_compare<T: Eq + Ord>(a: &T, b: &T, op: &CompareOp) -> bool {
match op {
CompareOp::Lt => a < b,
CompareOp::Le => a <= b,
CompareOp::Eq => a == b,
CompareOp::Ne => a != b,
CompareOp::Gt => a > b,
CompareOp::Ge => a >= b,
}
}
fn hash(t: &impl Hash) -> u64 {
let mut s = DefaultHasher::new();
t.hash(&mut s);
s.finish()
}
py_class!(class NamedNode |py| {
data inner: model::NamedNode;
def __new__(_cls, value: &str) -> PyResult<NamedNode> {
NamedNode::create_instance(py, model::NamedNode::from_str(value).map_err(|error| new_value_error(py, &error))?)
}
def value(&self) -> PyResult<String> {
Ok(self.inner(py).as_str().to_string())
}
def __str__(&self) -> PyResult<String> {
Ok(self.inner(py).to_string())
}
def __richcmp__(&self, other: &NamedNode, op: CompareOp) -> PyResult<bool> {
Ok(eq_compare(&self.inner(py), &other.inner(py), &op))
}
def __hash__(&self) -> PyResult<u64> {
Ok(hash(self.inner(py)))
}
});
py_class!(class BlankNode |py| {
data inner: model::BlankNode;
def __new__(_cls) -> PyResult<BlankNode> {
BlankNode::create_instance(py, model::BlankNode::default())
}
def __str__(&self) -> PyResult<String> {
Ok(self.inner(py).to_string())
}
def __richcmp__(&self, other: &BlankNode, op: CompareOp) -> PyResult<bool> {
Ok(eq_compare(&self.inner(py), &other.inner(py), &op))
}
def __hash__(&self) -> PyResult<u64> {
Ok(hash(self.inner(py)))
}
});
py_class!(class Literal |py| {
data inner: model::Literal;
def __new__(_cls, value: String, language: Option<String> = None, datatype: Option<NamedNode> = None) -> PyResult<Literal> {
Literal::create_instance(py, match language {
Some(language) => model::Literal::new_language_tagged_literal(value, language),
None => match datatype {
Some(datatype) => model::Literal::new_typed_literal(value, datatype.inner(py).clone()),
None => model::Literal::new_simple_literal(value)
}
})
}
def value(&self) -> PyResult<String> {
Ok(self.inner(py).value().to_string())
}
def language(&self) -> PyResult<Option<String>> {
Ok(self.inner(py).language().map(|l| l.to_string()))
}
def datatype(&self) -> PyResult<NamedNode> {
NamedNode::create_instance(py, self.inner(py).datatype().clone())
}
def __str__(&self) -> PyResult<String> {
Ok(self.inner(py).to_string())
}
def __richcmp__(&self, other: &Literal, op: CompareOp) -> PyResult<bool> {
Ok(eq_compare(&self.inner(py), &other.inner(py), &op))
}
def __hash__(&self) -> PyResult<u64> {
Ok(hash(self.inner(py)))
}
});