pyobject 1.2.7.1__tar.gz → 1.2.7.3__tar.gz

{pyobject-1.2.7.1/pyobject.egg-info → pyobject-1.2.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyobject
-Version: 1.2.7.1
+Version: 1.2.7.3
 Summary: 一个提供操作Python对象底层工具的模块。A utility tool with some submodules for operating internal python objects.
 Home-page: https://github.com/qfcy/pyobject
 Author: qfcy

{pyobject-1.2.7.1 → pyobject-1.2.7.3}/pyobject/__init__.py

@@ -6,7 +6,7 @@ from warnings import warn
 from pprint import pprint
 
 __author__="qfcy"
-__version__="1.2.7.1"
+__version__="1.2.7.3"
 
 _ignore_names=["__builtins__","__doc__"]
 __all__=["objectname","bases","describe","desc"]
@@ -109,7 +109,8 @@ try:
     __all__.extend(["convptr","py_incref","py_decref","getrealrefcount",
                     "setrefcount","list_in","getrefcount_nogil","setrefcount_nogil",
                     "get_type_flag","set_type_flag","set_type_base","set_type_bases",
-                    "set_type_mro"])
+                    "set_type_mro","get_type_subclasses","set_type_subclasses",
+                    "set_type_subclasses_by_cls"])
 except ImportError:warn("Failed to import module pyobj_extension.")
 
 def test():

{pyobject-1.2.7.1 → pyobject-1.2.7.3}/pyobject/pyobj_extension.c

