1 files changed, 117 insertions, 0 deletions
diff --git a/rust/src/python/parsing.rs b/rust/src/python/parsing.rs
new file mode 100644
index 0000000..48fa64c
--- /dev/null
+++ b/rust/src/python/parsing.rs
@@ -0,0 +1,117 @@
+use pyo3::exceptions;
+use pyo3::prelude::*;
+use pyo3::types::PyDate;
+use pyo3::types::PyDateTime;
+use pyo3::types::PyTime;
+
+use crate::parsing::Parser;
+use crate::python::types::{Duration, FixedTimezone};
+
+#[pyfunction]
+pub fn parse_iso8601(py: Python, input: &str) -> PyResult<PyObject> {
+    let parsed = Parser::new(input).parse();
+
+    match parsed {
+        Ok(parsed) => match (parsed.datetime, parsed.duration, parsed.second_datetime) {
+            (Some(datetime), None, None) => match (datetime.has_date, datetime.has_time) {
+                (true, true) => match datetime.offset {
+                    Some(offset) => {
+                        let dt = PyDateTime::new(
+                            py,
+                            datetime.year as i32,
+                            datetime.month as u8,
+                            datetime.day as u8,
+                            datetime.hour as u8,
+                            datetime.minute as u8,
+                            datetime.second as u8,
+                            datetime.microsecond,
+                            Some(
+                                Py::new(py, FixedTimezone::new(offset, datetime.tzname))?
+                                    .to_object(py)
+                                    .extract(py)?,
+                            ),
+                        )?;
+
+                        Ok(dt.to_object(py))
+                    }
+                    None => {
+                        let dt = PyDateTime::new(
+                            py,
+                            datetime.year as i32,
+                            datetime.month as u8,
+                            datetime.day as u8,
+                            datetime.hour as u8,
+                            datetime.minute as u8,
+                            datetime.second as u8,
+                            datetime.microsecond,
+                            None,
+                        )?;
+
+                        Ok(dt.to_object(py))
+                    }
+                },
+                (true, false) => {
+                    let dt = PyDate::new(
+                        py,
+                        datetime.year as i32,
+                        datetime.month as u8,
+                        datetime.day as u8,
+                    )?;
+
+                    Ok(dt.to_object(py))
+                }
+                (false, true) => match datetime.offset {
+                    Some(offset) => {
+                        let dt = PyTime::new(
+                            py,
+                            datetime.hour as u8,
+                            datetime.minute as u8,
+                            datetime.second as u8,
+                            datetime.microsecond,
+                            Some(
+                                Py::new(py, FixedTimezone::new(offset, datetime.tzname))?
+                                    .to_object(py)
+                                    .extract(py)?,
+                            ),
+                        )?;
+
+                        Ok(dt.to_object(py))
+                    }
+                    None => {
+                        let dt = PyTime::new(
+                            py,
+                            datetime.hour as u8,
+                            datetime.minute as u8,
+                            datetime.second as u8,
+                            datetime.microsecond,
+                            None,
+                        )?;
+
+                        Ok(dt.to_object(py))
+                    }
+                },
+                (_, _) => Err(exceptions::PyValueError::new_err(
+                    "Parsing error".to_string(),
+                )),
+            },
+            (None, Some(duration), None) => Ok(Py::new(
+                py,
+                Duration::new(
+                    Some(duration.years),
+                    Some(duration.months),
+                    Some(duration.weeks),
+                    Some(duration.days),
+                    Some(duration.hours),
+                    Some(duration.minutes),
+                    Some(duration.seconds),
+                    Some(duration.microseconds),
+                ),
+            )?
+            .to_object(py)),
+            (_, _, _) => Err(exceptions::PyValueError::new_err(
+                "Not yet implemented".to_string(),
+            )),
+        },
+        Err(error) => Err(exceptions::PyValueError::new_err(error.to_string())),
+    }
+}