Adding upstream version 3.0.0.upstream/3.0.0upstream

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

5 files changed, 217 insertions, 0 deletions

diff --git a/rust/src/python/types/duration.rs b/rust/src/python/types/duration.rs
new file mode 100644
index 0000000..fc18f4e
--- /dev/null
+++ b/rust/src/python/types/duration.rs
@@ -0,0 +1,59 @@
+use pyo3::prelude::*;
+
+#[pyclass(module = "_pendulum")]
+pub struct Duration {
+    #[pyo3(get, set)]
+    pub years: u32,
+    #[pyo3(get, set)]
+    pub months: u32,
+    #[pyo3(get, set)]
+    pub weeks: u32,
+    #[pyo3(get, set)]
+    pub days: u32,
+    #[pyo3(get, set)]
+    pub hours: u32,
+    #[pyo3(get, set)]
+    pub minutes: u32,
+    #[pyo3(get, set)]
+    pub seconds: u32,
+    #[pyo3(get, set)]
+    pub microseconds: u32,
+}
+
+#[pymethods]
+impl Duration {
+    #[new]
+    #[pyo3(signature = (years=0, months=0, weeks=0, days=0, hours=0, minutes=0, seconds=0, microseconds=0))]
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        years: Option<u32>,
+        months: Option<u32>,
+        weeks: Option<u32>,
+        days: Option<u32>,
+        hours: Option<u32>,
+        minutes: Option<u32>,
+        seconds: Option<u32>,
+        microseconds: Option<u32>,
+    ) -> Self {
+        Self {
+            years: years.unwrap_or(0),
+            months: months.unwrap_or(0),
+            weeks: weeks.unwrap_or(0),
+            days: days.unwrap_or(0),
+            hours: hours.unwrap_or(0),
+            minutes: minutes.unwrap_or(0),
+            seconds: seconds.unwrap_or(0),
+            microseconds: microseconds.unwrap_or(0),
+        }
+    }
+
+    #[getter]
+    fn remaining_days(&self) -> PyResult<u32> {
+        Ok(self.days)
+    }
+
+    #[getter]
+    fn remaining_seconds(&self) -> PyResult<u32> {
+        Ok(self.seconds)
+    }
+}
diff --git a/rust/src/python/types/interval.rs b/rust/src/python/types/interval.rs
new file mode 100644
index 0000000..7137493
--- /dev/null
+++ b/rust/src/python/types/interval.rs
@@ -0,0 +1,46 @@
+use pyo3::prelude::*;
+
+use pyo3::types::PyDelta;
+
+#[pyclass(extends=PyDelta)]
+#[derive(Default)]
+pub struct Interval {
+    pub days: i32,
+    pub seconds: i32,
+    pub microseconds: i32,
+}
+
+#[pymethods]
+impl Interval {
+    #[new]
+    #[pyo3(signature = (days=0, seconds=0, microseconds=0, milliseconds=0, minutes=0, hours=0, weeks=0))]
+    pub fn new(
+        py: Python,
+        days: Option<i32>,
+        seconds: Option<i32>,
+        microseconds: Option<i32>,
+        milliseconds: Option<i32>,
+        minutes: Option<i32>,
+        hours: Option<i32>,
+        weeks: Option<i32>,
+    ) -> PyResult<Self> {
+        println!("{} days", 31);
+        PyDelta::new(
+            py,
+            days.unwrap_or(0),
+            seconds.unwrap_or(0),
+            microseconds.unwrap_or(0),
+            true,
+        )?;
+
+        let f = Ok(Self {
+            days: days.unwrap_or(0),
+            seconds: seconds.unwrap_or(0),
+            microseconds: microseconds.unwrap_or(0),
+        });
+
+        println!("{} days", 31);
+
+        f
+    }
+}
diff --git a/rust/src/python/types/mod.rs b/rust/src/python/types/mod.rs
new file mode 100644
index 0000000..cba11df
--- /dev/null
+++ b/rust/src/python/types/mod.rs
@@ -0,0 +1,7 @@
+mod duration;
+mod precise_diff;
+mod timezone;
+
+pub use duration::Duration;
+pub use precise_diff::PreciseDiff;
+pub use timezone::FixedTimezone;
diff --git a/rust/src/python/types/precise_diff.rs b/rust/src/python/types/precise_diff.rs
new file mode 100644
index 0000000..64ca3a6
--- /dev/null
+++ b/rust/src/python/types/precise_diff.rs
@@ -0,0 +1,53 @@
+use pyo3::prelude::*;
+
+#[pyclass(module = "_pendulum")]
+pub struct PreciseDiff {
+    #[pyo3(get, set)]
+    pub years: i32,
+    #[pyo3(get, set)]
+    pub months: i32,
+    #[pyo3(get, set)]
+    pub days: i32,
+    #[pyo3(get, set)]
+    pub hours: i32,
+    #[pyo3(get, set)]
+    pub minutes: i32,
+    #[pyo3(get, set)]
+    pub seconds: i32,
+    #[pyo3(get, set)]
+    pub microseconds: i32,
+    #[pyo3(get, set)]
+    pub total_days: i32,
+}
+
+#[pymethods]
+impl PreciseDiff {
+    #[new]
+    #[pyo3(signature = (years=0, months=0, days=0, hours=0, minutes=0, seconds=0, microseconds=0, total_days=0))]
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        years: Option<i32>,
+        months: Option<i32>,
+        days: Option<i32>,
+        hours: Option<i32>,
+        minutes: Option<i32>,
+        seconds: Option<i32>,
+        microseconds: Option<i32>,
+        total_days: Option<i32>,
+    ) -> Self {
+        Self {
+            years: years.unwrap_or(0),
+            months: months.unwrap_or(0),
+            days: days.unwrap_or(0),
+            hours: hours.unwrap_or(0),
+            minutes: minutes.unwrap_or(0),
+            seconds: seconds.unwrap_or(0),
+            microseconds: microseconds.unwrap_or(0),
+            total_days: total_days.unwrap_or(0),
+        }
+    }
+
+    fn __repr__(&self) -> String {
+        format!("PreciseDiff(years={}, months={}, days={}, hours={}, minutes={}, seconds={}, microseconds={}, total_days={})", self.years, self.months, self.days, self.hours, self.minutes, self.seconds, self.microseconds, self.total_days)
+    }
+}
diff --git a/rust/src/python/types/timezone.rs b/rust/src/python/types/timezone.rs
new file mode 100644
index 0000000..1a8bbad
--- /dev/null
+++ b/rust/src/python/types/timezone.rs
@@ -0,0 +1,52 @@
+use pyo3::prelude::*;
+use pyo3::types::{PyDelta, PyDict, PyTzInfo};
+
+#[pyclass(module = "_pendulum", extends = PyTzInfo)]
+#[derive(Clone)]
+pub struct FixedTimezone {
+    offset: i32,
+    name: Option<String>,
+}
+
+#[pymethods]
+impl FixedTimezone {
+    #[new]
+    pub fn new(offset: i32, name: Option<String>) -> Self {
+        Self { offset, name }
+    }
+
+    fn utcoffset<'p>(&self, py: Python<'p>, _dt: &PyAny) -> PyResult<&'p PyDelta> {
+        PyDelta::new(py, 0, self.offset, 0, true)
+    }
+
+    fn tzname(&self, _dt: &PyAny) -> String {
+        self.__str__()
+    }
+
+    fn dst<'p>(&self, py: Python<'p>, _dt: &PyAny) -> PyResult<&'p PyDelta> {
+        PyDelta::new(py, 0, 0, 0, true)
+    }
+
+    fn __repr__(&self) -> String {
+        format!(
+            "FixedTimezone({}, name=\"{}\")",
+            self.offset,
+            self.__str__()
+        )
+    }
+
+    fn __str__(&self) -> String {
+        if let Some(n) = &self.name {
+            n.clone()
+        } else {
+            let sign = if self.offset < 0 { "-" } else { "+" };
+            let minutes = self.offset.abs() / 60;
+            let (hour, minute) = (minutes / 60, minutes % 60);
+            format!("{sign}{hour:.2}:{minute:.2}")
+        }
+    }
+
+    fn __deepcopy__(&self, py: Python, _memo: &PyDict) -> PyResult<Py<Self>> {
+        Py::new(py, self.clone())
+    }
+}