codeshift 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

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/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/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/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

codeshift/knowledge_base/libraries/attrs.yaml

@@ -0,0 +1,181 @@
+# attrs Knowledge Base
+# Breaking changes from 21.x to 23.x+
+
+name: attrs
+display_name: attrs
+description: Python Classes Without Boilerplate - declarative classes with validation
+migration_guide_url: https://www.attrs.org/en/stable/changelog.html
+
+supported_migrations:
+  - from: "21.0"
+    to: "23.0"
+
+breaking_changes:
+  # Decorator changes - attr.s to attrs.define
+  - symbol: "@attr.s"
+    change_type: renamed
+    severity: high
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "@attr.s decorator replaced with @attrs.define (modern API)"
+    replacement: "@attrs.define"
+    has_deterministic_transform: true
+    transform_name: attr_s_to_attrs_define
+    notes: "The modern API has slots=True and auto_attribs=True as defaults"
+
+  - symbol: "@attr.attrs"
+    change_type: renamed
+    severity: high
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "@attr.attrs decorator replaced with @attrs.define"
+    replacement: "@attrs.define"
+    has_deterministic_transform: true
+    transform_name: attr_attrs_to_attrs_define
+
+  - symbol: "@attr.s(auto_attribs=True)"
+    change_type: renamed
+    severity: high
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "@attr.s with auto_attribs=True becomes @attrs.define (auto_attribs is default)"
+    replacement: "@attrs.define"
+    has_deterministic_transform: true
+    transform_name: attr_s_auto_attribs_to_define
+    notes: "auto_attribs=True is the default in @attrs.define"
+
+  - symbol: "@attr.s(slots=True)"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "@attr.s with slots=True becomes @attrs.define (slots is default)"
+    replacement: "@attrs.define"
+    has_deterministic_transform: true
+    transform_name: attr_s_slots_to_define
+    notes: "slots=True is the default in @attrs.define"
+
+  - symbol: "@attr.s(frozen=True)"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "@attr.s with frozen=True becomes @attrs.frozen"
+    replacement: "@attrs.frozen"
+    has_deterministic_transform: true
+    transform_name: attr_s_frozen_to_attrs_frozen
+
+  # Field/attribute definition changes
+  - symbol: "attr.ib()"
+    change_type: renamed
+    severity: high
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.ib() replaced with attrs.field()"
+    replacement: "attrs.field()"
+    has_deterministic_transform: true
+    transform_name: attr_ib_to_attrs_field
+
+  - symbol: "attr.attrib()"
+    change_type: renamed
+    severity: high
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.attrib() replaced with attrs.field()"
+    replacement: "attrs.field()"
+    has_deterministic_transform: true
+    transform_name: attr_attrib_to_attrs_field
+
+  - symbol: "attr.Factory"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.Factory replaced with attrs.Factory"
+    replacement: "attrs.Factory"
+    has_deterministic_transform: true
+    transform_name: attr_factory_to_attrs_factory
+
+  # Function changes
+  - symbol: "attr.asdict()"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.asdict() replaced with attrs.asdict()"
+    replacement: "attrs.asdict()"
+    has_deterministic_transform: true
+    transform_name: attr_asdict_to_attrs_asdict
+
+  - symbol: "attr.astuple()"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.astuple() replaced with attrs.astuple()"
+    replacement: "attrs.astuple()"
+    has_deterministic_transform: true
+    transform_name: attr_astuple_to_attrs_astuple
+
+  - symbol: "attr.fields()"
+    change_type: renamed
+    severity: low
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.fields() replaced with attrs.fields()"
+    replacement: "attrs.fields()"
+    has_deterministic_transform: true
+    transform_name: attr_fields_to_attrs_fields
+
+  - symbol: "attr.has()"
+    change_type: renamed
+    severity: low
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.has() replaced with attrs.has()"
+    replacement: "attrs.has()"
+    has_deterministic_transform: true
+    transform_name: attr_has_to_attrs_has
+
+  - symbol: "attr.evolve()"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.evolve() replaced with attrs.evolve()"
+    replacement: "attrs.evolve()"
+    has_deterministic_transform: true
+    transform_name: attr_evolve_to_attrs_evolve
+
+  # Parameter deprecations
+  - symbol: "cmp"
+    change_type: removed
+    severity: high
+    from_version: "21.0"
+    to_version: "22.1"
+    description: "cmp parameter removed from @attr.s and attr.ib(), use eq and order instead"
+    replacement: "eq=..., order=..."
+    has_deterministic_transform: true
+    transform_name: cmp_to_eq_order
+    notes: "cmp=True becomes eq=True, order=True; cmp=False becomes eq=False, order=False"
+
+  # Import path changes for validators and converters
+  - symbol: "attr.validators"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.validators module replaced with attrs.validators"
+    replacement: "attrs.validators"
+    has_deterministic_transform: true
+    transform_name: attr_validators_to_attrs_validators
+
+  - symbol: "attr.converters"
+    change_type: renamed
+    severity: medium
+    from_version: "21.0"
+    to_version: "23.0"
+    description: "attr.converters module replaced with attrs.converters"
+    replacement: "attrs.converters"
+    has_deterministic_transform: true
+    transform_name: attr_converters_to_attrs_converters