raphtory-api 0.17.0

Raphtory common interface and APIs
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
31
32
33
34
35
36
37
38
39
40
41

use crate::core::entities::GID;
use pyo3::{exceptions::PyTypeError, prelude::*};
use std::convert::Infallible;

impl<'py> IntoPyObject<'py> for GID {
    type Target = PyAny;
    type Output = Bound<'py, PyAny>;
    type Error = Infallible;

    fn into_pyobject(self, py: Python<'py>) -> Result<Self::Output, Self::Error> {
        match self {
            GID::U64(v) => Ok(v.into_pyobject(py)?.into_any()),
            GID::Str(v) => Ok(v.into_pyobject(py)?.into_any()),
        }
    }
}

impl<'py> IntoPyObject<'py> for &GID {
    type Target = PyAny;
    type Output = Bound<'py, PyAny>;
    type Error = Infallible;

    fn into_pyobject(self, py: Python<'py>) -> Result<Self::Output, Self::Error> {
        match self {
            GID::U64(v) => Ok(v.into_pyobject(py)?.into_any()),
            GID::Str(v) => Ok(v.into_pyobject(py)?.into_any()),
        }
    }
}

impl<'source> FromPyObject<'source> for GID {
    fn extract_bound(id: &Bound<'source, PyAny>) -> PyResult<Self> {
        id.extract::<String>()
            .map(GID::Str)
            .or_else(|_| id.extract::<u64>().map(GID::U64))
            .map_err(|_| {
                let msg = "IDs need to be strings or an unsigned integers";
                PyTypeError::new_err(msg)
            })
    }
}