pyfixit-cli 0.1.0__py3-none-any.whl

pyfixit/imports.py

@@ -0,0 +1,95 @@
+import ast
+from pathlib import Path
+
+
+EXCLUDED_DIRECTORIES = {
+    ".venv",
+    "venv",
+    "env",
+    "test_env",
+    "__pycache__",
+    ".git",
+    "site-packages",
+    "build",
+    "dist",
+    ".pytest_cache",
+    "tests",
+}
+
+
+def find_python_files(project_directory):
+    project_path = Path(project_directory).resolve()
+
+    python_files = []
+
+    for path in project_path.rglob("*.py"):
+        path = path.resolve()
+
+        if any(part in EXCLUDED_DIRECTORIES for part in path.parts):
+            continue
+
+        # Don't analyze PyFixIt's own source code
+        if "src" in path.parts and "pyfixit" in path.parts:
+            continue
+
+        python_files.append(path)
+
+    return python_files
+
+
+def get_imports_from_file(file_path):
+    imports = []
+
+    try:
+        source = Path(file_path).read_text(
+            encoding="utf-8"
+        )
+
+        tree = ast.parse(source)
+
+    except (SyntaxError, UnicodeDecodeError):
+        return imports
+
+    for node in ast.walk(tree):
+
+        if isinstance(node, ast.Import):
+
+            for alias in node.names:
+
+                module = alias.name.split(".")[0]
+
+                if module not in imports:
+                    imports.append(module)
+
+        elif isinstance(node, ast.ImportFrom):
+
+            if node.module is not None:
+
+                module = node.module.split(".")[0]
+
+                if module not in imports:
+                    imports.append(module)
+
+    return imports
+
+
+def get_project_imports(project_directory="."):
+
+    python_files = find_python_files(
+        project_directory
+    )
+
+    all_imports = []
+
+    for file_path in python_files:
+
+        imports = get_imports_from_file(
+            file_path
+        )
+
+        for module in imports:
+
+            if module not in all_imports:
+                all_imports.append(module)
+
+    return all_imports

pyfixit/module_classifier.py

@@ -0,0 +1,62 @@
+import sys
+from pathlib import Path
+
+
+def is_standard_library(module_name):
+    if module_name in sys.stdlib_module_names:
+        return True
+
+    return False
+
+
+def is_local_module(module_name, project_directory="."):
+    project_directory = Path(project_directory)
+
+    possible_locations = [
+        project_directory,
+        project_directory / "src"
+    ]
+
+    for location in possible_locations:
+
+        module_file = location / f"{module_name}.py"
+        module_directory = location / module_name
+
+        if module_file.exists():
+            return True
+
+        if module_directory.is_dir():
+            init_file = module_directory / "__init__.py"
+
+            if init_file.exists():
+                return True
+
+    return False
+
+
+def classify_module(module_name, project_directory="."):
+    if is_standard_library(module_name):
+        return "standard_library"
+
+    if is_local_module(module_name, project_directory):
+        return "local"
+
+    return "third_party"
+
+
+def classify_modules(modules, project_directory="."):
+    classified = {
+        "standard_library": [],
+        "local": [],
+        "third_party": []
+    }
+
+    for module in modules:
+        category = classify_module(module, project_directory)
+
+        if module not in classified[category]:
+            classified[category].append(module)
+
+    return classified
+
+

pyfixit/package_mapping.py

@@ -0,0 +1,17 @@
+PACKAGE_IMPORT_MAP = {
+    "scikit-learn": "sklearn",
+    "beautifulsoup4": "bs4",
+    "python-dotenv": "dotenv",
+    "pillow": "PIL",
+    "opencv-python": "cv2",
+    "pyyaml": "yaml"
+}
+
+
+def get_import_name(package_name):
+    package_name = package_name.lower()
+
+    if package_name in PACKAGE_IMPORT_MAP:
+        return PACKAGE_IMPORT_MAP[package_name]
+
+    return package_name

pyfixit/package_names.py

@@ -0,0 +1,26 @@
+PACKAGE_IMPORT_NAMES = {
+    "scikit-learn": "sklearn",
+    "beautifulsoup4": "bs4",
+    "pillow": "PIL",
+    "python-dateutil": "dateutil",
+    "opencv-python": "cv2"
+}
+
+
+def get_import_name(package_name):
+    package_name = package_name.lower()
+
+    if package_name in PACKAGE_IMPORT_NAMES:
+        return PACKAGE_IMPORT_NAMES[package_name]
+
+    return package_name
+
+
+def get_package_name(import_name):
+    import_name = import_name.lower()
+
+    for package_name, mapped_import_name in PACKAGE_IMPORT_NAMES.items():
+        if mapped_import_name.lower() == import_name:
+            return package_name
+
+    return import_name

pyfixit/requirements_parser.py

@@ -0,0 +1,62 @@
+from packaging.version import Version
+
+
+def parse_requirement(requirement):
+    requirement = requirement.strip()
+
+    if not requirement:
+        return None
+
+    operators = ["==", ">=", "<=", "!=", ">", "<"]
+
+    for operator in operators:
+        if operator in requirement:
+            parts = requirement.split(operator, 1)
+
+            return {
+                "package": parts[0].strip(),
+                "operator": operator,
+                "version": parts[1].strip()
+            }
+
+    return {
+        "package": requirement,
+        "operator": None,
+        "version": None
+    }
+
+
+
+
+
+def check_version(installed_version, operator, required_version):
+    if installed_version is None:
+        return False
+
+    if operator is None or required_version is None:
+        return True
+
+    installed = Version(installed_version)
+    required = Version(required_version)
+
+    if operator == "==":
+        return installed == required
+
+    if operator == ">=":
+        return installed >= required
+
+    if operator == "<=":
+        return installed <= required
+
+    if operator == ">":
+        return installed > required
+
+    if operator == "<":
+        return installed < required
+
+    if operator == "!=":
+        return installed != required
+
+    return False
+
+