auto-code-fixer 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

auto_code_fixer/__init__.py

@@ -1,2 +1 @@
-__version__ = "0.2.5"
-
+__version__ = "0.2.6"

auto_code_fixer/cli.py

@@ -2,7 +2,6 @@ import argparse
 import shutil
 import tempfile
 import time
-import os
 from decouple import config
 
 from auto_code_fixer.runner import run_code
@@ -14,7 +13,6 @@ from auto_code_fixer.utils import (
     log,
     set_verbose,
 )
-from auto_code_fixer.guard import import_hijacked
 from auto_code_fixer import __version__
 
 MAX_RETRIES = 5
@@ -24,22 +22,16 @@ def fix_file(file_path, project_root, api_key, ask, verbose):
     log(f"Processing file: {file_path}")
 
     with open(file_path) as f:
-        current_code = f.read()   # 🔑 SOURCE OF TRUTH
+        original_code = f.read()
 
     temp_dir = tempfile.mkdtemp(prefix="codefix_")
-    temp_file = os.path.join(temp_dir, "temp.py")
-
-    # write initial code to temp
-    with open(temp_file, "w") as f:
-        f.write(current_code)
+    temp_file = f"{temp_dir}/temp.py"
+    shutil.copy(file_path, temp_file)
 
     for attempt in range(MAX_RETRIES):
         log(f"Run attempt #{attempt + 1}")
 
-        retcode, stdout, stderr = run_code(
-            temp_file,
-            project_root=project_root,
-        )
+        retcode, stdout, stderr = run_code(temp_file)
 
         if verbose:
             if stdout:
@@ -55,15 +47,13 @@ def fix_file(file_path, project_root, api_key, ask, verbose):
                     confirm = input(
                         f"Overwrite original file '{file_path}' with fixed version? (y/n): "
                     ).strip().lower()
+
                     if confirm != "y":
                         log("User declined overwrite", "WARN")
                         shutil.rmtree(temp_dir)
                         return
 
-                # ✅ WRITE FROM MEMORY, NOT TEMP FILE
-                with open(file_path, "w") as f:
-                    f.write(current_code)
-
+                shutil.copy(temp_file, file_path)
                 log(f"File updated: {file_path}")
 
             shutil.rmtree(temp_dir)
@@ -73,55 +63,24 @@ def fix_file(file_path, project_root, api_key, ask, verbose):
         log("Error detected ❌", "ERROR")
         print(stderr)
 
-        if check_and_install_missing_lib(stderr, project_root):
+        if check_and_install_missing_lib(stderr):
             log("Missing dependency installed, retrying…")
             time.sleep(1)
             continue
 
         log("Sending code & error to GPT 🧠", "DEBUG")
-
         fixed_code = fix_code_with_gpt(
-            current_code,
+            open(temp_file).read(),
             stderr,
             api_key,
         )
 
-        if import_hijacked(current_code, fixed_code):
-            log(
-                "GPT attempted to hijack local imports. Rejecting fix.",
-                "ERROR",
-            )
-            break
-
-        if fixed_code.strip() == current_code.strip():
+        if fixed_code.strip() == open(temp_file).read().strip():
             log("GPT returned no changes. Stopping.", "WARN")
             break
 
-        import ast
-
-        # Validate GPT output BEFORE accepting it
-        try:
-            ast.parse(fixed_code)
-        except SyntaxError:
-            log(
-                "GPT returned invalid or empty Python code. Rejecting fix.",
-                "ERROR",
-            )
-            break
-
-        if not fixed_code.strip():
-            log(
-                "GPT returned empty code. Rejecting fix.",
-                "ERROR",
-            )
-            break
-
-        # ✅ SAFE: update memory only after validation
-        current_code = fixed_code
-
-        # then update temp file for next execution
         with open(temp_file, "w") as f:
-            f.write(current_code)
+            f.write(fixed_code)
 
         log("Code updated by GPT ✏️")
         time.sleep(1)
@@ -130,7 +89,6 @@ def fix_file(file_path, project_root, api_key, ask, verbose):
     shutil.rmtree(temp_dir)
 
 
