Skip to content

Commit 2af0d78

Browse files
committed
Python: Add default field to the relevant AST nodes
1 parent 55ee3eb commit 2af0d78

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

python/extractor/semmle/python/ast.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -500,10 +500,11 @@ def __init__(self, n, text):
500500
self.text = text
501501

502502
class ParamSpec(type_parameter):
503-
__slots__ = "name",
503+
__slots__ = "name", "default",
504504

505-
def __init__(self, name):
505+
def __init__(self, name, default):
506506
self.name = name
507+
self.default = default
507508

508509

509510

@@ -607,17 +608,19 @@ def __init__(self, name, type_parameters, value):
607608
self.value = value
608609

609610
class TypeVar(type_parameter):
610-
__slots__ = "name", "bound",
611+
__slots__ = "name", "bound", "default"
611612

612-
def __init__(self, name, bound):
613+
def __init__(self, name, bound, default):
613614
self.name = name
614615
self.bound = bound
616+
self.default = default
615617

616618
class TypeVarTuple(type_parameter):
617-
__slots__ = "name",
619+
__slots__ = "name", "default",
618620

619-
def __init__(self, name):
621+
def __init__(self, name, default):
620622
self.name = name
623+
self.default = default
621624

622625
class UnaryOp(expr):
623626
__slots__ = "op", "operand",

python/extractor/semmle/python/master.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -397,6 +397,7 @@
397397
Num.field('text', number)
398398

399399
ParamSpec.field('name', expr)
400+
ParamSpec.field('default', expr)
400401

401402
Print.field('dest', expr, 'destination')
402403
Print.field('values', expr_list)
@@ -448,8 +449,10 @@
448449

449450
TypeVar.field('name', expr)
450451
TypeVar.field('bound', expr)
452+
TypeVar.field('default', expr)
451453

452454
TypeVarTuple.field('name', expr)
455+
TypeVarTuple.field('default', expr)
453456

454457
UnaryOp.field('op', unaryop, 'operator')
455458
UnaryOp.field('operand', expr)

0 commit comments

Comments
 (0)