Make #visitIdentifier: more robust

Author: jecisc

src/FAST-Python-Tools/FASTPythonImporter.class.st

@@ -8,6 +8,12 @@ Class {
 	#package : 'FAST-Python-Tools'
 }
 
+{ #category : 'parsing' }
+FASTPythonImporter class >> parse: aString [
+
+	^ self new parse: aString
+]
+
 { #category : 'parsing' }
 FASTPythonImporter >> deleteTreeSitterParser [
 

src/FAST-Python-Tools/TSFASTPythonImporter.class.st

@@ -26,7 +26,8 @@ TSFASTPythonImporter >> visitFunctionDefinition: aNode [
 TSFASTPythonImporter >> visitIdentifier: aNode [
 
 	"If we are in a field `name` then we are probably in a definition of an entity and we set my content as a name."
-	(context anySatisfy: [ :entry | entry field = #name ]) ifTrue: [ ^ context top fastEntity name: (self sourceCodeOf: aNode) ].
+	(context anySatisfy: [ :entry | entry field = #name and: [ #( 'function_definition' 'class_definition' ) includes: entry tsNode type ] ]) ifTrue: [
+		^ context top fastEntity name: (self sourceCodeOf: aNode) ].
 
 	^ self createEntityForNode: aNode
 ]