fix: method self

Author: nohehf

languages/tree-sitter-stack-graphs-python/query

@@ -0,0 +1,12 @@
+(class_definition
+    body: (
+        block
+            (function_definition
+                parameters: 
+                    (parameters
+                    . (identifier) @first_param
+                    )
+                body: (block) @method_body
+            )
+         )
+)

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

@@ -794,19 +794,29 @@ inherit .parent_module
   edge @name.def -> @func.call
 
   ; Prevent functions defined inside of method bodies from being treated like methods
+  ; There might be a bug here as calling .self_scope after this, will not fallback to the larger scope
   let @body.class_self_scope = #null
   let @body.class_member_attr_scope = #null
 }
 
-(function_definition
-  parameters: (parameters
-    . (identifier) @param)
-  body: (block) @body)
+; method definition
+(class_definition
+    body: (
+        block
+            (function_definition
+                parameters: 
+                    (parameters
+                    . (identifier) @first_param
+                    )
+                body: (block) @method_body
+            )
+         )
+) 
 {
-  edge @param.def -> @param.class_self_scope
-  edge @param.class_member_attr_scope -> @param.output
-  edge @param.output -> @body.after_scope
-  attr (@param.output) push_node = @param
+  edge @first_param.def -> @first_param.class_self_scope
+  edge @first_param.class_member_attr_scope -> @first_param.output
+  edge @first_param.output -> @method_body.after_scope
+  attr (@first_param.output) push_node = @first_param
 }
 
 [

languages/tree-sitter-stack-graphs-python/test/not_a_method.py

@@ -1,13 +1,20 @@
 class Foo:
     a = 1
 
-    # Self can be named anything
-    def first_method(actually_self):
-        return actually_self.a
-        #                    ^ defined: 2
+    def first_method(self):
+        self.a
+        #    ^ defined: 2
 
     def second_method(self):
+        self.a
+        #    ^ defined: 2
+
         # First argument here is not self
         def not_a_method(not_self):
             return not_self.a
             #               ^ defined:
+
+
+def function(self):
+    self.a
+    #    ^ defined:

languages/tree-sitter-stack-graphs-python/test/self.py

@@ -0,0 +1,11 @@
+class Foo:
+    a = 1
+
+    def mathod_1(self):
+        return self.a
+        #           ^ defined: 2
+
+    # Self can be named anything
+    def method_2(actually_self):
+        return actually_self.a
+        #                    ^ defined: 2