code-towel 1.0.3__tar.gz → 1.0.4__tar.gz

{code_towel-1.0.3/src/code_towel.egg-info → code_towel-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-towel
-Version: 1.0.3
+Version: 1.0.4
 Summary: A Python tool that DRYs your code - finds and refactors repeated code using unification-based analysis
 Author: Eric Allen
 License:                                  Apache License

{code_towel-1.0.3 → code_towel-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "code-towel"
-version = "1.0.3"
+version = "1.0.4"
 description = "A Python tool that DRYs your code - finds and refactors repeated code using unification-based analysis"
 authors = [
     {name = "Eric Allen"}

{code_towel-1.0.3 → code_towel-1.0.4/src/code_towel.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-towel
-Version: 1.0.3
+Version: 1.0.4
 Summary: A Python tool that DRYs your code - finds and refactors repeated code using unification-based analysis
 Author: Eric Allen
 License:                                  Apache License

{code_towel-1.0.3 → code_towel-1.0.4}/src/towel/cli.py

@@ -700,7 +700,9 @@ def _run_interactive_llm_mode(
     print('  {"__extracted_func_1": "calculate_total"}')
     print('  {"src/utils.py:__extracted_func_2": "validate_input"}')
     print()
-    print("Paste the JSON response below, then press ENTER followed by Ctrl+D (or Ctrl+Z on Windows):")
+    print(
+        "Paste the JSON response below, then press ENTER followed by Ctrl+D (or Ctrl+Z on Windows):"
+    )
     print()
 
     # Read LLM response from stdin

{code_towel-1.0.3 → code_towel-1.0.4}/src/towel/unification/refactor_engine.py

@@ -538,6 +538,7 @@ class UnificationRefactorEngine:
         docstring_char: Optional[str] = None
         last_import_line = 0
         after_docstring = 0
+        in_multiline_import = False
 
         for i, line in enumerate(lines):
             stripped = line.strip()
@@ -557,8 +558,19 @@ class UnificationRefactorEngine:
                     after_docstring = i + 1
                 continue
 
+            # Check for start of multi-line import (has opening paren but no closing paren)
             if stripped.startswith("import ") or stripped.startswith("from "):
                 last_import_line = i + 1
+                # Check if this is a multi-line import
+                if "(" in line and ")" not in line:
+                    in_multiline_import = True
+                continue
+
+            # Inside a multi-line import - continue until we see closing paren
+            if in_multiline_import:
+                last_import_line = i + 1
+                if ")" in line:
+                    in_multiline_import = False
                 continue
 
             if last_import_line > 0 and stripped and not stripped.startswith("#"):