@@ -48,10 +48,11 @@ PyDoc_STRVAR(setrefcount_nogil_doc, u8"setrefcount_nogil(obj,ref_data)\n"
 #error "Python 3 is required"
 #endif
 
-#if PY_MINOR_VERSION >= 14
+#if PY_MINOR_VERSION >= 13
+#define _PY313
 #if defined(Py_GIL_DISABLED)
-// _PY314_NO_GIL: 是否启用无GIL的引用计数功能
-#define _PY314_NO_GIL
+// _PY313_NO_GIL: 是否启用无GIL的引用计数功能
+#define _PY313_NO_GIL
 #endif
 
 // 兼容旧版本Python
@@ -62,8 +63,21 @@ PyDoc_STRVAR(setrefcount_nogil_doc, u8"setrefcount_nogil(obj,ref_data)\n"
 #if PY_MINOR_VERSION <= 11
 #define Py_mod_multiple_interpreters 0
 #define Py_MOD_PER_INTERPRETER_GIL_SUPPORTED NULL
+#else
+#define _PY312
+#endif
+
+#endif
+
+#if PY_MINOR_VERSION >= 12
+#define _PY312P /* 3.12+所有版本 */
 #endif
 
+#ifdef _PY312P
+#define Py_BUILD_CORE
+#include <internal/pycore_object.h>        // _PyType_HasFeature()
+#include <internal/pycore_typeobject.h>    // struct type_cache
+#undef Py_BUILD_CORE
 #endif
 
 static const int REFCNT_DELTA=2; // 调用时新增的引用计数
@@ -98,8 +112,8 @@ PyObject *py_decref(PyObject *self, PyObject *args) {
 }
 
 PyObject *getrealrefcount(PyObject *self, PyObject *args) {
-#ifdef _PY314_NO_GIL
-    PyErr_SetString(PyExc_NotImplementedError, "getrealrefcount is not available in GIL-free versions of Python 3.14+");
+#ifdef _PY313_NO_GIL
+    PyErr_SetString(PyExc_NotImplementedError, "getrealrefcount is not available in GIL-free versions of Python 3.13+");
     return NULL;
 #else
     PyObject *obj;
@@ -110,8 +124,8 @@ PyObject *getrealrefcount(PyObject *self, PyObject *args) {
 #endif
 }
 PyObject *setrefcount(PyObject *self, PyObject *args, PyObject *kwargs) {
-#ifdef _PY314_NO_GIL
-    PyErr_SetString(PyExc_NotImplementedError, "setrefcount is not available in GIL-free versions of Python 3.14+");
+#ifdef _PY313_NO_GIL
+    PyErr_SetString(PyExc_NotImplementedError, "setrefcount is not available in GIL-free versions of Python 3.13+");
     return NULL;
 #else
     PyObject *obj;Py_ssize_t n;
@@ -124,8 +138,8 @@ PyObject *setrefcount(PyObject *self, PyObject *args, PyObject *kwargs) {
 #endif
 }
 PyObject *getrefcount_nogil(PyObject *self, PyObject *args){
-#ifndef _PY314_NO_GIL
-    PyErr_SetString(PyExc_NotImplementedError, "getrefcount_nogil is only available in GIL-free versions of Python 3.14+");
+#ifndef _PY313_NO_GIL
+    PyErr_SetString(PyExc_NotImplementedError, "getrefcount_nogil is only available in GIL-free versions of Python 3.13+");
     return NULL;
 #else
     PyObject *obj;
@@ -139,8 +153,8 @@ PyObject *getrefcount_nogil(PyObject *self, PyObject *args){
 #endif
 }
 PyObject *setrefcount_nogil(PyObject *self, PyObject *args, PyObject *kwargs){
-#ifndef _PY314_NO_GIL
-    PyErr_SetString(PyExc_NotImplementedError, "setrefcount_nogil is only available in GIL-free versions of Python 3.14+");
+#ifndef _PY313_NO_GIL
+    PyErr_SetString(PyExc_NotImplementedError, "setrefcount_nogil is only available in GIL-free versions of Python 3.13+");
     return NULL;
 #else
     PyObject *obj = NULL;
@@ -265,6 +279,106 @@ static PyObject* set_type_mro(PyObject *self, PyObject *args, PyObject *kwargs)
     Py_RETURN_NONE;
 }
 
+#ifdef _PY312
+#define managed_static_type_state static_builtin_state /* 兼容3.12 */
+#endif
+PyObject **_lookup_tp_subclasses(PyTypeObject *self){ // 从解释器的实现typeobject.c复制
+    #ifdef _PY312P
+    if (self->tp_flags & _Py_TPFLAGS_STATIC_BUILTIN) {
+        PyInterpreterState *interp = _PyInterpreterState_GET();
+        managed_static_type_state *state = _PyStaticType_GetState(interp, self);
+        // assert(state != NULL);
+        return &(state->tp_subclasses);
+    }
+    #endif
+    return (PyObject **)&(self->tp_subclasses);
+}
+#define getSubclasses(cls) (*_lookup_tp_subclasses(cls))
+static PyObject* get_type_subclasses(PyObject *self, PyObject *args, PyObject *kwargs) {
+    static char *kwlist[] = {"typeobj", NULL};
+    PyObject *typeobj;
+
+    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O", kwlist, &typeobj))
+        return NULL;
+
+    PyTypeObject *type = (PyTypeObject *)typeobj;
+    if(!getSubclasses(type)) Py_RETURN_NONE;
+    return getSubclasses(type);
+}
+
+static PyObject* set_type_subclasses(PyObject *self, PyObject *args, PyObject *kwargs) {
+    static char *kwlist[] = {"typeobj", "subclasses", NULL};
+    PyObject *typeobj, *subclasses;
+
+    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO", kwlist, &typeobj, &subclasses))
+        return NULL;
+
+    PyTypeObject *type = (PyTypeObject *)typeobj;
+    if(getSubclasses(type)) Py_DECREF(getSubclasses(type));
+    if (PyLong_Check(subclasses) && PyLong_AsLong(subclasses) == 0) {
+        getSubclasses(type) = NULL; // 传入0表示设为NULL
+    } else {
+        Py_INCREF(subclasses);
+        getSubclasses(type) = subclasses;
+    }
+    Py_RETURN_NONE;
+}
+static PyObject* set_type_subclasses_by_cls(PyObject *self, PyObject *args, PyObject *kwargs) {
+    static char *kwlist[] = {"typeobj", "cls", NULL};
+    PyObject *typeobj, *cls;
+
+    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO", kwlist, &typeobj, &cls))
+        return NULL;
+
+    PyTypeObject *dst = (PyTypeObject *)typeobj, *src = (PyTypeObject *)cls;
+    #ifndef _PY312
+    if(getSubclasses(src)) Py_INCREF(getSubclasses(src));
+    if(getSubclasses(dst)) Py_DECREF(getSubclasses(src));
+    #endif
+    getSubclasses(dst) = getSubclasses(src);
+    Py_RETURN_NONE;
+}
+
+// -- 3.12+的外部符号 --
+#ifdef _PY312P
+static inline size_t
+managed_static_type_index_get(PyTypeObject *self)
+{
+    // assert(managed_static_type_index_is_set(self));
+    /* We store a 1-based index so 0 can mean "not initialized". */
+    return (size_t)self->tp_subclasses - 1;
+}
+#ifdef _PY312
+static inline static_builtin_state *
+managed_static_type_state_get(PyInterpreterState *interp, PyTypeObject *self)
+{
+    return &(interp->types.builtins[managed_static_type_index_get(self)]);
+}
+#else
+static managed_static_type_state *
+managed_static_type_state_get(PyInterpreterState *interp, PyTypeObject *self)
+{
+    // It's probably a builtin type.
+    size_t index = managed_static_type_index_get(self);
+    managed_static_type_state *state =
+            &(interp->types.builtins.initialized[index]);
+    if (state->type == self) {
+        return state;
+    }
+    if (index > _Py_MAX_MANAGED_STATIC_EXT_TYPES) {
+        return state;
+    }
+    return &(interp->types.for_extensions.initialized[index]);
+}
+#endif
+/* For static types we store some state in an array on each interpreter. */
+managed_static_type_state *_PyStaticType_GetState(PyInterpreterState *interp, PyTypeObject *self){
+    assert(self->tp_flags & _Py_TPFLAGS_STATIC_BUILTIN);
+    return managed_static_type_state_get(interp, self);
+}
+#endif
+// ----
+
 /*
  * List of functions to add to pyobj_extension in exec_pyobj_extension().
  */
@@ -284,6 +398,10 @@ static PyMethodDef pyobj_extension_functions[] = {
     {"set_type_base", (PyCFunction)set_type_base, METH_VARARGS | METH_KEYWORDS, "set_type_base(typeobj: type, base: type)"},
     {"set_type_bases", (PyCFunction)set_type_bases, METH_VARARGS | METH_KEYWORDS, "set_type_bases(typeobj: type, bases: tuple)"},
     {"set_type_mro", (PyCFunction)set_type_mro, METH_VARARGS | METH_KEYWORDS, "set_type_mro(typeobj: type, mro: tuple)"},
+    {"get_type_subclasses", (PyCFunction)get_type_subclasses, METH_VARARGS | METH_KEYWORDS, "get_type_subclasses(typeobj: type)"},
+    {"set_type_subclasses", (PyCFunction)set_type_subclasses, METH_VARARGS | METH_KEYWORDS, "set_type_subclasses(typeobj: type, subclasses: dict)"},
+    {"set_type_subclasses_by_cls", (PyCFunction)set_type_subclasses_by_cls, METH_VARARGS | METH_KEYWORDS,
+     "set_type_subclasses_by_cls(typeobj: type, cls: type)"},
     { NULL, NULL, 0, NULL } /* marks end of array */
 };
 

{pyobject-1.2.7.1 → pyobject-1.2.7.3/pyobject.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyobject
-Version: 1.2.7.1
+Version: 1.2.7.3
 Summary: 一个提供操作Python对象底层工具的模块。A utility tool with some submodules for operating internal python objects.
 Home-page: https://github.com/qfcy/pyobject
 Author: qfcy