Something went wrong. Try again.
The format that's super!
Something went wrong. Try again.
6.5 kB · 187 lines
Rust
at commit eba20c8d
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188use pyo3::{exceptions::PyValueError, prelude::*};
mod de;mod ser;mod temporal;
#[pyclass(frozen)]pub(crate) struct DuperType;
#[pyclass(frozen, module = "duper")]#[derive(Debug, Clone)]pub(crate) struct Duper { pub(crate) identifier: Option<duper::DuperIdentifier<'static>>,}
impl Duper { pub(crate) fn from_identifier<'a>(identifier: &duper::DuperIdentifier<'a>) -> PyResult<Self> { Ok(Self { identifier: Some(identifier.static_clone()), }) }}
#[pymethods]impl Duper { #[new] fn new(identifier: Option<String>) -> PyResult<Self> { match identifier { Some(identifier) => match duper::DuperIdentifier::try_from(identifier) { Ok(identifier) => Self::from_identifier(&identifier), Err(error) => Err(PyErr::new::<PyValueError, String>(error.to_string())), }, None => Ok(Self { identifier: None }), } }
#[getter] fn identifier(&self) -> Option<&str> { self.identifier .as_ref() .map(|identifier| identifier.as_ref()) }
fn __repr__(&self) -> String { match self.identifier.as_ref() { Some(identifier) => format!("Duper('{}')", identifier.as_ref()), None => "Duper(None)".into(), } }
fn __str__(&self) -> String { self.__repr__() }}
#[pyo3::pymodule(name = "_duper")]mod duper_py { use duper::{DuperParser, DuperValue, PrettyPrinter, Serializer}; use pyo3::{ exceptions::PyValueError, prelude::*, types::{PyInt, PyString}, };
#[pymodule_export] use crate::{Duper, DuperType, temporal::TemporalString}; use crate::{de::Visitor, ser::serialize_pyany};
#[pyfunction] #[pyo3(signature = (obj, *, indent=None, strip_identifiers=false, minify=false))] fn dumps<'py>( obj: Bound<'py, PyAny>, indent: Option<Bound<'py, PyAny>>, strip_identifiers: bool, minify: bool, ) -> PyResult<String> { let value: DuperValue = serialize_pyany(obj)?; if let Some(indent) = indent { if minify { Err(PyValueError::new_err( "cannot stringify with both indent and minify options", )) } else if indent.is_instance_of::<PyInt>() { let indent: usize = indent.extract()?; Ok(PrettyPrinter::new( strip_identifiers, &(0..indent).map(|_| ' ').collect::<String>(), ) .map_err(|error| PyErr::new::<PyValueError, String>(error.into()))? .pretty_print(&value)) } else if indent.is_instance_of::<PyString>() { let indent: &str = indent.extract()?; Ok(PrettyPrinter::new(strip_identifiers, indent) .map_err(|error| PyErr::new::<PyValueError, String>(error.into()))? .pretty_print(&value)) } else { Err(PyErr::new::<PyValueError, String>(format!( "expect indent to be string or int, found {indent:?}" ))) } } else { Ok(Serializer::new(strip_identifiers, minify).serialize(&value)) } }
#[pyfunction] #[pyo3(signature = (obj, fp, *, indent=None, strip_identifiers=false, minify=false))] fn dump<'py>( obj: Bound<'py, PyAny>, fp: Bound<'py, PyAny>, indent: Option<Bound<'py, PyAny>>, strip_identifiers: bool, minify: bool, ) -> PyResult<()> { let value: DuperValue = serialize_pyany(obj)?; fp.call_method1( "write", (if let Some(indent) = indent { if minify { return Err(PyValueError::new_err( "cannot stringify with both indent and minify options", )); } else if indent.is_instance_of::<PyInt>() { let indent: usize = indent.extract()?; PrettyPrinter::new( strip_identifiers, &(0..indent).map(|_| ' ').collect::<String>(), ) .map_err(|error| PyErr::new::<PyValueError, String>(error.into()))? .pretty_print(&value) } else if indent.is_instance_of::<PyString>() { let indent: &str = indent.extract()?; PrettyPrinter::new(strip_identifiers, indent) .map_err(|error| PyErr::new::<PyValueError, String>(error.into()))? .pretty_print(&value) } else { return Err(PyErr::new::<PyValueError, String>(format!( "expect indent to be string or int, found {indent:?}" ))); } } else { Serializer::new(strip_identifiers, minify).serialize(&value) },), )?; Ok(()) }
#[pyfunction] #[pyo3(signature = (s, *, parse_any=false))] fn loads<'py>(py: Python<'py>, s: &str, parse_any: bool) -> PyResult<Bound<'py, PyAny>> { let value = match parse_any { true => DuperParser::parse_duper_value(s), false => DuperParser::parse_duper_trunk(s), } .map_err(|err| { PyErr::new::<PyValueError, String>( DuperParser::prettify_error(s, &err, None).unwrap_or_else(|_| format!("{err:?}")), ) })?; value .accept(&mut Visitor { py }) .map(|visitor_value| visitor_value.value) }
#[pyfunction] #[pyo3(signature = (fp, *, parse_any=false))] fn load<'py>( py: Python<'py>, fp: Bound<'py, PyAny>, parse_any: bool, ) -> PyResult<Bound<'py, PyAny>> { let read = fp.call_method0("read")?; let s: &str = read.extract()?; let value = match parse_any { true => DuperParser::parse_duper_value(s), false => DuperParser::parse_duper_trunk(s), } .map_err(|err| { PyErr::new::<PyValueError, String>( DuperParser::prettify_error(s, &err, None).unwrap_or_else(|_| format!("{err:?}")), ) })?; value .accept(&mut Visitor { py }) .map(|visitor_value| visitor_value.value) }}