Skip to content

Commit d604743

Browse files
author
Bastiaan Marinus van de Weerd
committed
Fix supercclass keyword arguments in Python.
1 parent d0b5947 commit d604743

File tree

2 files changed

+11
-5
lines changed

2 files changed

+11
-5
lines changed

languages/tree-sitter-stack-graphs-python/src/stack-graphs.tsg

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -933,7 +933,10 @@ inherit .parent_module
933933

934934
(class_definition
935935
superclasses: (argument_list
936-
(_) @superclass)) @class
936+
[
937+
(identifier)
938+
(attribute)
939+
] @superclass)) @class
937940
{
938941
edge @class.super_scope -> @superclass.output
939942
}
Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,22 @@
11
class A:
2+
def __init_subclass__(cls, foo):
3+
pass
4+
25
def __init__(self):
36
self.some_attr = 2
47

58
def some_method(self):
69
print self
710

8-
class B(A):
11+
class B(A, foo="Bar"):
912
def method2(self):
1013
print self.some_attr, self.some_method()
11-
# ^ defined: 3
12-
# ^ defined: 5, 14
14+
# ^ defined: 6
15+
# ^ defined: 8, 17
1316

1417
def some_method(self):
1518
pass
1619

1720
def other(self):
1821
super().some_method()
19-
# ^ defined: 5
22+
# ^ defined: 8

0 commit comments

Comments
 (0)