skylos 1.1.12__py3-none-any.whl → 1.2.1__py3-none-any.whl

skylos/test_aware.py

@@ -0,0 +1,66 @@
+import ast
+from skylos.constants import TEST_IMPORT_RE, TEST_DECOR_RE, TEST_FILE_RE
+
+class TestAwareVisitor:
+    
+    def __init__(self, filename=None):
+        self.is_test_file = False
+        self.test_decorated_lines = set()
+        
+        # mark as test file based on the file path/name NOT imports
+        if filename and TEST_FILE_RE.search(str(filename)):
+            self.is_test_file = True
+
+    def visit(self, node):
+        method = 'visit_' + node.__class__.__name__
+        visitor = getattr(self, method, self.generic_visit)
+        return visitor(node)
+
+    def generic_visit(self, node):
+        for field, value in ast.iter_fields(node):
+            if isinstance(value, list):
+                for item in value:
+                    if isinstance(item, ast.AST):
+                        self.visit(item)
+            elif isinstance(value, ast.AST):
+                self.visit(value)
+
+    def visit_Import(self, node):
+  
+        if self.is_test_file: 
+            for alias in node.names:
+                if TEST_IMPORT_RE.match(alias.name):
+                    pass  
+        self.generic_visit(node)
+
+    def visit_ImportFrom(self, node):
+        if self.is_test_file:
+            if node.module and TEST_IMPORT_RE.match(node.module):
+                pass 
+        self.generic_visit(node)
+
+
+    def visit_FunctionDef(self, node):
+        if (node.name.startswith('test_') or 
+            node.name.endswith('_test') or
+            any(node.name.startswith(prefix) for prefix in ['setup', 'teardown']) or
+            node.name in ['setUp', 'tearDown', 'setUpClass', 'tearDownClass', 
+                        'setUpModule', 'tearDownModule']):
+            self.test_decorated_lines.add(node.lineno)
+        
+        for deco in node.decorator_list:
+            name = self._decorator_name(deco)
+            if name and (TEST_DECOR_RE.match(name) or 'pytest' in name or 'fixture' in name):
+                self.test_decorated_lines.add(node.lineno)
+        self.generic_visit(node)
+
+    def visit_AsyncFunctionDef(self, node):
+        self.visit_FunctionDef(node)
+
+    def _decorator_name(self, deco):
+        if isinstance(deco, ast.Name):
+            return deco.id
+        if isinstance(deco, ast.Attribute):
+            parent = self._decorator_name(deco.value)
+            return f"{parent}.{deco.attr}" if parent else deco.attr
+        return ""

skylos/visitor.py

@@ -1,6 +1,11 @@
 #!/usr/bin/env python3
 import ast
 from pathlib import Path
+import os 
+import re, logging, traceback
+DBG = bool(int(os.getenv("SKYLOS_DEBUG", "0")))
+log = logging.getLogger("Skylos")
+
 
 PYTHON_BUILTINS={"print","len","str","int","float","list","dict","set","tuple","range","open","super","object","type","enumerate","zip","map","filter","sorted","reversed","sum","min","max","all","any","next","iter","repr","chr","ord","bytes","bytearray","memoryview","format","round","abs","pow","divmod","complex","hash","id","bool","callable","getattr","setattr","delattr","hasattr","isinstance","issubclass","globals","locals","vars","dir","property","classmethod","staticmethod"}
 DYNAMIC_PATTERNS={"getattr","globals","eval","exec"}
@@ -74,11 +79,23 @@ class Visitor(ast.NodeVisitor):
                 self.visit(node)
 
     def visit_string_annotation(self, annotation_str):
+        if not isinstance(annotation_str, str):
+            return
+
+        if DBG:
+            log.debug(f"[visitor] parsing string annotation {annotation_str!r} "
+                    f"in {self.filename}:{getattr(self, 'line', '?')}")
+
         try:
-            parsed = ast.parse(annotation_str, mode='eval')
+            parsed = ast.parse(annotation_str, mode="eval")
             self.visit(parsed.body)
-        except:
-            pass
+        except Exception:
+            if DBG:
+                log.debug("[visitor] inner-annotation parse failed:\n" +
+                        traceback.format_exc())
+            # keep going but dont swallow symbol names:
+            for tok in re.findall(r"[A-Za-z_][A-Za-z0-9_]*", annotation_str):
+                self.add_ref(tok)
 
     def visit_Import(self,node):
         for a in node.names:
@@ -250,6 +267,11 @@ class Visitor(ast.NodeVisitor):
                     if module_name != "self": 
                         qualified_name = f"{self.qual(module_name)}.{attr_name}"
                         self.add_ref(qualified_name)
+            
+            elif isinstance(node.args[0], ast.Name):
+                target_name = node.args[0].id
+                if target_name != "self":
+                    self.dyn.add(self.mod.split(".")[0] if self.mod else "")
         
         elif isinstance(node.func, ast.Name) and node.func.id == "globals":
             parent = getattr(node, 'parent', None)
@@ -259,6 +281,9 @@ class Visitor(ast.NodeVisitor):
                 func_name = parent.slice.value
                 self.add_ref(func_name)
                 self.add_ref(f"{self.mod}.{func_name}")
