graphrecords_python/graphrecord/
value.rs

1use super::{Lut, traits::DeepFrom};
2use crate::{conversion_lut::ConversionLut, graphrecord::errors::PyGraphRecordError};
3use chrono::{NaiveDateTime, TimeDelta};
4use graphrecords_core::graphrecord::GraphRecordValue;
5use pyo3::{
6    Borrowed, Bound, FromPyObject, IntoPyObject, IntoPyObjectExt, PyAny, PyErr, PyResult, Python,
7    types::{PyAnyMethods, PyBool, PyDateTime, PyDelta, PyFloat, PyInt, PyString},
8};
9use std::ops::Deref;
10
11#[repr(transparent)]
12#[derive(Clone, Debug)]
13pub struct PyGraphRecordValue(GraphRecordValue);
14
15impl From<GraphRecordValue> for PyGraphRecordValue {
16    fn from(value: GraphRecordValue) -> Self {
17        Self(value)
18    }
19}
20
21impl From<PyGraphRecordValue> for GraphRecordValue {
22    fn from(value: PyGraphRecordValue) -> Self {
23        value.0
24    }
25}
26
27impl DeepFrom<PyGraphRecordValue> for GraphRecordValue {
28    fn deep_from(value: PyGraphRecordValue) -> Self {
29        value.into()
30    }
31}
32
33impl DeepFrom<GraphRecordValue> for PyGraphRecordValue {
34    fn deep_from(value: GraphRecordValue) -> Self {
35        value.into()
36    }
37}
38
39impl Deref for PyGraphRecordValue {
40    type Target = GraphRecordValue;
41
42    fn deref(&self) -> &Self::Target {
43        &self.0
44    }
45}
46
47static GRAPHRECORDVALUE_CONVERSION_LUT: Lut<GraphRecordValue> = ConversionLut::new();
48
49#[allow(clippy::unnecessary_wraps)]
50pub(crate) fn convert_pyobject_to_graphrecordvalue(
51    ob: &Bound<'_, PyAny>,
52) -> PyResult<GraphRecordValue> {
53    fn convert_string(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
54        Ok(GraphRecordValue::String(
55            ob.extract::<String>().expect("Extraction must succeed"),
56        ))
57    }
58
59    fn convert_int(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
60        Ok(GraphRecordValue::Int(
61            ob.extract::<i64>().expect("Extraction must succeed"),
62        ))
63    }
64
65    fn convert_float(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
66        Ok(GraphRecordValue::Float(
67            ob.extract::<f64>().expect("Extraction must succeed"),
68        ))
69    }
70
71    fn convert_bool(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
72        Ok(GraphRecordValue::Bool(
73            ob.extract::<bool>().expect("Extraction must succeed"),
74        ))
75    }
76
77    fn convert_datetime(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
78        Ok(GraphRecordValue::DateTime(
79            ob.extract::<NaiveDateTime>()
80                .expect("Extraction must succeed"),
81        ))
82    }
83
84    fn convert_duration(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
85        Ok(GraphRecordValue::Duration(
86            ob.extract::<TimeDelta>().expect("Extraction must succeed"),
87        ))
88    }
89
90    const fn convert_null(_ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
91        Ok(GraphRecordValue::Null)
92    }
93
94    fn throw_error(ob: &Bound<'_, PyAny>) -> PyResult<GraphRecordValue> {
95        Err(
96            PyGraphRecordError::Conversion(format!("Failed to convert {ob} into GraphRecordValue"))
97                .into(),
98        )
99    }
100
101    let type_pointer = ob.get_type_ptr() as usize;
102
103    let conversion_function = GRAPHRECORDVALUE_CONVERSION_LUT.get_or_insert(type_pointer, || {
104        if ob.is_instance_of::<PyString>() {
105            convert_string
106        } else if ob.is_instance_of::<PyBool>() {
107            convert_bool
108        } else if ob.is_instance_of::<PyInt>() {
109            convert_int
110        } else if ob.is_instance_of::<PyFloat>() {
111            convert_float
112        } else if ob.is_instance_of::<PyDateTime>() {
113            convert_datetime
114        } else if ob.is_instance_of::<PyDelta>() {
115            convert_duration
116        } else if ob.is_none() {
117            convert_null
118        } else {
119            throw_error
120        }
121    });
122
123    conversion_function(ob)
124}
125
126impl FromPyObject<'_, '_> for PyGraphRecordValue {
127    type Error = PyErr;
128
129    fn extract(ob: Borrowed<'_, '_, PyAny>) -> PyResult<Self> {
130        convert_pyobject_to_graphrecordvalue(&ob).map(Self::from)
131    }
132}
133
134impl<'py> IntoPyObject<'py> for PyGraphRecordValue {
135    type Error = PyErr;
136    type Output = Bound<'py, Self::Target>;
137    type Target = PyAny;
138
139    fn into_pyobject(self, py: Python<'py>) -> Result<Self::Output, Self::Error> {
140        match self.0 {
141            GraphRecordValue::String(value) => value.into_bound_py_any(py),
142            GraphRecordValue::Int(value) => value.into_bound_py_any(py),
143            GraphRecordValue::Float(value) => value.into_bound_py_any(py),
144            GraphRecordValue::Bool(value) => value.into_bound_py_any(py),
145            GraphRecordValue::DateTime(value) => value.into_bound_py_any(py),
146            GraphRecordValue::Duration(value) => value.into_bound_py_any(py),
147            GraphRecordValue::Null => py.None().into_bound_py_any(py),
148        }
149    }
150}
graphrecords_python/graphrecord/value.rs

graphrecords_python/graphrecord/
value.rs
