#![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(())
}