summaryrefslogtreecommitdiffstats
path: root/rust/src/python/mod.rs
diff options
context:
space:
mode:
Diffstat (limited to 'rust/src/python/mod.rs')
-rw-r--r--rust/src/python/mod.rs27
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(())
+}