datafusion-server 0.21.0

Web server library for session-based queries using Arrow and other large datasets as data sources.
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

// convert_py_data.rs - Converts rust from to Python data structure.
// Sasaki, Naoki <nsasaki@sal.co.jp> February 23, 2023
//

#[cfg(feature = "plugin")]
use pyo3::{
    types::{PyAnyMethods, PyDict, PyDictMethods, PyString},
    Bound, IntoPyObject, Py, PyAny, PyErr, Python,
};

#[cfg(feature = "plugin")]
pub fn append_to_py_dict(
    py: Python,
    json_any_array: &[&serde_json::Value],
    py_dict: &Bound<'_, PyAny>,
) -> Result<(), PyErr> {
    for json_any in json_any_array {
        match json_any {
            serde_json::Value::Object(json_obj) => {
                for (key, value) in json_obj {
                    py_dict.set_item(key, to_py_any(py, value)?)?;
                }
            }
            _ => py_dict.set_item("value", to_py_any(py, json_any)?)?,
        }
    }

    Ok(())
}

#[cfg(feature = "plugin")]
fn to_py_any(py: Python, value: &serde_json::Value) -> Result<Py<PyAny>, PyErr> {
    use pyo3::types::PyList;

    Ok(match value {
        serde_json::Value::Null => py.None(),
        serde_json::Value::Bool(v) => v.into_pyobject(py).map(|b| b.to_owned().into())?,
        serde_json::Value::String(v) => PyString::new(py, v)
            .into_pyobject(py)
            .map(|b| b.clone().into())?,
        serde_json::Value::Number(v) => match v.as_f64() {
            Some(f) => f.into_pyobject(py).map(|b| b.clone().into())?,
            None => py.None(),
        },
        serde_json::Value::Array(values) => {
            let mut py_list = Vec::<Py<PyAny>>::new();
            for value in values {
                py_list.push(to_py_any(py, value)?);
            }
            PyList::new(py, &py_list)?.clone().into()
        }
        serde_json::Value::Object(map) => {
            let py_dict = PyDict::new(py);
            for (key, value) in map {
                py_dict.set_item(key, to_py_any(py, value)?)?;
            }
            py_dict.into_pyobject(py).map(|b| b.clone().into())?
        }
    })
}