diff options
Diffstat (limited to 'rust/src/python/mod.rs')
-rw-r--r-- | rust/src/python/mod.rs | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/rust/src/python/mod.rs b/rust/src/python/mod.rs new file mode 100644 index 0000000..8d3cd41 --- /dev/null +++ b/rust/src/python/mod.rs @@ -0,0 +1,27 @@ +use pyo3::prelude::*; + +mod helpers; +mod parsing; +mod types; + +use helpers::{days_in_year, is_leap, is_long_year, local_time, precise_diff, week_day}; +use parsing::parse_iso8601; +use types::{Duration, PreciseDiff}; + +#[pymodule] +pub fn _pendulum(_py: Python<'_>, m: &PyModule) -> PyResult<()> { + m.add_function(wrap_pyfunction!(days_in_year, m)?)?; + m.add_function(wrap_pyfunction!(is_leap, m)?)?; + m.add_function(wrap_pyfunction!(is_long_year, m)?)?; + m.add_function(wrap_pyfunction!(local_time, m)?)?; + m.add_function(wrap_pyfunction!(week_day, m)?)?; + m.add_function(wrap_pyfunction!(parse_iso8601, m)?)?; + m.add_function(wrap_pyfunction!(precise_diff, m)?)?; + m.add_class::<Duration>()?; + m.add_class::<PreciseDiff>()?; + + #[cfg(not(feature = "mimalloc"))] + m.setattr("__pendulum_default_allocator__", true)?; // uses setattr so this is not in __all__ + + Ok(()) +} |