py-flow-mapper 0.1.0b3.dev0__tar.gz → 0.1.0b4.dev0__tar.gz

{py_flow_mapper-0.1.0b3.dev0/src/py_flow_mapper.egg-info → py_flow_mapper-0.1.0b4.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-flow-mapper
-Version: 0.1.0b3.dev0
+Version: 0.1.0b4.dev0
 Summary: Python project analyzer and visualization tool
 Home-page: https://github.com/ArunKoundinya/py-flow-mapper
 Author: Arun Koundinya Parasa

{py_flow_mapper-0.1.0b3.dev0 → py_flow_mapper-0.1.0b4.dev0}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="py-flow-mapper",
-    version="0.1.0b3.dev",
+    version="0.1.0b4.dev",
     author="Arun Koundinya Parasa",
     author_email="parasa.arunkoundinya@gmail.com",
     description="Python project analyzer and visualization tool",

{py_flow_mapper-0.1.0b3.dev0 → py_flow_mapper-0.1.0b4.dev0}/src/py_flow_mapper/__init__.py

@@ -2,7 +2,7 @@
 PyFlowMapper - A Python project analyzer and visualization tool.
 """
 
-__version__ = "0.1.0b3.dev"
+__version__ = "0.1.0b4.dev"
 __author__ = "Arun Koundinya Parasa"
 __description__ = "Analyze Python projects and generate dependency graphs"
 __github__ = "https://github.com/ArunKoundinya/py-flow-mapper"

{py_flow_mapper-0.1.0b3.dev0 → py_flow_mapper-0.1.0b4.dev0}/src/py_flow_mapper/analyzer.py

@@ -107,8 +107,24 @@ class ProjectAnalyzer:
         """Find all Python files in the project, excluding virtual environments and other unwanted directories."""
         python_files = []
         
-        # Prefixes of directories to exclude
-        exclude_prefixes = ('venv', '.venv', 'env', '.env', '__pycache__', '.git', 'node_modules')
+        # Prefixes of directories to exclude        
+        exclude_prefixes = (
+            "venv", ".venv", "env", ".env", "virtualenv", ".virtualenv",
+            "__pycache__", ".mypy_cache", ".pytest_cache", ".ruff_cache",
+            ".coverage", "htmlcov", ".hypothesis",
+            ".git", ".hg", ".svn",
+            "build", "dist", ".eggs", "egg-info", "*.egg-info",
+            ".tox", ".nox", ".pdm-build",
+            "site-packages", "node_modules", "bower_components",
+            "docs", "doc", "site", "_site", "mkdocs",
+            ".ipynb_checkpoints", "notebooks", "examples", "experiments", "scratch",
+            "data", "datasets", "models", "checkpoints", "outputs", "results",
+            "logs", "tmp", "temp", "cache",
+            ".idea", ".vscode", ".DS_Store", "__MACOSX",
+            ".github", ".gitlab", ".circleci",
+            "tests", "test", "testing",
+            )
+
         
         for root, dirs, files in os.walk(self.base_path):
             # Filter out directories that start with any of the exclude prefixes
@@ -505,25 +521,19 @@ class DataFlowAnalyzer(ast.NodeVisitor):
         
     def visit_Assign(self, node):
         """Track assignments of function return values."""
-        # Get the variable name being assigned to
-        if isinstance(node.targets[0], ast.Name):
+        if node.targets and isinstance(node.targets[0], ast.Name):
             var_name = node.targets[0].id
-            
-            # Check if the value is a function call
+
             if isinstance(node.value, ast.Call):
                 call_info = self._extract_call_info(node.value)
                 if call_info:
-                    self.calls.append(call_info['func_name'])
-                    if var_name not in self.return_assignments:
-                        self.return_assignments[var_name] = []
-                    self.return_assignments[var_name].append(call_info['func_name'])
-            
-            # Check if the value is a Name (could be a variable holding a return value)
+                    self.return_assignments.setdefault(var_name, []).append(call_info["func_name"])
+
             elif isinstance(node.value, ast.Name):
-                # Track variable assignments for data flow
                 pass
-        
+
         self.generic_visit(node)
+
     
     def visit_Call(self, node):
         call_info = self._extract_call_info(node)

{py_flow_mapper-0.1.0b3.dev0 → py_flow_mapper-0.1.0b4.dev0}/src/py_flow_mapper/utils.py

@@ -120,17 +120,36 @@ def load_json(file_path: Path) -> Any:
 def get_project_structure(base_path: Path, exclude_dirs: List[str] = None) -> Dict[str, Any]:
     """Get the project directory structure."""
     if exclude_dirs is None:
-        exclude_dirs = ['venv', '.venv', '__pycache__', '.git', '.idea', '.vscode']
-    
-    structure = {}
-    
-    for item in sorted(base_path.iterdir()):
-        if item.name in exclude_dirs:
+        exclude_dirs = [
+            "venv", ".venv", "env", ".env", "virtualenv",
+            "__pycache__", ".mypy_cache", ".pytest_cache", ".ruff_cache",
+            ".coverage", "htmlcov",
+            ".git", ".hg", ".svn",
+            ".idea", ".vscode", ".DS_Store", "__MACOSX",
+            "build", "dist", ".eggs", ".tox", ".nox",
+            "node_modules", "site-packages",
+            "docs", "doc", "notebooks", ".ipynb_checkpoints",
+            "models", "outputs", "results",
+            "logs", "tmp", "temp",
+        ]
+
+    exclude_set = set(exclude_dirs)
+    structure: Dict[str, Any] = {}
+
+    for item in sorted(base_path.iterdir(), key=lambda p: p.name):
+        name = item.name
+
+        # Unified exclusion logic (works for both files and dirs)
+        if (
+            name in exclude_set
+            or name.endswith(".egg-info")                          # e.g., mypkg.egg-info
+            or any(name.startswith(excl) for excl in exclude_set)  # e.g., venv311, build_temp
+        ):
             continue
-        
+
         if item.is_dir():
-            structure[item.name] = get_project_structure(item, exclude_dirs)
+            structure[name] = get_project_structure(item, exclude_dirs)
         else:
-            structure[item.name] = 'file'
-    
-    return structure
+            structure[name] = "file"
+
+    return structure

{py_flow_mapper-0.1.0b3.dev0 → py_flow_mapper-0.1.0b4.dev0/src/py_flow_mapper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-flow-mapper
-Version: 0.1.0b3.dev0
+Version: 0.1.0b4.dev0
 Summary: Python project analyzer and visualization tool
 Home-page: https://github.com/ArunKoundinya/py-flow-mapper
 Author: Arun Koundinya Parasa