codeshift 0.3.2__tar.gz → 0.3.4__tar.gz

{codeshift-0.3.2 → codeshift-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeshift
-Version: 0.3.2
+Version: 0.3.4
 Summary: AI-powered CLI tool that migrates Python code to handle breaking dependency changes
 Author: PyResolve Team
 License: MIT
@@ -84,6 +84,16 @@ PyResolve is an AI-powered CLI tool that migrates Python code to handle breaking
 | SQLAlchemy | 1.4 → 2.0 | ✅ Supported |
 | Pandas | 1.x → 2.x | ✅ Supported |
 | Requests | Various | ✅ Supported |
+| Django | 3.x → 4.x/5.x | ✅ Supported |
+| Flask | 1.x → 2.x/3.x | ✅ Supported |
+| NumPy | 1.x → 2.x | ✅ Supported |
+| attrs | attr → attrs | ✅ Supported |
+| Celery | 4.x → 5.x | ✅ Supported |
+| Click | 7.x → 8.x | ✅ Supported |
+| aiohttp | 2.x → 3.x | ✅ Supported |
+| httpx | 0.x → 0.24+ | ✅ Supported |
+| Marshmallow | 2.x → 3.x | ✅ Supported |
+| pytest | 6.x → 7.x/8.x | ✅ Supported |
 
 ### Any Library (Auto-Generated Knowledge Base)
 
@@ -104,7 +114,10 @@ pyresolve scan
 # Scan with detailed breaking change analysis
 pyresolve scan --fetch-changes
 
-# Analyze and propose changes for a specific library
+# Upgrade all outdated packages at once
+pyresolve upgrade-all
+
+# Or analyze and propose changes for a specific library
 pyresolve upgrade pydantic --target 2.5.0
 
 # View detailed diff of proposed changes
@@ -231,6 +244,35 @@ pyresolve apply
 #   --file, -f      Apply changes to specific file only
 ```
 
+### Upgrade-All Command
+
+Upgrade all outdated packages to their latest versions in one go:
+
+```bash
+pyresolve upgrade-all
+
+# Options:
+#   --all, -a       Include all outdated packages (not just Tier 1)
+#   --path, -p      Path to analyze (default: current directory)
+#   --dry-run       Show what would be changed without saving state
+```
+
+### Libraries Command
+
+List all supported libraries and their migration paths:
+
+```bash
+pyresolve libraries
+```
+
+### Status Command
+
+Show current migration status, pending changes, and quota information:
+
+```bash
+pyresolve status
+```
+
 ## How It Works
 
 ```text
@@ -261,7 +303,7 @@ pyresolve apply
 2. **Parse Changes**: Uses LLM to extract breaking changes with confidence levels (HIGH/MEDIUM/LOW)
 3. **Scan Codebase**: Finds imports and usage of the target library
 4. **Tiered Migration**:
-   - **Tier 1**: Deterministic AST transforms for known libraries (pydantic, fastapi, sqlalchemy, pandas, requests)
+   - **Tier 1**: Deterministic AST transforms for 15 supported libraries (Pydantic, FastAPI, SQLAlchemy, Django, Flask, NumPy, Pandas, Requests, attrs, Celery, Click, aiohttp, httpx, Marshmallow, pytest)
    - **Tier 2**: Knowledge base guided migration with LLM assistance
    - **Tier 3**: Pure LLM migration for unknown patterns
 5. **Validate**: Runs syntax checks and optionally your test suite
@@ -303,7 +345,7 @@ PyResolve uses a tiered pricing model:
 
 | Tier | Price | Features |
 |------|-------|----------|
-| **Free** | $0/month | Tier 1 deterministic transforms (Pydantic, FastAPI, SQLAlchemy, Pandas, Requests) |
+| **Free** | $0/month | Tier 1 deterministic transforms (15 libraries including Pydantic, Django, Flask, SQLAlchemy, and more) |
 | **Pro** | $19/month | Tier 2 KB-guided LLM migrations for any library |
 | **Unlimited** | $49/month | Tier 3 pure LLM migrations + priority support |
 

{codeshift-0.3.2 → codeshift-0.3.4}/README.md

@@ -28,6 +28,16 @@ PyResolve is an AI-powered CLI tool that migrates Python code to handle breaking
 | SQLAlchemy | 1.4 → 2.0 | ✅ Supported |
 | Pandas | 1.x → 2.x | ✅ Supported |
 | Requests | Various | ✅ Supported |
+| Django | 3.x → 4.x/5.x | ✅ Supported |
+| Flask | 1.x → 2.x/3.x | ✅ Supported |
+| NumPy | 1.x → 2.x | ✅ Supported |
+| attrs | attr → attrs | ✅ Supported |
+| Celery | 4.x → 5.x | ✅ Supported |
+| Click | 7.x → 8.x | ✅ Supported |
+| aiohttp | 2.x → 3.x | ✅ Supported |
+| httpx | 0.x → 0.24+ | ✅ Supported |
+| Marshmallow | 2.x → 3.x | ✅ Supported |
+| pytest | 6.x → 7.x/8.x | ✅ Supported |
 
 ### Any Library (Auto-Generated Knowledge Base)
 
@@ -48,7 +58,10 @@ pyresolve scan
 # Scan with detailed breaking change analysis
 pyresolve scan --fetch-changes
 
-# Analyze and propose changes for a specific library
+# Upgrade all outdated packages at once
+pyresolve upgrade-all
+
+# Or analyze and propose changes for a specific library
 pyresolve upgrade pydantic --target 2.5.0
 
 # View detailed diff of proposed changes
@@ -175,6 +188,35 @@ pyresolve apply
 #   --file, -f      Apply changes to specific file only
 ```
 
+### Upgrade-All Command
+
+Upgrade all outdated packages to their latest versions in one go:
+
+```bash
+pyresolve upgrade-all
+
+# Options:
+#   --all, -a       Include all outdated packages (not just Tier 1)
+#   --path, -p      Path to analyze (default: current directory)
+#   --dry-run       Show what would be changed without saving state
+```
+
+### Libraries Command
+
+List all supported libraries and their migration paths:
+
+```bash
+pyresolve libraries
+```
+
+### Status Command
+
+Show current migration status, pending changes, and quota information:
+
+```bash
+pyresolve status
+```
+
 ## How It Works
 
 ```text
@@ -205,7 +247,7 @@ pyresolve apply
 2. **Parse Changes**: Uses LLM to extract breaking changes with confidence levels (HIGH/MEDIUM/LOW)
 3. **Scan Codebase**: Finds imports and usage of the target library
 4. **Tiered Migration**:
-   - **Tier 1**: Deterministic AST transforms for known libraries (pydantic, fastapi, sqlalchemy, pandas, requests)
+   - **Tier 1**: Deterministic AST transforms for 15 supported libraries (Pydantic, FastAPI, SQLAlchemy, Django, Flask, NumPy, Pandas, Requests, attrs, Celery, Click, aiohttp, httpx, Marshmallow, pytest)
    - **Tier 2**: Knowledge base guided migration with LLM assistance
    - **Tier 3**: Pure LLM migration for unknown patterns
 5. **Validate**: Runs syntax checks and optionally your test suite
@@ -247,7 +289,7 @@ PyResolve uses a tiered pricing model:
 
 | Tier | Price | Features |
 |------|-------|----------|
-| **Free** | $0/month | Tier 1 deterministic transforms (Pydantic, FastAPI, SQLAlchemy, Pandas, Requests) |
+| **Free** | $0/month | Tier 1 deterministic transforms (15 libraries including Pydantic, Django, Flask, SQLAlchemy, and more) |
 | **Pro** | $19/month | Tier 2 KB-guided LLM migrations for any library |
 | **Unlimited** | $49/month | Tier 3 pure LLM migrations + priority support |
 

{codeshift-0.3.2 → codeshift-0.3.4}/codeshift/cli/commands/apply.py

@@ -10,6 +10,7 @@ from rich.panel import Panel
 from rich.prompt import Confirm
 
 from codeshift.cli.commands.upgrade import load_state, save_state
+from codeshift.cli.package_manager import get_install_commands, get_sync_command
 from codeshift.cli.quota import (
     QuotaError,
     check_quota,
@@ -226,11 +227,32 @@ def apply(
             state_file.unlink()
         console.print("\n[green]Migration complete![/]")
 
-    # Next steps
+    # Next steps - generate dynamic dependency sync command
     console.print("\n[bold]Recommended next steps:[/]")
     console.print("  1. Review the changes in your editor")
     console.print("  2. Run your test suite: [cyan]pytest[/]")
-    console.print("  3. Update your dependencies: [cyan]pip install pydantic>=2.0[/]")
+
+    # Generate dependency update commands based on library/libraries and package manager
+    sync_command = get_sync_command(project_path)
+
+    # Check if this is a multi-library migration (upgrade-all)
+    migrations = state.get("migrations", [])
+    if library == "multiple" and migrations:
+        # Multi-library case: show sync command and list all libraries
+        console.print(f"  3. Sync your dependencies: [cyan]{sync_command}[/]")
+        console.print("\n     [dim]Upgraded libraries:[/]")
+        for migration in migrations:
+            lib_name = migration.get("library", "unknown")
+            to_version = migration.get("to_version", "unknown")
+            console.print(f"       • {lib_name} → {to_version}")
+    else:
+        # Single library case: show sync command with specific library info
+        console.print(f"  3. Sync your dependencies: [cyan]{sync_command}[/]")
+        if library != "unknown" and target_version != "unknown":
+            libraries = [{"name": library, "version": target_version}]
+            install_commands = get_install_commands(project_path, libraries)
+            if install_commands:
+                console.print(f"\n     [dim]Or install directly:[/] {install_commands[0]}")
 
 
 @click.command()

{codeshift-0.3.2 → codeshift-0.3.4}/codeshift/cli/commands/upgrade_all.py

@@ -257,6 +257,7 @@ def upgrade_all(
         BarColumn(),
         TaskProgressColumn(),
         console=console,
+        transient=True,
     ) as progress:
         task = progress.add_task("Checking for updates...", total=len(dependencies))
 
@@ -347,9 +348,10 @@ def upgrade_all(
         table.add_row(str(pkg["name"]), str(pkg["current"]), str(pkg["latest"]), type_str, tier_str)
 
     console.print(table)
+    console.print()  # Ensure table is fully rendered before progress bar
 
     # Run upgrades for each package
-    console.print("\n[bold]Running migrations...[/]\n")
+    console.print("[bold]Running migrations...[/]\n")
 
     all_results: dict[str, list[dict]] = {}
     migration_summary: list[dict] = []
@@ -360,6 +362,7 @@ def upgrade_all(
         BarColumn(),
         TaskProgressColumn(),
         console=console,
+        transient=True,
     ) as progress:
         task = progress.add_task("Upgrading packages...", total=len(upgradeable))
 

codeshift-0.3.4/codeshift/cli/package_manager.py

@@ -0,0 +1,102 @@
+"""Package manager detection and sync command generation utilities."""
+
+import shutil
+from pathlib import Path
+from typing import Literal
+
+PackageManager = Literal["uv", "poetry", "pip"]
+
+
+def detect_package_manager(project_path: Path) -> PackageManager:
+    """Detect the package manager used in a project.
+
+    Detection order:
+    1. uv.lock file presence → uv
+    2. poetry.lock file presence → poetry
+    3. Default → pip
+
+    Args:
+        project_path: Path to the project root directory.
+
+    Returns:
+        The detected package manager name.
+    """
+    # Check for uv (uv.lock file)
+    if (project_path / "uv.lock").exists():
+        return "uv"
+
+    # Check for poetry (poetry.lock file)
+    if (project_path / "poetry.lock").exists():
+        return "poetry"
+
+    # Default to pip
+    return "pip"
+
+
+def get_sync_command(project_path: Path) -> str:
+    """Get the appropriate dependency sync command for the project.
+
+    Args:
+        project_path: Path to the project root directory.
+
+    Returns:
+        The shell command to sync/install dependencies.
+    """
+    manager = detect_package_manager(project_path)
+
+    if manager == "uv":
+        return "uv sync"
+    elif manager == "poetry":
+        return "poetry install"
+    else:
+        # pip - check what kind of project it is
+        if (project_path / "pyproject.toml").exists():
+            return "pip install -e ."
+        elif (project_path / "requirements.txt").exists():
+            return "pip install -r requirements.txt"
+        else:
+            return "pip install -e ."
+
+
+def is_package_manager_available(manager: PackageManager) -> bool:
+    """Check if a package manager is available in the system.
+
+    Args:
+        manager: The package manager to check.
+
+    Returns:
+        True if the package manager is available, False otherwise.
+    """
+    return shutil.which(manager) is not None
+
+
+def get_install_commands(
+    project_path: Path,
+    libraries: list[dict[str, str]],
+) -> list[str]:
+    """Generate install commands for the upgraded libraries.
+
+    This is a fallback for when users want to install specific packages
+    rather than sync the entire project.
+
+    Args:
+        project_path: Path to the project root directory.
+        libraries: List of dicts with 'name' and 'version' keys.
+
+    Returns:
+        List of install commands for each library.
+    """
+    manager = detect_package_manager(project_path)
+
+    commands = []
+    for lib in libraries:
+        name = lib["name"]
+        version = lib["version"]
+        if manager == "uv":
+            commands.append(f"uv add {name}>={version}")
+        elif manager == "poetry":
+            commands.append(f"poetry add {name}>={version}")
+        else:
+            commands.append(f"pip install {name}>={version}")
+
+    return commands

{codeshift-0.3.2 → codeshift-0.3.4}/codeshift/knowledge/generator.py

@@ -162,7 +162,17 @@ class KnowledgeGenerator:
 
 
 # Tier 1 libraries with deterministic AST transforms
-TIER_1_LIBRARIES = {"pydantic", "fastapi", "sqlalchemy", "pandas", "requests"}
+TIER_1_LIBRARIES = {
+    "pydantic",
+    "fastapi",
+    "sqlalchemy",
+    "pandas",
+    "requests",
+    "numpy",
+    "pytest",
+    "marshmallow",
+    "flask",
+}
 
 
 def is_tier_1_library(library: str) -> bool:

codeshift-0.3.4/codeshift/knowledge_base/libraries/aiohttp.yaml

@@ -0,0 +1,186 @@
+# aiohttp Knowledge Base
+# Breaking changes from v3.7 to v3.9+
+
+name: aiohttp
+display_name: aiohttp
+description: Async HTTP client/server framework for asyncio
+migration_guide_url: https://docs.aiohttp.org/en/stable/changes.html
+
+supported_migrations:
+  - from: "3.7"
+    to: "3.9"
+
+breaking_changes:
+  # Loop parameter removal - ClientSession
+  - symbol: "ClientSession(loop=...)"
+    change_type: removed
+    severity: high
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "loop parameter removed from ClientSession constructor"
+    replacement: "ClientSession()"
+    has_deterministic_transform: true
+    transform_name: remove_loop_param_client_session
+    notes: "Event loop is now obtained automatically via asyncio.get_event_loop()"
+
+  # Loop parameter removal - TCPConnector
+  - symbol: "TCPConnector(loop=...)"
+    change_type: removed
+    severity: high
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "loop parameter removed from TCPConnector constructor"
+    replacement: "TCPConnector()"
+    has_deterministic_transform: true
+    transform_name: remove_loop_param_tcp_connector
+
+  # Loop parameter removal - UnixConnector
+  - symbol: "UnixConnector(loop=...)"
+    change_type: removed
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "loop parameter removed from UnixConnector constructor"
+    replacement: "UnixConnector()"
+    has_deterministic_transform: true
+    transform_name: remove_loop_param_unix_connector
+
+  # Loop parameter removal - web.Application
+  - symbol: "web.Application(loop=...)"
+    change_type: removed
+    severity: high
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "loop parameter removed from web.Application constructor"
+    replacement: "web.Application()"
+    has_deterministic_transform: true
+    transform_name: remove_loop_param_web_application
+
+  # Loop parameter removal - ClientTimeout
+  - symbol: "ClientTimeout(loop=...)"
+    change_type: removed
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "loop parameter removed from ClientTimeout constructor"
+    replacement: "ClientTimeout()"
+    has_deterministic_transform: true
+    transform_name: remove_loop_param_client_timeout
+
+  # connector_owner default change
+  - symbol: "ClientSession(connector=..., connector_owner=None)"
+    change_type: behavior_change
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "connector_owner now defaults to True; explicitly set to False if connector is shared"
+    replacement: "ClientSession(connector=..., connector_owner=False)"
+    has_deterministic_transform: false
+    notes: "Manual review needed - depends on whether connector is shared across sessions"
+
+  # BasicAuth.encode() removal
+  - symbol: "BasicAuth.encode()"
+    change_type: removed
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "BasicAuth.encode() method removed"
+    replacement: "str(BasicAuth(...))"
+    has_deterministic_transform: true
+    transform_name: basicauth_encode_to_str
+
+  # Deprecated timeout parameters
+  - symbol: "ClientSession(read_timeout=...)"
+    change_type: removed
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "read_timeout parameter deprecated, use ClientTimeout instead"
+    replacement: "ClientSession(timeout=ClientTimeout(total=...))"
+    has_deterministic_transform: true
+    transform_name: read_timeout_to_client_timeout
+
+  - symbol: "ClientSession(conn_timeout=...)"
+    change_type: removed
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "conn_timeout parameter deprecated, use ClientTimeout instead"
+    replacement: "ClientSession(timeout=ClientTimeout(connect=...))"
+    has_deterministic_transform: true
+    transform_name: conn_timeout_to_client_timeout
+
+  # app.loop, request.loop, client.loop deprecation
+  - symbol: "app.loop"
+    change_type: deprecated
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "app.loop property deprecated"
+    replacement: "asyncio.get_event_loop()"
+    has_deterministic_transform: true
+    transform_name: app_loop_to_get_event_loop
+
+  - symbol: "request.app.loop"
+    change_type: deprecated
+    severity: medium
+    from_version: "3.7"
+    to_version: "3.8"
+    description: "request.app.loop property deprecated"
+    replacement: "asyncio.get_event_loop()"
+    has_deterministic_transform: true
+    transform_name: request_loop_to_get_event_loop
+
+  # Middleware signature changes (old-style to new-style)
+  - symbol: "@middleware"
+    change_type: signature_change
+    severity: high
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "Old-style middleware decorator deprecated; use @web.middleware with async def(request, handler)"
+    replacement: "@web.middleware async def middleware(request, handler)"
+    has_deterministic_transform: false
+    notes: "Middleware function signature changed; manual review recommended"
+
+  # ws_connect timeout deprecation
+  - symbol: "ws_connect(timeout=...)"
+    change_type: deprecated
+    severity: low
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "timeout parameter in ws_connect deprecated"
+    replacement: "ws_connect(receive_timeout=...)"
+    has_deterministic_transform: true
+    transform_name: ws_connect_timeout_rename
+
+  - symbol: "ws_connect(receive_timeout=...)"
+    change_type: behavior_change
+    severity: low
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "Default websocket receive_timeout changed from None to 10.0 seconds"
+    replacement: "ws_connect(receive_timeout=None)"
+    has_deterministic_transform: false
+    notes: "Set receive_timeout=None explicitly if you want no timeout"
+
+  # Response URL attribute change
+  - symbol: "response.url_obj"
+    change_type: renamed
+    severity: low
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "url_obj attribute renamed to url (now returns yarl.URL)"
+    replacement: "response.url"
+    has_deterministic_transform: true
+    transform_name: url_obj_to_url
+
+  # WebSocket protocol attribute rename
+  - symbol: "ws_response.protocol"
+    change_type: renamed
+    severity: low
+    from_version: "3.7"
+    to_version: "3.9"
+    description: "WebSocketResponse.protocol renamed to ws_protocol"
+    replacement: "ws_response.ws_protocol"
+    has_deterministic_transform: true
+    transform_name: ws_protocol_rename