auto-code-fixer 0.3.0__tar.gz → 0.3.1__tar.gz

{auto_code_fixer-0.3.0 → auto_code_fixer-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: auto-code-fixer
-Version: 0.3.0
+Version: 0.3.1
 Summary: Automatically fix Python code using ChatGPT
 Author-email: Arif Shah <ashah7775@gmail.com>
 License: MIT

auto_code_fixer-0.3.1/auto_code_fixer/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.1"

{auto_code_fixer-0.3.0 → auto_code_fixer-0.3.1}/auto_code_fixer/installer.py

@@ -4,6 +4,15 @@ import subprocess
 
 _MISSING_RE = re.compile(r"No module named '([^']+)'")
 
+# common import-name -> pip-package mappings
+IMPORT_TO_PIP = {
+    "PIL": "Pillow",
+    "cv2": "opencv-python",
+    "sklearn": "scikit-learn",
+    "yaml": "PyYAML",
+    "Crypto": "pycryptodome",
+}
+
 
 def parse_missing_module(stderr: str) -> str | None:
     m = _MISSING_RE.search(stderr or "")
@@ -43,5 +52,5 @@ def check_and_install_missing_lib(stderr: str, *, python_exe: str, project_root:
     if is_local_module(lib, project_root=project_root):
         return False
 
-    # naive mapping; user can always fix the requirement name manually.
-    return install_package(lib, python_exe=python_exe)
+    pkg = IMPORT_TO_PIP.get(lib, lib)
+    return install_package(pkg, python_exe=python_exe)

{auto_code_fixer-0.3.0 → auto_code_fixer-0.3.1}/auto_code_fixer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: auto-code-fixer
-Version: 0.3.0
+Version: 0.3.1
 Summary: Automatically fix Python code using ChatGPT
 Author-email: Arif Shah <ashah7775@gmail.com>
 License: MIT

{auto_code_fixer-0.3.0 → auto_code_fixer-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "auto-code-fixer"
-version = "0.3.0"
+version = "0.3.1"
 description = "Automatically fix Python code using ChatGPT"
 readme = "README.md"
 

auto_code_fixer-0.3.0/auto_code_fixer/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.3.0"