1

为了创建一个使用泛型类型的结构的 pyo3 驱动的 Python 类,我想使用包装器来生成不需要为每个特定类型执行此操作所需的代码。

我创建了一个生成代码的宏,但我需要将宏生成的函数注册为 Python 模块的函数。

一种方法是跟踪宏中使用的标识以使用它们并wrap_pyfunction使用另一个宏生成,但我找不到任何相关的东西。

(当然,任何其他生成代码的解决方案都会受到热烈欢迎)

我现在拥有的(简化的)代码:

macro_rules! create_python_function {
    ($name:ident, $objtype:expr) => {
        paste!{
            #[pyclass(unsendable)]
            pub struct [<$name PyIface>] {
                obj: GenericStruct<$objtype>,
            }

            impl [<$name PyIface>]{
                pub fn new() -> [<$name PyIface>]{
                    [<$name PyIface>] {}
                }
            }

            pub fn [<create_object_ $name>]() -> [<$name PyIface>]{
                [<$name PyIface>]::new()
            }
        }
    };
}

create_python_function!(name, SpecificType);

#[pymodule]
fn mymodule(_py: Python, m: &PyModule) -> PyResult<()> {
    /* I want to auto-generate this with macro
     * m.add_function(wrap_pyfunction!(create_object_name, m)?).unwrap(); 
     */
    Ok(())
}
4

1 回答 1

1

宏不能共享它们的参数或状态。如果您不想重复标识符,请将mymodule定义移动到create_python_function宏中并更改宏以使用重复(The Rust Reference)

macro_rules! create_python_function {
    ($($name:ident => $objtype:ty),* $(,)?) => {
        $(
            paste! {
                #[pyclass(unsendable)]
                pub struct [<$name PyIface>] {
                    obj: GenericStruct<$objtype>,
                }

                impl [<$name PyIface>]{
                    pub fn new() -> [<$name PyIface>]{
                        [<$name PyIface>] { obj: todo!() }
                    }
                }

                pub fn [<create_object_ $name>]() -> [<$name PyIface>]{
                    [<$name PyIface>]::new()
                }
            }
        )*

        #[pymodule]
        fn mymodule(_py: Python, m: &PyModule) -> Result<(), ()> {
            $(
                paste! {
                    m.add_function(wrap_pyfunction!([<create_object_ $name>], m)?).unwrap();
                }
            )*
            Ok(())
        }
    };
}

struct Foo;
create_python_function!(
    foo => Foo,
    v => Vec<()>,
);
于 2021-05-26T07:49:15.207 回答