+        
+        elif isinstance(node.func, ast.Name) and node.func.id in ("eval", "exec"):
+            self.dyn.add(self.mod.split(".")[0] if self.mod else "")
 
     def visit_Name(self,node):
         if isinstance(node.ctx,ast.Load):

{skylos-1.1.12.dist-info → skylos-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: skylos
-Version: 1.1.12
+Version: 1.2.1
 Summary: A static analysis tool for Python codebases
 Author-email: oha <aaronoh2015@gmail.com>
 Requires-Python: >=3.9

skylos-1.2.1.dist-info/RECORD

@@ -0,0 +1,32 @@
+skylos/__init__.py,sha256=ZZWhq0TZ3G-yDi9inbYvMn8OBes-pqo1aYB5EivuxFI,152
+skylos/analyzer.py,sha256=zbWheTzHGbgfwbW0Z2euQyca1dDZL4BPscIHvAMVkME,15079
+skylos/cli.py,sha256=6udZY4vLU6PFzVMkaiCLCRcLXgquyHdmf4rIOAPosWc,18266
+skylos/constants.py,sha256=F1kMjuTxfw2hJjd0SeOQcgex5WhHMUhXCzOlVmwuACs,1230
+skylos/framework_aware.py,sha256=p7BGoFnzkpaLJoE3M5qgyIeZvXx17tOkdyXgeqGKmqU,5804
+skylos/test_aware.py,sha256=cduaWMcFsuzIEQWAMFPC58xGk8NUU3SbUS0ChRPedv8,2372
+skylos/visitor.py,sha256=MnWyzs0b2JOD2Nj1Iu7ZIZHCr7fRC92PY1bsT8bdgXg,12796
+test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test/compare_tools.py,sha256=0g9PDeJlbst-7hOaQzrL4MiJFQKpqM8q8VeBGzpPczg,22738
+test/conftest.py,sha256=57sTF6vLL5U0CVKwGQFJcRs6n7t1dEHIriQoSluNmAI,6418
+test/diagnostics.py,sha256=ExuFOCVpc9BDwNYapU96vj9RXLqxji32Sv6wVF4nJYU,13802
+test/test_analyzer.py,sha256=vZ-0cSF_otm7ZYdp6U3a8eWGwRZ6Tf0tVuJxYqcxdqM,22816
+test/test_changes_analyzer.py,sha256=l1hspCFz-sF8gqOilJvntUDuGckwhYsVtzvSRB13ZWw,5085
+test/test_cli.py,sha256=rtdKzanDRJT_F92jKkCQFdhvlfwVJxfXKO8Hrbn-mIg,13180
+test/test_constants.py,sha256=pMuDy0UpC81zENMDCeK6Bqmm3BR_HHZQSlMG-9TgOm0,12602
+test/test_framework_aware.py,sha256=tJ7bnhiGeSdsAvrWaGJO5ovTrw9n9BBk6o1HCw15yDA,11693
+test/test_integration.py,sha256=bNKGUe-w0xEZEdnoQNHbssvKMGs9u9fmFQTOz1lX9_k,12398
+test/test_skylos.py,sha256=kz77STrS4k3Eez5RDYwGxOg2WH3e7zNZPUYEaTLbGTs,15608
+test/test_test_aware.py,sha256=VmbR_MQY0m941CAxxux8OxJHIr7l8crfWRouSeBMhIo,9390
+test/test_visitor.py,sha256=xAbGv-XaozKm_0WJJhr0hMb6mLaJcbPz57G9-SWkxFU,22764
+test/sample_repo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test/sample_repo/app.py,sha256=M5XgoAn-LPz50mKAj_ZacRKf-Pg7I4HbjWP7Z9jE4a0,226
+test/sample_repo/sample_repo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test/sample_repo/sample_repo/commands.py,sha256=b6gQ9YDabt2yyfqGbOpLo0osF7wya8O4Lm7m8gtCr3g,2575
+test/sample_repo/sample_repo/models.py,sha256=xXIg3pToEZwKuUCmKX2vTlCF_VeFA0yZlvlBVPIy5Qw,3320
+test/sample_repo/sample_repo/routes.py,sha256=8yITrt55BwS01G7nWdESdx8LuxmReqop1zrGUKPeLi8,2475
+test/sample_repo/sample_repo/utils.py,sha256=S56hEYh8wkzwsD260MvQcmUFOkw2EjFU27nMLFE6G2k,1103
+skylos-1.2.1.dist-info/METADATA,sha256=DY1_WEnwZbz1xpWVbV7WwRRmxPqg8pVdS8DlTWs62Sc,224
+skylos-1.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+skylos-1.2.1.dist-info/entry_points.txt,sha256=zzRpN2ByznlQoLeuLolS_TFNYSQxUGBL1EXQsAd6bIA,43
+skylos-1.2.1.dist-info/top_level.txt,sha256=f8GA_7KwfaEopPMP8-EXDQXaqd4IbsOQPakZy01LkdQ,12
+skylos-1.2.1.dist-info/RECORD,,