validate_stub - add wrappers constants

2026-08-11 18:16:40 +00:00 · 2025-05-29 18:07:52 +05:00
parent bd5f5a30b3
commit 5fc549453c
2 changed files with 78 additions and 0 deletions
@@ -171,6 +171,17 @@ def get_names_tree_lines(tree: ast.Module) -> list[str]:
            node_name = get_function_node_name(node)
            assert isinstance(node_name, str)

+        elif isinstance(node, ast.Assign):
+            targets = node.targets
+            if not len(targets) == 1 or not isinstance(target := targets[0], ast.Name):
+                continue
+            node_name = target.id
+
+        elif isinstance(node, ast.AnnAssign):
+            target = node.target
+            assert isinstance(target, ast.Name)
+            node_name = target.id
+
        if node_name is not None:
            names_tree[node_name] = subnames