stock-trek 0.2.1

Stock Trek time-series analysis
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

use pyo3::prelude::*;
use pyo3::types::{PyDict, PyList, PyModule};
use pyo3::IntoPyObject;
use rust_decimal::Decimal;
use std::collections::HashMap;
use std::sync::OnceLock;

static DECIMAL_CLS: OnceLock<Py<PyAny>> = OnceLock::new();

pub fn hashmap_to_dict<K, V>(py: Python<'_>, map: &HashMap<K, V>) -> Py<PyDict>
where
    K: for<'py> IntoPyObject<'py> + Clone + Sync,
    V: for<'py> IntoPyObject<'py> + Clone + Sync,
{
    let dict = PyDict::new(py);
    for (key, value) in map {
        dict.set_item(key.clone(), value.clone()).unwrap();
    }
    dict.into()
}

pub fn vec_to_list<V>(py: Python<'_>, vec: &Vec<V>) -> Py<PyList>
where
    V: for<'py> IntoPyObject<'py> + Clone + Sync,
{
    let list = PyList::empty(py);
    for value in vec {
        list.add(value.clone()).unwrap();
    }
    list.into()
}

pub fn rust_decimal_to_py(py: Python<'_>, val: Decimal) -> PyResult<Py<PyAny>> {
    let s = val.to_string();
    let cls = get_decimal_cls(py).bind(py);
    let obj = cls.call1((s,))?;
    Ok(obj.unbind())
}

fn get_decimal_cls(py: Python<'_>) -> &Py<PyAny> {
    DECIMAL_CLS.get_or_init(|| {
        let module = PyModule::import(py, "decimal").unwrap();
        module.getattr("Decimal").unwrap().unbind()
    })
}