subclasses 方法在 CPython 中是如何实现的？

Question

文档说：

Each class keeps a list of weak references to its immediate subclasses. This method returns a list of all those references still alive.

但是每个 class 首先如何获得对其子 class 的弱引用列表？换句话说，当我创建

class B(A):
  pass

A 如何发现 B 刚刚对它进行了子class？这种机制是否足够强大以应对边缘情况（自定义元classes、分配给__bases__等）？

Answer 1

作为新 class 初始化的一部分，对该 class 的弱引用被添加到其每个基 class 的 tp_subclasses 成员中.您可以在 Objects/typeobject.c:

中的 Python 源代码中看到这一点

int
PyType_Ready(PyTypeObject *type)
{
    ...
    /* Link into each base class's list of subclasses */
    bases = type->tp_bases;
    n = PyTuple_GET_SIZE(bases);
    for (i = 0; i < n; i++) {
        PyObject *b = PyTuple_GET_ITEM(bases, i);
        if (PyType_Check(b) &&
            add_subclass((PyTypeObject *)b, type) < 0)
            goto error;
    }
    ...
}

static int
add_subclass(PyTypeObject *base, PyTypeObject *type)
{
    int result = -1;
    PyObject *dict, *key, *newobj;

    dict = base->tp_subclasses;
    if (dict == NULL) {
        base->tp_subclasses = dict = PyDict_New();
        if (dict == NULL)
            return -1;
    }
    assert(PyDict_CheckExact(dict));
    key = PyLong_FromVoidPtr((void *) type);
    if (key == NULL)
        return -1;
    newobj = PyWeakref_NewRef((PyObject *)type, NULL);
    if (newobj != NULL) {
        result = PyDict_SetItem(dict, key, newobj);
        Py_DECREF(newobj);
    }
    Py_DECREF(key);
    return result;
}

__bases__ 的 setter 也更新了每个新旧基地的 subclass 列表：

static int
type_set_bases(PyTypeObject *type, PyObject *new_bases, void *context)
{
    ...
    if (type->tp_bases == new_bases) {
        /* any base that was in __bases__ but now isn't, we
           need to remove |type| from its tp_subclasses.
           conversely, any class now in __bases__ that wasn't
           needs to have |type| added to its subclasses. */

        /* for now, sod that: just remove from all old_bases,
           add to all new_bases */
        remove_all_subclasses(type, old_bases);
        res = add_all_subclasses(type, new_bases);
        update_all_slots(type);
    }
    ...
}

请注意，如果元class 做了一些事情来自定义子class 关系的含义，__subclasses__ 将不会反映这一点。例如，issubclass(list, collections.abc.Iterable) 是 True，但 list 不会出现在从 collections.abc.Iterable.

开始的 __subclasses__ 树的搜索中

subclasses 方法在 CPython 中是如何实现的？

How is subclasses method implemented in CPython?

python

python-3.x

python-internals

__subclasses__ 方法在 CPython 中是如何实现的？

How is __subclasses__ method implemented in CPython?

python

python-3.x

python-internals

subclasses 方法在 CPython 中是如何实现的？

How is subclasses method implemented in CPython?