Merge pull request #240 from digma-ai/bug/python_method_co_lookup_ignoring_class

added class aliases

Author: doppleware

src/services/languages/python/symbolAliasExtractor.ts

@@ -2,12 +2,28 @@ import { ISymbolAliasExtractor, SymbolInfo } from "../extractors";
 
 export class PythonSymbolAliasExtractor implements ISymbolAliasExtractor{
     public extractAliases(symbol: SymbolInfo): string[] {
-        var folders = symbol.id.split("/");
         let aliases = [];
-        //aliases.push(symbol.id);
+
+        const folderAndName = symbol.id.split("$_$");
+        if (folderAndName.length<2){
+            return [];
+        }
+        const folderOnly = folderAndName[0];
+        var folders = folderOnly.split("/");
+
+        if (symbol.name!=symbol.displayName){
+            aliases.push(folderOnly + "$_$" + symbol.displayName);
+        }  
+              //aliases.push(symbol.id);
         for (let i=1;i<folders.length;i++){
-            aliases.push(folders.slice(i, folders.length).join("/"));
+            aliases.push(folders.slice(i, folders.length).join("/") + "$_$" + symbol.name);
+            if (symbol.name!=symbol.displayName){
+                aliases.push(folders.slice(i, folders.length).join("/") + "$_$" + symbol.displayName);
+
+            }
         }
+
+        
         return aliases;
     }
 }

src/services/languages/symbolProvider.ts

@@ -150,7 +150,7 @@ export class SymbolProvider
                 const extractedMethods = await Promise.all(
                     methodExtractors.map(async (x) => await x.extractMethods(document, allDocSymbols, tokens))
                 );
-                const methods = extractedMethods.flat()
+                const methods = extractedMethods.flat();
                 return methods;
             }
             else {