partially revert changes to PyTzInfoAccess
trait
This commit is contained in:
parent
7d245842d4
commit
f5b18468bc
|
@ -1 +1 @@
|
|||
Add lifetime parameter to `PyTzInfoAccess` trait and change the return type of `PyTzInfoAccess::get_tzinfo` to `Option<Bound<PyTzInfo>>`. For the deprecated gil-ref API, the trait is now implemented for `&'py PyTime` and `&'py PyDateTime` instead of `PyTime` and `PyDate`.
|
||||
Add lifetime parameter to `PyTzInfoAccess` trait. For the deprecated gil-ref API, the trait is now implemented for `&'py PyTime` and `&'py PyDateTime` instead of `PyTime` and `PyDate`.
|
||||
|
|
|
@ -161,12 +161,12 @@ fn datetime_from_timestamp<'p>(
|
|||
|
||||
#[pyfunction]
|
||||
fn get_datetime_tzinfo(dt: &PyDateTime) -> Option<Bound<'_, PyTzInfo>> {
|
||||
dt.get_tzinfo()
|
||||
dt.get_tzinfo_bound()
|
||||
}
|
||||
|
||||
#[pyfunction]
|
||||
fn get_time_tzinfo(dt: &PyTime) -> Option<Bound<'_, PyTzInfo>> {
|
||||
dt.get_tzinfo()
|
||||
dt.get_tzinfo_bound()
|
||||
}
|
||||
|
||||
#[pyclass(extends=PyTzInfo)]
|
||||
|
|
|
@ -248,7 +248,7 @@ impl FromPyObject<'_> for NaiveDateTime {
|
|||
// we return a hard error. We could silently remove tzinfo, or assume local timezone
|
||||
// and do a conversion, but better leave this decision to the user of the library.
|
||||
#[cfg(not(Py_LIMITED_API))]
|
||||
let has_tzinfo = dt.get_tzinfo().is_some();
|
||||
let has_tzinfo = dt.get_tzinfo_bound().is_some();
|
||||
#[cfg(Py_LIMITED_API)]
|
||||
let has_tzinfo = !dt.getattr(intern!(dt.py(), "tzinfo"))?.is_none();
|
||||
if has_tzinfo {
|
||||
|
@ -284,7 +284,7 @@ impl<Tz: TimeZone + for<'a> FromPyObject<'a>> FromPyObject<'_> for DateTime<Tz>
|
|||
check_type(dt, &DatetimeTypes::get(dt.py()).datetime, "PyDateTime")?;
|
||||
|
||||
#[cfg(not(Py_LIMITED_API))]
|
||||
let tzinfo = dt.get_tzinfo();
|
||||
let tzinfo = dt.get_tzinfo_bound();
|
||||
#[cfg(Py_LIMITED_API)]
|
||||
let tzinfo: Option<&PyAny> = dt.getattr(intern!(dt.py(), "tzinfo"))?.extract()?;
|
||||
|
||||
|
|
|
@ -163,12 +163,21 @@ pub trait PyTimeAccess {
|
|||
|
||||
/// Trait for accessing the components of a struct containing a tzinfo.
|
||||
pub trait PyTzInfoAccess<'py> {
|
||||
/// Deprecated form of `get_tzinfo_bound`.
|
||||
#[deprecated(
|
||||
since = "0.21.0",
|
||||
note = "`get_tzinfo` will be replaced by `get_tzinfo_bound` in a future PyO3 version"
|
||||
)]
|
||||
fn get_tzinfo(&self) -> Option<&'py PyTzInfo> {
|
||||
self.get_tzinfo_bound().map(Bound::into_gil_ref)
|
||||
}
|
||||
|
||||
/// Returns the tzinfo (which may be None).
|
||||
///
|
||||
/// Implementations should conform to the upstream documentation:
|
||||
/// <https://docs.python.org/3/c-api/datetime.html#c.PyDateTime_DATE_GET_TZINFO>
|
||||
/// <https://docs.python.org/3/c-api/datetime.html#c.PyDateTime_TIME_GET_TZINFO>
|
||||
fn get_tzinfo(&self) -> Option<Bound<'py, PyTzInfo>>;
|
||||
fn get_tzinfo_bound(&self) -> Option<Bound<'py, PyTzInfo>>;
|
||||
}
|
||||
|
||||
/// Bindings around `datetime.date`
|
||||
|
@ -413,13 +422,13 @@ impl PyTimeAccess for Bound<'_, PyDateTime> {
|
|||
}
|
||||
|
||||
impl<'py> PyTzInfoAccess<'py> for &'py PyDateTime {
|
||||
fn get_tzinfo(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
Bound::borrowed_from_gil_ref(self).get_tzinfo()
|
||||
fn get_tzinfo_bound(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
Bound::borrowed_from_gil_ref(self).get_tzinfo_bound()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'py> PyTzInfoAccess<'py> for Bound<'py, PyDateTime> {
|
||||
fn get_tzinfo(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
fn get_tzinfo_bound(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
let ptr = self.as_ptr() as *mut ffi::PyDateTime_DateTime;
|
||||
unsafe {
|
||||
if (*ptr).hastzinfo != 0 {
|
||||
|
@ -543,13 +552,13 @@ impl PyTimeAccess for Bound<'_, PyTime> {
|
|||
}
|
||||
|
||||
impl<'py> PyTzInfoAccess<'py> for &'py PyTime {
|
||||
fn get_tzinfo(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
Bound::borrowed_from_gil_ref(self).get_tzinfo()
|
||||
fn get_tzinfo_bound(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
Bound::borrowed_from_gil_ref(self).get_tzinfo_bound()
|
||||
}
|
||||
}
|
||||
|
||||
impl<'py> PyTzInfoAccess<'py> for Bound<'py, PyTime> {
|
||||
fn get_tzinfo(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
fn get_tzinfo_bound(&self) -> Option<Bound<'py, PyTzInfo>> {
|
||||
let ptr = self.as_ptr() as *mut ffi::PyDateTime_Time;
|
||||
unsafe {
|
||||
if (*ptr).hastzinfo != 0 {
|
||||
|
@ -725,6 +734,7 @@ mod tests {
|
|||
|
||||
#[test]
|
||||
#[cfg_attr(target_arch = "wasm32", ignore)] // DateTime import fails on wasm for mysterious reasons
|
||||
#[allow(deprecated)]
|
||||
fn test_get_tzinfo() {
|
||||
crate::Python::with_gil(|py| {
|
||||
let utc = timezone_utc(py);
|
||||
|
|
Loading…
Reference in a new issue