-
 def main():
     parser = argparse.ArgumentParser(
         description="Auto-fix Python code using ChatGPT"
@@ -151,23 +109,35 @@ def main():
     parser.add_argument("--project-root", default=".")
     parser.add_argument("--api-key")
 
+    # ✅ Proper boolean flags
     ask_group = parser.add_mutually_exclusive_group()
-    ask_group.add_argument("--ask", action="store_true")
-    ask_group.add_argument("--no-ask", action="store_true")
+    ask_group.add_argument(
+        "--ask",
+        action="store_true",
+        help="Ask before overwriting files",
+    )
+    ask_group.add_argument(
+        "--no-ask",
+        action="store_true",
+        help="Do not ask before overwriting files",
+    )
 
-    parser.add_argument("--verbose", action="store_true")
+    parser.add_argument(
+        "--verbose",
+        action="store_true",
+        help="Enable verbose/debug output",
+    )
 
     args = parser.parse_args()
 
     if not args.entry_file:
         parser.error("the following arguments are required: entry_file")
 
-    # 🔑 NORMALIZE ROOT PATH ONCE
-    project_root = os.path.abspath(args.project_root)
-
+    # ENV defaults
     env_ask = config("AUTO_CODE_FIXER_ASK", default=False, cast=bool)
     env_verbose = config("AUTO_CODE_FIXER_VERBOSE", default=False, cast=bool)
 
+    # Final ask resolution (CLI overrides ENV)
     if args.ask:
         ask = True
     elif args.no_ask:
@@ -180,4 +150,4 @@ def main():
 
     files = discover_all_files(args.entry_file)
     for file in files:
-        fix_file(file, project_root, args.api_key, ask, verbose)
+        fix_file(file, args.project_root, args.api_key, ask, verbose)

auto_code_fixer/fixer.py

@@ -1,60 +1,55 @@
-from openai import OpenAI
+import os
+import time
 from decouple import config
+from openai import OpenAI
 
-client = OpenAI(api_key=config("chat_gpt_secret_key"))
 
+def get_openai_client(api_key=None):
+    key = (
+            api_key
+            or os.getenv("OPENAI_API_KEY")
+            or config("OPENAI_API_KEY", default=None)
+    )
 
-def fix_code_with_gpt(original_code: str, error_log: str, api_key: str | None):
-    """
-    Send code + error to GPT and return fixed code.
-    Enforces strict rules to prevent semantic-breaking fixes.
-    """
+    if not key:
+        raise RuntimeError(
+            "OpenAI API key not found. "
+            "Set OPENAI_API_KEY env var or .env file or pass --api-key"
+        )
 
-    system_prompt = """
-You are a senior Python engineer fixing code.
+    return OpenAI(api_key=key)
 
-STRICT RULES (DO NOT VIOLATE):
-- NEVER replace or wrap local imports with try/except.
-- NEVER inline, duplicate, or redefine code from other local files.
-- NEVER change import statements for local modules.
-- Assume all local imports are valid and must remain unchanged.
-- If an import fails, assume environment configuration is incorrect.
-- Only fix syntax errors or runtime errors in the current file.
-- Return ONLY valid Python code. No markdown. No explanation.
-"""
 
-    user_prompt = f"""
-The following Python code has an error.
+def fix_code_with_gpt(original_code, error_log, api_key=None):
+    client = get_openai_client(api_key)
 
-CODE:
+    prompt = f"""
+You are a senior Python engineer.
+Fix the following Python code so it runs without errors.
+
+Code:
 {original_code}
 
-ERROR:
+Error:
 {error_log}
 
-Fix the code according to the rules.
+Return ONLY the full corrected Python code.
 """
 
-    try:
-        response = client.chat.completions.create(
-            model="gpt-5",
-            messages=[
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt},
-            ],
-            max_completion_tokens=1500,
-        )
-
-        fixed_code = response.choices[0].message.content.strip()
+    response = client.chat.completions.create(
+        model="gpt-5",
+        messages=[
+            {"role": "system", "content": "You fix broken Python code."},
+            {"role": "user", "content": prompt},
+        ],
+        max_completion_tokens=1500,
+    )
 
-        # strip accidental markdown fences
-        if fixed_code.startswith("```"):
-            fixed_code = "\n".join(fixed_code.splitlines()[1:])
-        if fixed_code.endswith("```"):
-            fixed_code = "\n".join(fixed_code.splitlines()[:-1])
+    fixed_code = response.choices[0].message.content.strip()
 
-        return fixed_code.strip()
+    if fixed_code.startswith("```"):
+        fixed_code = "\n".join(fixed_code.split("\n")[1:])
+    if fixed_code.endswith("```"):
+        fixed_code = "\n".join(fixed_code.split("\n")[:-1])
 
-    except Exception as e:
-        print(f"Error calling OpenAI API: {e}")
-        return original_code
+    return fixed_code.strip()

auto_code_fixer/installer.py

@@ -1,25 +1,16 @@
-import os
 import re
-import sys
 import subprocess
+import sys
 
 
-def check_and_install_missing_lib(stderr, project_root):
+def check_and_install_missing_lib(stderr):
     match = re.search(r"No module named '([^']+)'", stderr)
     if not match:
         return False
 
-    missing = match.group(1)
-
-    # 🔒 CRITICAL: skip local project modules
-    local_file = os.path.join(project_root, f"{missing}.py")
-    if os.path.exists(local_file):
-        return False
-
+    lib = match.group(1)
     try:
-        subprocess.check_call(
-            [sys.executable, "-m", "pip", "install", missing]
-        )
+        subprocess.check_call([sys.executable, "-m", "pip", "install", lib])
         return True
     except subprocess.CalledProcessError:
         return False

auto_code_fixer/runner.py

@@ -1,29 +1,13 @@
 import subprocess
-import os
-import sys
 
 
-def run_code(script_path, project_root):
-    env = os.environ.copy()
-
-    script_dir = os.path.dirname(script_path)
-
-    # 🔑 CRITICAL: include BOTH project root and script dir
-    paths = [project_root, script_dir]
-
-    existing = env.get("PYTHONPATH", "")
-    if existing:
-        paths.append(existing)
-
-    env["PYTHONPATH"] = os.pathsep.join(paths)
-
+def run_code(path_to_code):
     try:
         result = subprocess.run(
-            [sys.executable, script_path],
+            ["python3", path_to_code],
             capture_output=True,
             text=True,
             timeout=15,
-            env=env,
         )
         return result.returncode, result.stdout, result.stderr
     except subprocess.TimeoutExpired:

{auto_code_fixer-0.2.5.dist-info → auto_code_fixer-0.2.6.dist-info}/METADATA

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

auto_code_fixer-0.2.6.dist-info/RECORD

@@ -0,0 +1,12 @@
+auto_code_fixer/__init__.py,sha256=Oz5HbwHMyE87nmwV80AZzpkJPf-wBg7eDuJr_BXZkhU,22
+auto_code_fixer/cli.py,sha256=xUVwCBTHxqxicAFqU_rbVigxP8fPL0L9m73NAUIWYy0,4147
+auto_code_fixer/fixer.py,sha256=aoPAzUmaMSSEO4kpJ6PMWOOpWhnd7W1ikfFr9fQsHrg,1308
+auto_code_fixer/installer.py,sha256=25rlboOI6SgJyVWDQ1NFGHukZkpIREsHa2abMYAqq7A,378
+auto_code_fixer/runner.py,sha256=n5GR03-ZUnti0L9do0_BTsislrFzARjuP9lm_iaWsKc,388
+auto_code_fixer/utils.py,sha256=ovWyNf3B7xMVq8qjylrPzNhrmLFaHKFF1G0vIp_GGSI,1425
+auto_code_fixer-0.2.6.dist-info/licenses/LICENSE,sha256=hgchJNa26tjXuLztwSUDbYQxNLnAPnLk6kDXNIkC8xc,1066
+auto_code_fixer-0.2.6.dist-info/METADATA,sha256=kGzKHtewk_p-y9gChVamngU9PScAIELAH4Edl5RNrHs,1522
+auto_code_fixer-0.2.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+auto_code_fixer-0.2.6.dist-info/entry_points.txt,sha256=a-j2rkfwkrhXZ5Qbz_6_gwk6Bj7nijYR1DALjWp5Myk,61
+auto_code_fixer-0.2.6.dist-info/top_level.txt,sha256=qUk1qznb6Qxqmxy2A3z_5dpOZlmNKHwUiLuJwH-CrAk,16
+auto_code_fixer-0.2.6.dist-info/RECORD,,

auto_code_fixer/guard.py

@@ -1,19 +0,0 @@
-def import_hijacked(original_code: str, fixed_code: str) -> bool:
-    """
-    Detects GPT cheating by replacing imports with try/except fallbacks
-    or inlining local module logic.
-    """
-
-    if "from " not in original_code:
-        return False
-
-    # common cheating patterns
-    bad_patterns = [
-        "try:",
-        "except ImportError",
-        "except ModuleNotFoundError",
-        "def greet(",
-        "def add(",
-    ]
-
-    return any(p in fixed_code for p in bad_patterns)

auto_code_fixer-0.2.5.dist-info/RECORD

@@ -1,13 +0,0 @@
-auto_code_fixer/__init__.py,sha256=S5Nlc-a2E4VkueiTguk-_xKzhUnSyTj5fGeoME1VKts,23
-auto_code_fixer/cli.py,sha256=cg86GiEuihOHnzqlP0ocEdmpgFtDigQ20pY7oazH-_c,5049
-auto_code_fixer/fixer.py,sha256=Lqb9IQiSvXIo168wv9cE3uyM3F_FVmu681PnnAmCFLI,1767
-auto_code_fixer/guard.py,sha256=Qfyqj2H1VeqBHFQxBkMy-L7Rz6ULXMV3uPAz7IMyTRs,489
-auto_code_fixer/installer.py,sha256=FYPD2r3zJ24MOPDlB_HWOTr2lTXKFLc73vMiQIChsCk,599
-auto_code_fixer/runner.py,sha256=EOQERj9lspNHpDd5QOvoB0XqbCRCed1vu1EAb6nJZsE,763
-auto_code_fixer/utils.py,sha256=ovWyNf3B7xMVq8qjylrPzNhrmLFaHKFF1G0vIp_GGSI,1425
-auto_code_fixer-0.2.5.dist-info/licenses/LICENSE,sha256=hgchJNa26tjXuLztwSUDbYQxNLnAPnLk6kDXNIkC8xc,1066
-auto_code_fixer-0.2.5.dist-info/METADATA,sha256=fJDgSRlKDTuBPzlNwpK3BIMwk1l7_uSwnTujsv3ex9c,1522
-auto_code_fixer-0.2.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-auto_code_fixer-0.2.5.dist-info/entry_points.txt,sha256=a-j2rkfwkrhXZ5Qbz_6_gwk6Bj7nijYR1DALjWp5Myk,61
-auto_code_fixer-0.2.5.dist-info/top_level.txt,sha256=qUk1qznb6Qxqmxy2A3z_5dpOZlmNKHwUiLuJwH-CrAk,16
-auto_code_fixer-0.2.5.dist-info/RECORD,,