Add all_methods_ids property to ClassDefinition and update unique_id handling in PythonParser

Author: BrunoV21

codetide/core/models.py

@@ -133,6 +133,12 @@ def references(self)->List[CodeReference]:
             sum([method.references for method in self.methods], [])
         )
         return all_references
+    
+    @property
+    def all_methods_ids(self)->List[str]:
+        return [
+            method.unique_id for method in self.methods
+        ]
 
 class CodeFileModel(BaseModel):
     """Representation of a single code file"""
@@ -658,7 +664,7 @@ def get(self, unique_id :Union[str, List[str]], degree :int=0, as_string :bool=F
             new_references_ids = []
             for reference in references_ids:
                 element = self._cached_elements.get(reference)
-                if element is not None and element.unique_id not in retrieved_ids:
+                if element is not None and (element.unique_id not in retrieved_ids):
                     retrieved_elements.append(element)
                     retrieved_ids.append(element.unique_id)
 

codetide/parsers/python_parser.py

@@ -306,15 +306,19 @@ def _process_function_definition(cls, node: Node, code: bytes, codeFile: CodeFil
         if is_class_method:
             if raw is None:
                 raw = cls._get_content(code, node, preserve_indentation=True)
-            codeFile.classes[-1].add_method(
-                MethodDefinition(
+            
+            codeFile.classes[-1].add_method(MethodDefinition(
                     name=definition,
                     signature=signature,
                     decorators=decorators,
                     modifiers=modifiers,
                     raw=raw
                 )
             )
+            currentMethod = codeFile.classes[-1].methods[-1]
+            if currentMethod.unique_id in codeFile.classes[-1].all_methods_ids[:-1] and currentMethod.decorators:
+                currentMethod.unique_id = f"{currentMethod.unique_id}{''.join(currentMethod.decorators)}"
+
         else:
             if raw is None:
                 raw = cls._get_content(code, node)