msyrs/src/lib.rs
2024-11-17 20:44:05 +00:00

31 lines
673 B
Rust

#![doc = include_str!("../README.md")]
pub mod download;
pub mod utils;
use pyo3::{prelude::*, wrap_pymodule};
use pyo3_polars::PyDataFrame;
#[pyfunction]
pub fn load_qdf(file_path: &str) -> PyResult<PyDataFrame> {
Ok(PyDataFrame(
utils::qdf::load_quantamental_dataframe(file_path).unwrap(),
))
}
// ignore deprecated warning
#[allow(deprecated)]
#[pymodule]
pub fn qdf(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_function(wrap_pyfunction!(load_qdf, m)?)?;
Ok(())
}
#[allow(deprecated)]
#[pymodule]
pub fn msyrs(_py: Python, m: &PyModule) -> PyResult<()> {
// add qdf as a submodule
m.add_wrapped(wrap_pymodule!(qdf))?;
Ok(())
}