feat: [CG-10557] Codegen update command

pyproject.toml

@@ -62,6 +62,7 @@ dependencies = [
   "tomlkit>=0.13.2",
   "python-semantic-release",
   "uvicorn[standard]>=0.30.0",
+  "packaging>=24.2",
 ]
 
 license = { text = "Apache-2.0" }

src/codegen/cli/cli.py

@@ -14,6 +14,7 @@
 from codegen.cli.commands.run.main import run_command
 from codegen.cli.commands.run_on_pr.main import run_on_pr_command
 from codegen.cli.commands.style_debug.main import style_debug_command
+from codegen.cli.commands.update.main import update_command
 
 click.rich_click.USE_RICH_MARKUP = True
 install(show_locals=True)
@@ -39,6 +40,7 @@ def main():
 main.add_command(run_on_pr_command)
 main.add_command(notebook_command)
 main.add_command(reset_command)
+main.add_command(update_command)
 
 
 if __name__ == "__main__":

src/codegen/cli/commands/update/main.py

@@ -0,0 +1,60 @@
+import subprocess
+import sys
+from importlib.metadata import distribution
+
+import requests
+import rich
+import rich_click as click
+from packaging.version import Version
+
+import codegen
+
+
+def fetch_pypi_releases(package: str) -> list[str]:
+    response = requests.get(f"https://pypi.org/pypi/{package}/json")
+    response.raise_for_status()
+    return response.json()["releases"].keys()
+
+
+def filter_versions(versions: list[Version], current_version: Version, num_prev_minor_version: int = 1) -> list[Version]:
+    descending_minor_versions = [v_tuple for v_tuple in sorted(set(v.release[:2] for v in versions), reverse=True) if v_tuple < current_version.release[:2]]
+    try:
+        compare_tuple = descending_minor_versions[:num_prev_minor_version][-1] + (0,)
+    except IndexError:
+        compare_tuple = (current_version.major, current_version.minor, 0)
+
+    return [v for v in versions if (v.major, v.minor, v.micro) >= compare_tuple]  # v.release will only show major,minor if micro doesn't exist.
+
+
+def install_package(package: str, *args: str) -> None:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package, *args])
+
+
+@click.command(name="update")
+@click.option("--list", "-l", "list_", is_flag=True, help="List all supported versions of the codegen")
+@click.option("--version", "-v", type=str, help="Update to a specific version of the codegen")
+def update_command(list_: bool = False, version: str | None = None):
+    """Update Codegen to the latest or specified version
+
+    --list: List all supported versions of the codegen
+    --version: Update to a specific version of the codegen
+    """
+    if list_ and version:
+        msg = "Cannot specify both --list and --version"
+        raise click.ClickException(msg)
+
+    package_info = distribution(codegen.__package__)
+    current_version = Version(package_info.version)
+
+    if list_:
+        releases = fetch_pypi_releases(package_info.name)
+        filtered_releases = filter_versions([Version(r) for r in releases], current_version, num_prev_minor_version=2)
+        for release in filtered_releases:
+            if release.release == current_version.release:
+                release = f"[bold]{release}[/bold] (current)"
+            rich.print(release)
+    elif version:
+        install_package(f"{package_info.name}=={version}")
+    else:
+        # Update to latest version
+        install_package(package_info.name, "--upgrade")