commit 71ba5f52d2a80e7beffc923c54c0b6345cd0637a
author kj <28750310+Fidget-Spinner@users.noreply.github.com> Wed Nov 18 05:45:08 2020 +0700
committer GitHub <noreply@github.com> Tue Nov 17 14:45:08 2020 -0800
tree d8a107c47353ca2c875f417321172d85221247ec
parent 48a9c0eb2a3304ea64d1b32fdf9db853d5d8c429

[3.9] bpo-42332: Add weakref slot to types.GenericAlias (GH-23250) (GH-23309)



(cherry picked from commit 384b7a4bd988986bca227c7e85c32d766da74708)

Objects/genericaliasobject.c

diff --git a/Objects/genericaliasobject.c b/Objects/genericaliasobject.c
index 3e850b5..4b6c8c6 100644
--- a/Objects/genericaliasobject.c
+++ b/Objects/genericaliasobject.c
@@ -9,6 +9,7 @@
     PyObject *origin;
     PyObject *args;
     PyObject *parameters;
+    PyObject* weakreflist;
 } gaobject;
 
 static void
@@ -17,6 +18,9 @@
     gaobject *alias = (gaobject *)self;
 
     _PyObject_GC_UNTRACK(self);
+    if (alias->weakreflist != NULL) {
+        PyObject_ClearWeakRefs((PyObject *)alias);
+    }
     Py_XDECREF(alias->origin);
     Py_XDECREF(alias->args);
     Py_XDECREF(alias->parameters);
@@ -593,6 +597,7 @@
     .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,
     .tp_traverse = ga_traverse,
     .tp_richcompare = ga_richcompare,
+    .tp_weaklistoffset = offsetof(gaobject, weakreflist),
     .tp_methods = ga_methods,
     .tp_members = ga_members,
     .tp_alloc = PyType_GenericAlloc,
@@ -624,6 +629,7 @@
     alias->origin = origin;
     alias->args = args;
     alias->parameters = NULL;
+    alias->weakreflist = NULL;
     _PyObject_GC_TRACK(alias);
     return (PyObject *)alias;
 }