Handle name mangling

JelleZijlstra · JelleZijlstra · commit bd496227ebfe · 2023-04-26T19:10:52.000-06:00
diff --git a/Lib/test/test_type_params.py b/Lib/test/test_type_params.py
@@ -341,6 +341,29 @@ def foo[T: glb, U: cls](self): ...
         exec(code, {})
 
 
+class ManglingTest(unittest.TestCase):
+    def test_mangling(self):
+        class Foo[__T]:
+            param = __T
+            def meth[__U](self, arg: __T, arg2: __U):
+                return (__T, __U)
+            type Alias[__V] = (__T, __V)
+
+        T = Foo.__type_params__[0]
+        self.assertEqual(T.__name__, "__T")
+        U = Foo.meth.__type_params__[0]
+        self.assertEqual(U.__name__, "__U")
+        V = Foo.Alias.__type_params__[0]
+        self.assertEqual(V.__name__, "__V")
+
+        anno = Foo.meth.__annotations__
+        self.assertIs(anno["arg"], T)
+        self.assertIs(anno["arg2"], U)
+        self.assertEqual(Foo().meth(1, 2), (T, U))
+
+        self.assertEqual(Foo.Alias.__value__, (T, V))
+
+
 class TypeParamsTraditionalTypeVars(unittest.TestCase):
     def test_traditional_01(self):
         code = textwrap.dedent("""\
diff --git a/Python/compile.c b/Python/compile.c
@@ -2339,6 +2339,7 @@ compiler_class(struct compiler *c, stmt_ty s)
 
     asdl_typeparam_seq *typeparams = s->v.ClassDef.typeparams;
     if (asdl_seq_LEN(typeparams) > 0) {
+        Py_XSETREF(c->u->u_private, Py_NewRef(s->v.ClassDef.name));
         ADDOP(c, loc, PUSH_NULL);
         PyObject *typeparams_name = PyUnicode_FromFormat("<generic parameters of %U>",
                                                          s->v.ClassDef.name);
diff --git a/Python/symtable.c b/Python/symtable.c
@@ -1193,6 +1193,7 @@ symtable_enter_typeparam_block(struct symtable *st, identifier name,
         }
     }
     if (kind == ClassDef_kind) {
+        st->st_private = name;
         // This is used for setting the generic base
         _Py_DECLARE_STR(generic_base, ".generic_base");
         if (!symtable_add_def(st, &_Py_STR(generic_base), DEF_LOCAL,

Original file line number	Diff line number	Diff line change
`@@ -1193,6 +1193,7 @@ symtable_enter_typeparam_block(struct symtable *st, identifier name,`
`1193`	`1193`	`}`
`1194`	`1194`	`}`
`1195`	`1195`	`if (kind == ClassDef_kind) {`
	`1196`	`+ st->st_private = name;`
`1196`	`1197`	`// This is used for setting the generic base`
`1197`	`1198`	`_Py_DECLARE_STR(generic_base, ".generic_base");`
`1198`	`1199`	`if (!symtable_add_def(st, &_Py_STR(generic_base), DEF_LOCAL,`