Function opt_py_to_timestamp

crates/python/src/convert.rs:46–68

Convert an optional timezone-aware Python datetime to a UTC timestamp.

(value: Option<&Bound<'_, PyAny>>)

/// Convert an optional timezone-aware Python datetime to a UTC timestamp.
pub fn opt_py_to_timestamp(value: Option<&Bound<'_, PyAny>>) -> PyResult<Option<DateTime<Utc>>> {
    let Some(timestamp) = value.filter(|timestamp| !timestamp.is_none()) else {
        return Ok(None);
    };

    let py = timestamp.py();
    let datetime_type = py.import("datetime")?.getattr("datetime")?;
    if !timestamp.is_instance(&datetime_type)? {
        return Err(pyo3::exceptions::PyTypeError::new_err(
            "timestamp must be a datetime.datetime object",
        ));
    }
    if timestamp.getattr("tzinfo")?.is_none() || timestamp.call_method0("utcoffset")?.is_none() {
        return Err(pyo3::exceptions::PyValueError::new_err(
            "timestamp datetime must be timezone-aware",
        ));
    }

    let iso_timestamp: String = timestamp.call_method0("isoformat")?.extract()?;
    DateTime::parse_from_rfc3339(&iso_timestamp)
        .map(|timestamp| Some(timestamp.with_timezone(&Utc)))
        .map_err(|e| pyo3::exceptions::PyValueError::new_err(format!("invalid timestamp: {e}")))
}

#[cfg(test)]
#[path = "../tests/unit/convert_tests.rs"]