Skip nontrivial class creation signatures

This commit is contained in:
2026-07-02 17:12:50 +02:00
parent 34e53e8692
commit 065c9ae7ec
2 changed files with 71 additions and 4 deletions
+14 -4
View File
@@ -605,6 +605,16 @@ def _invalidate_class_bindings(class_bindings, names):
class_bindings.pop(name, None)
def _is_trivially_safe_class_def(stmt):
return (
isinstance(stmt, ast.ClassDef)
and not stmt.decorator_list
and not stmt.bases
and not stmt.keywords
and not getattr(stmt, "type_params", ())
)
def _apply_module_stmt_to_env(stmt, env, class_bindings=None):
names = _mutating_call_target_names(stmt)
if _DYNAMIC_NAMESPACE_MUTATION in names:
@@ -616,10 +626,10 @@ def _apply_module_stmt_to_env(stmt, env, class_bindings=None):
_invalidate_env_names(env, names)
if isinstance(stmt, ast.ClassDef):
if class_bindings is not None:
if stmt.decorator_list:
class_bindings.pop(stmt.name, None)
else:
if _is_trivially_safe_class_def(stmt):
class_bindings[stmt.name] = (stmt, dict(env))
else:
class_bindings.pop(stmt.name, None)
_invalidate_env_name(env, stmt.name)
return
if isinstance(stmt, ast.Assign):
@@ -999,7 +1009,7 @@ def _update_class_aliases(stmt, class_aliases, class_bindings):
class_aliases.pop(name, None)
if isinstance(stmt, ast.ClassDef):
if stmt.name in class_bindings and not stmt.decorator_list:
if stmt.name in class_bindings and _is_trivially_safe_class_def(stmt):
class_aliases[stmt.name] = {stmt.name}
return