#![cfg(not(Py_LIMITED_API))]
use crate::ffi_ptr_ext::FfiPtrExt;
use crate::py_result_ext::PyResultExt;
use crate::types::{PyAny, PyBytes};
use crate::{ffi, Bound};
use crate::{AsPyPointer, PyResult, Python};
use std::os::raw::c_int;
pub const VERSION: i32 = 4;
#[cfg(feature = "gil-refs")]
#[deprecated(
since = "0.21.0",
note = "`dumps` will be replaced by `dumps_bound` in a future PyO3 version"
)]
pub fn dumps<'py>(
py: Python<'py>,
object: &impl AsPyPointer,
version: i32,
) -> PyResult<&'py PyBytes> {
dumps_bound(py, object, version).map(Bound::into_gil_ref)
}
pub fn dumps_bound<'py>(
py: Python<'py>,
object: &impl AsPyPointer,
version: i32,
) -> PyResult<Bound<'py, PyBytes>> {
unsafe {
ffi::PyMarshal_WriteObjectToString(object.as_ptr(), version as c_int)
.assume_owned_or_err(py)
.downcast_into_unchecked()
}
}
#[cfg(feature = "gil-refs")]
#[deprecated(
since = "0.21.0",
note = "`loads` will be replaced by `loads_bound` in a future PyO3 version"
)]
pub fn loads<'py, B>(py: Python<'py>, data: &B) -> PyResult<&'py PyAny>
where
B: AsRef<[u8]> + ?Sized,
{
loads_bound(py, data).map(Bound::into_gil_ref)
}
pub fn loads_bound<'py, B>(py: Python<'py>, data: &B) -> PyResult<Bound<'py, PyAny>>
where
B: AsRef<[u8]> + ?Sized,
{
let data = data.as_ref();
unsafe {
ffi::PyMarshal_ReadObjectFromString(data.as_ptr().cast(), data.len() as isize)
.assume_owned_or_err(py)
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::types::{bytes::PyBytesMethods, dict::PyDictMethods, PyDict};
#[test]
fn marshal_roundtrip() {
Python::with_gil(|py| {
let dict = PyDict::new_bound(py);
dict.set_item("aap", "noot").unwrap();
dict.set_item("mies", "wim").unwrap();
dict.set_item("zus", "jet").unwrap();
let pybytes = dumps_bound(py, &dict, VERSION).expect("marshalling failed");
let deserialized = loads_bound(py, pybytes.as_bytes()).expect("unmarshalling failed");
assert!(equal(py, &dict, &deserialized));
});
}
fn equal(_py: Python<'_>, a: &impl AsPyPointer, b: &impl AsPyPointer) -> bool {
unsafe { ffi::PyObject_RichCompareBool(a.as_ptr(), b.as_ptr(), ffi::Py_EQ) != 0 }
}
}