Set namespace for TypeParam nodes

While technically the value should be in an annotation scope,
I can't see any difference for our use. They are supposed to behave as if executing the parent's namespace.

Author: dflook

src/python_minifier/rename/mapper.py

@@ -61,6 +61,10 @@ def add_parent_to_functiondef(functiondef):
     for node in functiondef.decorator_list:
         add_parent(node, parent=functiondef, namespace=functiondef.namespace)
 
+    if hasattr(functiondef, 'type_params') and functiondef.type_params is not None:
+        for node in functiondef.type_params:
+            add_parent(node, parent=functiondef, namespace=functiondef.namespace)
+
     if hasattr(functiondef, 'returns') and functiondef.returns is not None:
         add_parent(functiondef.returns, parent=functiondef, namespace=functiondef.namespace)
 
@@ -89,6 +93,10 @@ def add_parent_to_classdef(classdef):
     for node in classdef.decorator_list:
         add_parent(node, parent=classdef, namespace=classdef.namespace)
 
+    if hasattr(classdef, 'type_params') and classdef.type_params is not None:
+        for node in classdef.type_params:
+            add_parent(node, parent=classdef, namespace=classdef.namespace)
+
 def add_parent_to_comprehension(node, namespace):
     assert is_ast_node(node, (ast.GeneratorExp, 'SetComp', 'DictComp', 'ListComp'))
 

src/python_minifier/transforms/remove_annotations.py

@@ -25,6 +25,9 @@ def visit_FunctionDef(self, node):
         node.body = self.suite(node.body, parent=node)
         node.decorator_list = [self.visit(d) for d in node.decorator_list]
 
+        if hasattr(node, 'type_params') and node.type_params is not None:
+            node.type_params = [self.visit(t) for t in node.type_params]
+
         if hasattr(node, 'returns') and self._options.remove_return_annotations:
             node.returns = None
 

src/python_minifier/transforms/remove_object_base.py

@@ -16,6 +16,9 @@ def visit_ClassDef(self, node):
             b for b in node.bases if not isinstance(b, ast.Name) or (isinstance(b, ast.Name) and b.id != 'object')
         ]
 
+        if hasattr(node, 'type_params') and node.type_params is not None:
+            node.type_params = [self.visit(t) for t in node.type_params]
+
         node.body = [self.visit(n) for n in node.body]
 
         return node

src/python_minifier/transforms/suite_transformer.py

@@ -50,6 +50,9 @@ def __call__(self, node):
     def visit_ClassDef(self, node):
         node.bases = [self.visit(b) for b in node.bases]
 
+        if hasattr(node, 'type_params') and node.type_params is not None:
+            node.type_params = [self.visit(t) for t in node.type_params]
+
         node.body = self.suite(node.body, parent=node)
         node.decorator_list = [self.visit(d) for d in node.decorator_list]