Skip to content

gh-109219: propagate used free vars through type param scopes [v2] #109389

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions Lib/test/test_type_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -694,6 +694,20 @@ class Cls:
cls = ns["outer"]()
self.assertEqual(cls.Alias.__value__, "class")

def test_nested_free(self):
ns = run_code("""
def f():
T = str
class C:
T = int
class D[U](T):
x = T
return C
""")
C = ns["f"]()
self.assertIn(int, C.D.__bases__)
self.assertIs(C.D.x, str)


class TypeParamsManglingTest(unittest.TestCase):
def test_mangling(self):
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Fix compiling type param scopes that use a name which is also free in an
inner scope.
17 changes: 13 additions & 4 deletions Python/symtable.c
Original file line number Diff line number Diff line change
Expand Up @@ -784,7 +784,8 @@ drop_class_free(PySTEntryObject *ste, PyObject *free)
static int
update_symbols(PyObject *symbols, PyObject *scopes,
PyObject *bound, PyObject *free,
PyObject *inlined_cells, int classflag)
PyObject *inlined_cells, int classflag,
PySTEntryObject *class_entry)
{
PyObject *name = NULL, *itr = NULL;
PyObject *v = NULL, *v_scope = NULL, *v_new = NULL, *v_free = NULL;
Expand Down Expand Up @@ -836,8 +837,16 @@ update_symbols(PyObject *symbols, PyObject *scopes,
the class that has the same name as a local
or global in the class scope.
*/
if (classflag &&
PyLong_AS_LONG(v) & (DEF_BOUND | DEF_GLOBAL)) {
PyObject *class_v = NULL;
if (class_entry) {
class_v = PyDict_GetItemWithError(class_entry->ste_symbols, name);
if (!class_v && PyErr_Occurred()) {
goto error;
}
}
if ((classflag &&
PyLong_AS_LONG(v) & (DEF_BOUND | DEF_GLOBAL)) ||
(class_v && PyLong_AS_LONG(class_v) & (DEF_BOUND | DEF_GLOBAL))) {
long flags = PyLong_AS_LONG(v) | DEF_FREE_CLASS;
v_new = PyLong_FromLong(flags);
if (!v_new) {
Expand Down Expand Up @@ -1078,7 +1087,7 @@ analyze_block(PySTEntryObject *ste, PyObject *bound, PyObject *free,
goto error;
/* Records the results of the analysis in the symbol table entry */
if (!update_symbols(ste->ste_symbols, scopes, bound, newfree, inlined_cells,
ste->ste_type == ClassBlock))
ste->ste_type == ClassBlock, class_entry))
goto error;

temp = PyNumber_InPlaceOr(free, newfree);
Expand Down