skylos 1.0.7__py3-none-any.whl

skylos/visitor.py

@@ -0,0 +1,181 @@
+#!/usr/bin/env python3
+import ast,re
+from pathlib import Path
+
+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"}
+
+class Definition:
+    __slots__ = ('name', 'type', 'filename', 'line', 'simple_name', 'confidence', 'references', 'is_exported', 'in_init')
+    
+    def __init__(self, n, t, f, l):
+        self.name = n
+        self.type = t
+        self.filename = f
+        self.line = l
+        self.simple_name = n.split('.')[-1]
+        self.confidence = 100
+        self.references = 0
+        self.is_exported = False
+        self.in_init = "__init__.py" in str(f)
+    
+    def to_dict(self):
+        if self.type == "method" and "." in self.name:
+            parts = self.name.split(".")
+            if len(parts) >= 3:
+                output_name = ".".join(parts[-2:])
+            else:
+                output_name = self.name
+        else:
+            output_name = self.simple_name
+            
+        return{
+            "name": output_name,
+            "full_name": self.name,
+            "simple_name": self.simple_name,
+            "type": self.type,
+            "file": str(self.filename),
+            "basename": Path(self.filename).name,
+            "line": self.line,
+            "confidence": self.confidence,
+            "references": self.references
+        }
+
+class Visitor(ast.NodeVisitor):
+    def __init__(self,mod,file):
+        self.mod=mod
+        self.file=file
+        self.defs=[]
+        self.refs=[]
+        self.cls=None
+        self.alias={}
+        self.dyn=set()
+        self.exports=set()
+        self.current_function_scope = []
+
+    def add_def(self,n,t,l):
+        if n not in{d.name for d in self.defs}:self.defs.append(Definition(n,t,self.file,l))
+
+    def add_ref(self,n):self.refs.append((n,self.file))
+
+    def qual(self,n):
+        if n in self.alias:return self.alias[n]
+        if n in PYTHON_BUILTINS:return n
+        return f"{self.mod}.{n}"if self.mod else n
+    
+    def visit_Import(self,node):
+        for a in node.names:
+            full=a.name
+            self.alias[a.asname or a.name.split(".")[-1]]=full
+            self.add_def(full,"import",node.lineno)
+
+    def visit_ImportFrom(self,node):
+        if node.module is None:return
+        for a in node.names:
+            if a.name=="*":continue
+            base=node.module
+            if node.level:
+                parts=self.mod.split(".")
+                base=".".join(parts[:-node.level])+(f".{node.module}"if node.module else"")
+            full=f"{base}.{a.name}"
+            self.alias[a.asname or a.name]=full
+            self.add_def(full,"import",node.lineno)
+
+    def visit_FunctionDef(self,node):
+        outer_scope_prefix = '.'.join(self.current_function_scope) + '.' if self.current_function_scope else ''
+        
+        if self.cls:
+            name_parts = [self.mod, self.cls, outer_scope_prefix + node.name]
+        else:
+            name_parts = [self.mod, outer_scope_prefix + node.name]
+        
+        qualified_name = ".".join(filter(None, name_parts))
+
+        self.add_def(qualified_name,"method"if self.cls else"function",node.lineno)
+        
+        self.current_function_scope.append(node.name)
+        for d_node in node.decorator_list:
+            self.visit(d_node)
+        for stmt in node.body:
+            self.visit(stmt)
+        self.current_function_scope.pop()
+        
+    visit_AsyncFunctionDef=visit_FunctionDef
+
+    def visit_ClassDef(self,node):
+        cname=f"{self.mod}.{node.name}"
+        self.add_def(cname,"class",node.lineno)
+        prev=self.cls;self.cls=node.name
+        for b in node.body:self.visit(b)
+        self.cls=prev
+
+    def visit_Assign(self, node):
+        for target in node.targets:
+            if isinstance(target, ast.Name) and target.id == "__all__":
+                if isinstance(node.value, (ast.List, ast.Tuple)):
+                    for elt in node.value.elts:
+                        value = None
+                        if isinstance(elt, ast.Constant) and isinstance(elt.value, str):
+                            value = elt.value
+                        elif hasattr(elt, 's') and isinstance(elt.s, str):
+                            value = elt.s
+                            
+                        if value is not None:
+                            full_name = f"{self.mod}.{value}"
+                            self.add_ref(full_name)
+                            self.add_ref(value)
+        self.generic_visit(node)
+
+    def visit_Call(self, node):
+        self.generic_visit(node)
+        
+        if isinstance(node.func, ast.Name) and node.func.id in ("getattr", "hasattr") and len(node.args) >= 2:
+            if isinstance(node.args[1], ast.Constant) and isinstance(node.args[1].value, str):
+                attr_name = node.args[1].value
+                self.add_ref(attr_name)
+                
+                if isinstance(node.args[0], ast.Name):
+                    module_name = node.args[0].id
+                    if module_name != "self": 
+                        qualified_name = f"{self.qual(module_name)}.{attr_name}"
+                        self.add_ref(qualified_name)
+        
+        elif isinstance(node.func, ast.Name) and node.func.id == "globals":
+            parent = getattr(node, 'parent', None)
+            if (isinstance(parent, ast.Subscript) and 
+                isinstance(parent.slice, ast.Constant) and 
+                isinstance(parent.slice.value, str)):
+                func_name = parent.slice.value
+                self.add_ref(func_name)
+                self.add_ref(f"{self.mod}.{func_name}")
+        
+        elif (isinstance(node.func, ast.Attribute) and 
+            node.func.attr == "format" and 
+            isinstance(node.func.value, ast.Constant) and 
+            isinstance(node.func.value.value, str)):
+            fmt = node.func.value.value
+            if any(isinstance(k.arg, str) and k.arg is None for k in node.keywords):
+                for _, n, _, _ in re.findall(r'\{([^}:!]+)', fmt):
+                    if n:
+                        self.add_ref(self.qual(n))
+
+    def visit_Name(self,node):
+        if isinstance(node.ctx,ast.Load):
+            self.add_ref(self.qual(node.id))
+            if node.id in DYNAMIC_PATTERNS:self.dyn.add(self.mod.split(".")[0])
+
+    def visit_Attribute(self,node):
+        self.generic_visit(node)
+        if isinstance(node.ctx,ast.Load)and isinstance(node.value,ast.Name):
+            self.add_ref(f"{self.qual(node.value.id)}.{node.attr}")
+
+    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):
+                        item.parent = node
+                        self.visit(item)
+            elif isinstance(value, ast.AST):
+                value.parent = node
+                self.visit(value)

