Adding upstream version 3.0.0.upstream/3.0.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

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(())
+}