apply code review

jsbautista · jsbautista · commit b8f544e2cacb · 2025-05-13T11:53:55.000-05:00
diff --git a/pylsp/plugins/symbols.py b/pylsp/plugins/symbols.py
@@ -20,6 +20,10 @@ def pylsp_document_symbols(config, document):
     symbols = []
     exclude = set({})
     redefinitions = {}
+    pattern_import = (
+        re.compile
+        (r'^\s*(?!#)\s*(from\s+[.\w]+(\.[\w]+)*\s+import\s+[\w\s,()*]+|import\s+[\w\s,.*]+)')
+        )
 
     while definitions != []:
         d = definitions.pop(0)
@@ -28,11 +32,8 @@ def pylsp_document_symbols(config, document):
         if not add_import_symbols:
             # Skip if there's an import in the code the symbol is defined.
             code = d.get_line_code()
-            pattern = (
-                re.compile
-                (r'^\s*(?!#)(from\s+\w+(\.\w+)*\s+import\s+[\w,\s*]+|import\s+[\w,\s]+)')
-                )
-            if pattern.match(code):
+            
+            if pattern_import.match(code):
                 continue
 
             # Skip imported symbols comparing module names.