skylos-1.0.7.dist-info/METADATA

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: skylos
+Version: 1.0.7
+Summary: A static analysis tool for Python codebases
+Author-email: oha <aaronoh2015@gmail.com>
+Requires-Python: >=3.9
+Requires-Dist: inquirer>=3.0.0
+Dynamic: requires-python

skylos-1.0.7.dist-info/RECORD

@@ -0,0 +1,21 @@
+skylos/__init__.py,sha256=QzhqFsdZqpu52fXFnvDem4P56R3SOC0dKOHfekwFuEQ,151
+skylos/analyzer.py,sha256=EtL0IIw1NYzdGpXiOFQk4SXTSuTayVGpmGi0mZXL_Hk,7347
+skylos/cli.py,sha256=Pq71Gtc3d8E7JVYnEORK_LoLclG96A6b76SIGZh-C6s,13821
+skylos/visitor.py,sha256=DvRLXI-N2B6XG7VAB7cXEooD52oG1YNkl86iJbXe2-w,7336
+test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test/compare_tools.py,sha256=0g9PDeJlbst-7hOaQzrL4MiJFQKpqM8q8VeBGzpPczg,22738
+test/diagnostics.py,sha256=ExuFOCVpc9BDwNYapU96vj9RXLqxji32Sv6wVF4nJYU,13802
+test/test_skylos.py,sha256=kz77STrS4k3Eez5RDYwGxOg2WH3e7zNZPUYEaTLbGTs,15608
+test/test_visitor.py,sha256=bxUY_Zn_gLadZlz_n3Mu6rhVcExqElISwwVBo4eqVAY,7337
+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.0.7.dist-info/METADATA,sha256=Aqcg8zkHwx-l7ZGIXVVUPmXNQn9DZzZACiGOqM-YIks,224
+skylos-1.0.7.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+skylos-1.0.7.dist-info/entry_points.txt,sha256=zzRpN2ByznlQoLeuLolS_TFNYSQxUGBL1EXQsAd6bIA,43
+skylos-1.0.7.dist-info/top_level.txt,sha256=f8GA_7KwfaEopPMP8-EXDQXaqd4IbsOQPakZy01LkdQ,12
+skylos-1.0.7.dist-info/RECORD,,

skylos-1.0.7.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.7.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

skylos-1.0.7.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+skylos = skylos.cli:main

skylos-1.0.7.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+skylos
+test