fastapi-spawn 0.4.38__tar.gz → 0.4.39__tar.gz

{fastapi_spawn-0.4.38 → fastapi_spawn-0.4.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-spawn
-Version: 0.4.38
+Version: 0.4.39
 Summary: A powerful CLI tool to scaffold production-ready FastAPI projects with flexible database, auth, broker, and deployment options.
 Project-URL: Homepage, https://github.com/Bishwajitgarai/fastapi-spawn
 Project-URL: Documentation, https://github.com/Bishwajitgarai/fastapi-spawn#readme

{fastapi_spawn-0.4.38 → fastapi_spawn-0.4.39}/fastapi_spawn/cli.py

@@ -523,7 +523,6 @@ def add_feature(
     project_dir: Path = typer.Option(Path("."), "--dir", "-d", help="Path to the existing project"),
 ) -> None:
     _print_banner()
-    
     if not feature:
         console.print("\n[bold cyan]Available features to add:[/bold cyan]\n")
         table = Table(box=None)
@@ -548,6 +547,58 @@ def add_feature(
         console.print(f"[bold red]✗ Directory not found:[/bold red] {project_dir.resolve()}")
         raise typer.Exit(1)
 
+    if feature == "alembic":
+        console.print("[bold cyan]→ Automatically adding Alembic...[/bold cyan]")
+        
+        # 1. Prompt for DB type
+        from fastapi_spawn.constants import Database
+        import questionary
+        
+        db_type = questionary.select(
+            "Which database are you using?",
+            choices=[Database.postgresql.value, Database.mysql.value, Database.sqlite.value],
+            default=Database.postgresql.value
+        ).ask()
+        
+        if not db_type:
+            console.print("[bold red]✗ Database selection cancelled[/bold red]")
+            raise typer.Exit(1)
+            
+        # 2. Run uv add alembic
+        import subprocess
+        try:
+            console.print("[dim]Running: uv add alembic[/dim]")
+            subprocess.run(["uv", "add", "alembic"], cwd=project_dir, check=True)
+        except subprocess.CalledProcessError:
+            console.print("[bold red]✗ Failed to add alembic dependency[/bold red]")
+            raise typer.Exit(1)
+            
+        # 3. Run alembic init
+        try:
+            console.print("[dim]Running: uv run alembic init migrations[/dim]")
+            subprocess.run(["uv", "run", "alembic", "init", "migrations"], cwd=project_dir, check=True)
+        except subprocess.CalledProcessError:
+            console.print("[bold red]✗ Failed to initialize alembic[/bold red]")
+            raise typer.Exit(1)
+            
+        # 4. Render env.py
+        console.print("[dim]Rendering async env.py...[/dim]")
+        from fastapi_spawn.generator import ProjectGenerator
+        from fastapi_spawn.config import ProjectConfig
+        
+        cfg = ProjectConfig(project_name=project_dir.name, db=Database(db_type), orm=ORM.sqlalchemy)
+        cfg.has_alembic = True
+        generator = ProjectGenerator(cfg, project_dir)
+        
+        generator._render_to(project_dir / "migrations" / "env.py", "alembic/env.py.j2")
+        
+        console.print("[bold green]✓ Alembic added successfully![/bold green]")
+        console.print("\n[bold yellow]Next Steps:[/bold yellow]")
+        console.print("  1. Update [bold]alembic.ini[/bold] with your database URL.")
+        console.print("  2. Run [bold]uv run alembic revision --autogenerate -m 'initial'[/bold]")
+        console.print("  3. Run [bold]uv run alembic upgrade head[/bold]\n")
+        raise typer.Exit(0)
+
     console.print(f"[bold cyan]→ Adding feature:[/bold cyan] [bold]{feature}[/bold] — {_ADDABLE_FEATURES[feature]}")
     console.print(f"[dim]Target project:[/dim] {project_dir.resolve()}\n")
     _feature_guidance(feature, project_dir)

{fastapi_spawn-0.4.38 → fastapi_spawn-0.4.39}/fastapi_spawn/generator.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 import shutil
 import tempfile
 from pathlib import Path
+from dataclasses import asdict
+import json
 
 from jinja2 import Environment, PackageLoader, StrictUndefined, TemplateNotFound
 from rich.console import Console
@@ -132,6 +134,20 @@ class ProjectGenerator:
         self._render_to(root / ".gitignore", "base/gitignore.j2")
         self._render_to(root / ".pre-commit-config.yaml", "base/pre_commit.j2")
         self._render_to(root / "README.md", "base/README.md.j2")
+        
+        # Save tracking log for future add commands
+        def _enum_to_str(obj):
+            if isinstance(obj, dict):
+                return {k: _enum_to_str(v) for k, v in obj.items()}
+            elif isinstance(obj, list):
+                return [_enum_to_str(v) for v in obj]
+            elif hasattr(obj, "value") and not isinstance(obj, str):  # Enum
+                return obj.value
+            return obj
+            
+        config_dict = _enum_to_str(asdict(self.config))
+        config_path = root / ".fastapi-spawn.json"
+        config_path.write_text(json.dumps(config_dict, indent=2), encoding="utf-8")
 
     def _generate_app(self, root: Path) -> None:
         pkg = root / "app"

{fastapi_spawn-0.4.38 → fastapi_spawn-0.4.39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "fastapi-spawn"
-version = "0.4.38"
+version = "0.4.39"
 description = "A powerful CLI tool to scaffold production-ready FastAPI projects with flexible database, auth, broker, and deployment options."
 readme = "README.md"
 license = { text = "MIT" }