Merge pull request #6 from FSoft-AI4Code/dev/extended

Python: Handle class definitions with empty argument list

Author: minhna1112

src/codetext/parser/python_parser.py

@@ -114,7 +114,9 @@ def get_class_metadata(class_node, blob: str=None) -> Dict[str, str]:
                 argument_list = get_node_text(child).split(',')
                 for arg in argument_list:
                     item = re.sub(r'[^a-zA-Z0-9\_]', ' ', arg).split()
-                    metadata['parameters'][item[0].strip()] = None
+                    # Handle class definitions with empty argument list class ABC()
+                    if len(item) > 0:
+                        metadata['parameters'][item[0].strip()] = None
 
         # get __init__ function
         return metadata

tests/test_parser/test_python.py

@@ -27,7 +27,6 @@ def test_get_class_list(self):
         root = self.root_node
 
         class_list = PythonParser.get_class_list(root)
-        
         self.assertEqual(len(class_list), 1)
 
     def test_get_docstring(self):
@@ -59,21 +58,41 @@ def test_sample(arg1: str = "string", arg2 = "another_string"):
 
     def test_get_class_metadata(self):
         code_sample = '''
+        class ABC():
+            pass
+            
         class Sample(ABC):
             def __init__(self):
                 pass
 
             def test_sample(self, arg1: str = "string", arg2 = "another_string"):
                 return NotImplement()
+        
+        class ThisIsalsoAclass(ABC, Sample):
+            pass
         '''
         root = parse_code(code_sample, 'python').root_node
 
-        classes = list(PythonParser.get_class_list(root))[0]
-        metadata = PythonParser.get_class_metadata(classes)
-
+        
+        classes = list(PythonParser.get_class_list(root))
+        self.assertEqual(len(classes), 3)
+        
+        metadata = PythonParser.get_class_metadata(classes[0])
+        self.assertEqual(metadata['parameters'], {})
+        self.assertEqual(metadata['identifier'], 'ABC')
+        
+        
+        metadata = PythonParser.get_class_metadata(classes[1])
         self.assertEqual(metadata['parameters'], {'ABC': None})
         self.assertEqual(metadata['identifier'], 'Sample')
 
+        
+        metadata = PythonParser.get_class_metadata(classes[2])
+        self.assertEqual(metadata['parameters'], {'ABC': None, 'Sample': None})
+        self.assertEqual(metadata['identifier'], 'ThisIsalsoAclass')
+        
+        
+        
     def test_get_comment_list(self):
         root = self.root_node