llm-code-validator 0.1.0__py3-none-any.whl

llm_code_validator/fixes.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+from .core import check_file
+from .diagnostics import Diagnostic
+from .versioning import VersionContext
+
+
+@dataclass(frozen=True)
+class FixResult:
+    path: str
+    changed: bool
+    previews: list[str]
+    skipped: list[str]
+
+
+def _line_replacement(line: str, diagnostic: Diagnostic) -> str | None:
+    if diagnostic.fix.safety != "safe_fix" or not diagnostic.replacement:
+        return None
+    if line.lstrip().startswith("from ") and diagnostic.replacement.startswith("from "):
+        indent = line[: len(line) - len(line.lstrip())]
+        return indent + diagnostic.replacement
+    if diagnostic.symbol in line:
+        return line.replace(diagnostic.symbol, diagnostic.replacement, 1)
+    old_token = diagnostic.symbol.split(".")[-1]
+    if old_token not in line:
+        return None
+    return line.replace(old_token, diagnostic.replacement, 1)
+
+
+def fix_file(path: str | Path, write: bool = False, version_context: VersionContext | None = None) -> FixResult:
+    file_path = Path(path)
+    source = file_path.read_text(encoding="utf-8")
+    keep_final_newline = source.endswith("\n")
+    lines = source.splitlines()
+    result = check_file(file_path, version_context)
+    previews: list[str] = []
+    skipped: list[str] = []
+    changed = False
+
+    for diagnostic in result.diagnostics:
+        if diagnostic.line < 1 or diagnostic.line > len(lines):
+            skipped.append(f"{diagnostic.path}:{diagnostic.line} {diagnostic.code} cannot locate line")
+            continue
+        old_line = lines[diagnostic.line - 1]
+        new_line = _line_replacement(old_line, diagnostic)
+        if new_line is None:
+            skipped.append(
+                f"{diagnostic.path}:{diagnostic.line} {diagnostic.code} skipped "
+                f"({diagnostic.fix.safety})"
+            )
+            continue
+        previews.append(f"{diagnostic.path}:{diagnostic.line}\n  old: {old_line}\n  new: {new_line}")
+        if write:
+            lines[diagnostic.line - 1] = new_line
+            changed = True
+
+    if write and changed:
+        updated = "\n".join(lines)
+        if keep_final_newline:
+            updated += "\n"
+        file_path.write_text(updated, encoding="utf-8")
+
+    return FixResult(str(file_path), changed, previews, skipped)

llm_code_validator/formatting.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+import json
+
+from .diagnostics import CheckResult
+
+
+def format_text(result: CheckResult) -> str:
+    lines: list[str] = []
+    for diagnostic in result.diagnostics:
+        qualified_symbol = (
+            diagnostic.symbol
+            if diagnostic.symbol.startswith(f"{diagnostic.library}.")
+            else f"{diagnostic.library}.{diagnostic.symbol}"
+        )
+        lines.append(
+            f"{diagnostic.path}:{diagnostic.line} {diagnostic.code} "
+            f"{diagnostic.severity} {qualified_symbol} {diagnostic.message}"
+        )
+        if diagnostic.version_assumption:
+            lines.append(f"  version: {diagnostic.version_assumption}")
+        if diagnostic.replacement:
+            lines.append(f"  fix: {diagnostic.replacement}")
+    if not result.diagnostics:
+        lines.append(f"OK: checked {result.checked_files} file(s), no diagnostics")
+    for warning in result.warnings:
+        lines.append(f"warning: {warning}")
+    return "\n".join(lines)
+
+
+def format_json(result: CheckResult) -> str:
+    return json.dumps(result.to_dict(), indent=2, sort_keys=True)
+
+
+def format_github(result: CheckResult) -> str:
+    lines: list[str] = []
+    for diagnostic in result.diagnostics:
+        message = diagnostic.message.replace("\n", " ")
+        lines.append(
+            f"::{diagnostic.severity} file={diagnostic.path},line={diagnostic.line},"
+            f"col={diagnostic.column},title={diagnostic.code}::{message}"
+        )
+    return "\n".join(lines)