Proof of concept of using PEP384s PyType_Spec

PyO3 · Sep 1, 2020 · 5ce76e9 · 5ce76e9
1 parent 4a05f27
commit 5ce76e9
Show file tree

Hide file tree

Showing 3 changed files with 317 additions and 102 deletions.
diff --git a/src/class/basic.rs b/src/class/basic.rs
@@ -9,8 +9,9 @@
 //! [typeobj docs](https://docs.python.org/3/c-api/typeobj.html)
 
 use crate::callback::{HashCallbackOutput, IntoPyCallbackOutput};
+use crate::pyclass::maybe_push_slot;
 use crate::{exceptions, ffi, FromPyObject, PyAny, PyCell, PyClass, PyErr, PyObject, PyResult};
-use std::os::raw::c_int;
+use std::os::raw::{c_int, c_void};
 
 /// Operators for the __richcmp__ method
 #[derive(Debug)]
@@ -147,13 +148,38 @@ pub struct PyObjectMethods {
 
 #[doc(hidden)]
 impl PyObjectMethods {
-    pub(crate) fn update_typeobj(&self, type_object: &mut ffi::PyTypeObject) {
-        type_object.tp_str = self.tp_str;
-        type_object.tp_repr = self.tp_repr;
-        type_object.tp_hash = self.tp_hash;
-        type_object.tp_getattro = self.tp_getattro;
-        type_object.tp_richcompare = self.tp_richcompare;
-        type_object.tp_setattro = self.tp_setattro;
+    pub(crate) fn update_slots(&self, slots: &mut Vec<ffi::PyType_Slot>) {
+        maybe_push_slot(slots, ffi::Py_tp_str, self.tp_str.map(|v| v as *mut c_void));
+        maybe_push_slot(
+            slots,
+            ffi::Py_tp_repr,
+            self.tp_repr.map(|v| v as *mut c_void),
+        );
+        maybe_push_slot(
+            slots,
+            ffi::Py_tp_hash,
+            self.tp_hash.map(|v| v as *mut c_void),
+        );
+        maybe_push_slot(
+            slots,
+            ffi::Py_tp_getattro,
+            self.tp_getattro.map(|v| v as *mut c_void),
+        );
+        maybe_push_slot(
+            slots,
+            ffi::Py_tp_richcompare,
+            self.tp_richcompare.map(|v| v as *mut c_void),
+        );
+        maybe_push_slot(
+            slots,
+            ffi::Py_tp_setattro,
+            self.tp_setattro.map(|v| v as *mut c_void),
+        );
+        maybe_push_slot(
+            slots,
+            ffi::Py_nb_bool,
+            self.nb_bool.map(|v| v as *mut c_void),
+        );
     }
     // Set functions used by `#[pyproto]`.
     pub fn set_str<T>(&mut self)