Python: Add default field to the relevant AST nodes

tausbn · tausbn · commit 2af0d7843582 · 2024-10-15T11:22:32.000Z
diff --git a/python/extractor/semmle/python/ast.py b/python/extractor/semmle/python/ast.py
@@ -500,10 +500,11 @@ def __init__(self, n, text):
         self.text = text
 
 class ParamSpec(type_parameter):
-    __slots__ = "name",
+    __slots__ = "name", "default",
 
-    def __init__(self, name):
+    def __init__(self, name, default):
         self.name = name
+        self.default = default
 
 
 
@@ -607,17 +608,19 @@ def __init__(self, name, type_parameters, value):
         self.value = value
 
 class TypeVar(type_parameter):
-    __slots__ = "name", "bound",
+    __slots__ = "name", "bound", "default"
 
-    def __init__(self, name, bound):
+    def __init__(self, name, bound, default):
         self.name = name
         self.bound = bound
+        self.default = default
 
 class TypeVarTuple(type_parameter):
-    __slots__ = "name",
+    __slots__ = "name", "default",
 
-    def __init__(self, name):
+    def __init__(self, name, default):
         self.name = name
+        self.default = default
 
 class UnaryOp(expr):
     __slots__ = "op", "operand",
diff --git a/python/extractor/semmle/python/master.py b/python/extractor/semmle/python/master.py
@@ -397,6 +397,7 @@
 Num.field('text', number)
 
 ParamSpec.field('name', expr)
+ParamSpec.field('default', expr)
 
 Print.field('dest', expr, 'destination')
 Print.field('values', expr_list)
@@ -448,8 +449,10 @@
 
 TypeVar.field('name', expr)
 TypeVar.field('bound', expr)
+TypeVar.field('default', expr)
 
 TypeVarTuple.field('name', expr)
+TypeVarTuple.field('default', expr)
 
 UnaryOp.field('op', unaryop, 'operator')
 UnaryOp.field('operand', expr)
