WIP: PyFunction

pyo3-derive-backend/src/module.rs

@@ -209,7 +209,7 @@ pub fn add_fn_to_module(
     Ok(quote! {
         fn #function_wrapper_ident<'a>(
             args: impl Into<pyo3::derive_utils::WrapPyFunctionArguments<'a>>
-        ) -> pyo3::PyResult<pyo3::PyObject> {
+        ) -> pyo3::PyResult<&'a pyo3::types::PyFunction> {
             let arg = args.into();
             let (py, maybe_module) = arg.into_py_and_maybe_module();
             #wrapper
@@ -231,8 +231,7 @@ pub fn add_fn_to_module(
             };
 
             let function = unsafe {
-                pyo3::PyObject::from_owned_ptr(
-                    py,
+                py.from_owned_ptr::<pyo3::types::PyFunction>(
                     pyo3::ffi::PyCFunction_NewEx(
                         Box::into_raw(Box::new(_def.as_method_def())),
                         mod_ptr,

src/types/function.rs

@@ -0,0 +1,8 @@
+use crate::ffi;
+use crate::prelude::*;
+
+/// Represents a builtin Python function object.
+#[repr(transparent)]
+pub struct PyFunction(PyAny);
+
+pyobject_native_var_type!(PyFunction, ffi::PyCFunction_Type, ffi::PyCFunction_Check);

src/types/mod.rs

@@ -13,6 +13,7 @@ pub use self::datetime::{
 };
 pub use self::dict::{IntoPyDict, PyDict};
 pub use self::floatob::PyFloat;
+pub use self::function::PyFunction;
 pub use self::iterator::PyIterator;
 pub use self::list::PyList;
 pub use self::module::PyModule;
@@ -225,6 +226,7 @@ mod complex;
 mod datetime;
 mod dict;
 mod floatob;
+mod function;
 mod iterator;
 mod list;
 mod module;

src/types/module.rs

@@ -9,8 +9,8 @@ use crate::ffi;
 use crate::instance::PyNativeType;
 use crate::pyclass::PyClass;
 use crate::type_object::PyTypeObject;
-use crate::types::PyTuple;
 use crate::types::{PyAny, PyDict, PyList};
+use crate::types::{PyFunction, PyTuple};
 use crate::{AsPyPointer, IntoPy, Py, PyObject, Python};
 use std::ffi::{CStr, CString};
 use std::os::raw::c_char;
@@ -275,12 +275,11 @@ impl PyModule {
     /// ```
     pub fn add_function<'a>(
         &'a self,
-        wrapper: &impl Fn(&'a Self) -> PyResult<PyObject>,
+        wrapper: &impl Fn(&'a Self) -> PyResult<&'a PyFunction>,
     ) -> PyResult<()> {
-        let py = self.py();
         let function = wrapper(self)?;
-        let name = function.getattr(py, "__name__")?;
-        let name = name.extract(py)?;
+        let name = function.getattr("__name__")?;
+        let name = name.extract()?;
         self.add(name, function)
     }
 }