monarch_types/
lib.rs

1/*
2 * Copyright (c) Meta Platforms, Inc. and affiliates.
3 * All rights reserved.
4 *
5 * This source code is licensed under the BSD-style license found in the
6 * LICENSE file in the root directory of this source tree.
7 */
8
9#![feature(assert_matches)]
10
11mod pyobject;
12mod python;
13mod pytree;
14
15use pyo3::PyErr;
16use pyo3::exceptions::PyValueError;
17pub use pyobject::PickledPyObject;
18pub use python::SerializablePyErr;
19pub use python::TryIntoPyObjectUnsafe;
20pub use pytree::PyTree;
21
22/// Macro to generate a Python object lookup function with caching
23///
24/// # Arguments
25/// * `$fn_name` - Name of the Rust function to generate
26/// * `$python_path` - Path to the Python object as a string (e.g., "module.submodule.function")
27#[macro_export]
28macro_rules! py_global {
29    ($fn_name:ident, $python_module:literal, $python_class:literal) => {
30        fn $fn_name<'py>(py: ::pyo3::Python<'py>) -> ::pyo3::Bound<'py, ::pyo3::PyAny> {
31            static CACHE: ::pyo3::sync::GILOnceCell<::pyo3::PyObject> =
32                ::pyo3::sync::GILOnceCell::new();
33            CACHE
34                .import(py, $python_module, $python_class)
35                .unwrap()
36                .clone()
37        }
38    };
39}
40
41pub trait MapPyErr<T> {
42    fn map_pyerr(self) -> Result<T, PyErr>;
43}
44impl<T, E> MapPyErr<T> for Result<T, E>
45where
46    E: ToString,
47{
48    fn map_pyerr(self) -> Result<T, PyErr> {
49        self.map_err(|err| PyErr::new::<PyValueError, _>(err.to_string()))
50    }
51}
monarch_types/lib.rs

monarch_types/
lib.rs
