mirror of
https://github.com/d3vyce/fastapi-toolsets.git
synced 2026-03-02 01:10:47 +01:00
34ef4da317
* chore: cleanup + add tests * chore: remove graph and show fixtures commands * feat: add async_command wrapper
120 lines
3.3 KiB
Python
120 lines
3.3 KiB
Python
"""Fixture management commands."""
|
|
|
|
from typing import Annotated
|
|
|
|
import typer
|
|
from rich.console import Console
|
|
from rich.table import Table
|
|
|
|
from ...fixtures import Context, LoadStrategy, load_fixtures_by_context
|
|
from ..config import CliConfig
|
|
from ..utils import async_command
|
|
|
|
fixture_cli = typer.Typer(
|
|
name="fixtures",
|
|
help="Manage database fixtures.",
|
|
no_args_is_help=True,
|
|
)
|
|
console = Console()
|
|
|
|
|
|
def _get_config(ctx: typer.Context) -> CliConfig:
|
|
"""Get CLI config from context."""
|
|
return ctx.obj["config"]
|
|
|
|
|
|
@fixture_cli.command("list")
|
|
def list_fixtures(
|
|
ctx: typer.Context,
|
|
context: Annotated[
|
|
str | None,
|
|
typer.Option(
|
|
"--context",
|
|
"-c",
|
|
help="Filter by context (base, production, development, testing).",
|
|
),
|
|
] = None,
|
|
) -> None:
|
|
"""List all registered fixtures."""
|
|
config = _get_config(ctx)
|
|
registry = config.get_fixtures_registry()
|
|
fixtures = registry.get_by_context(context) if context else registry.get_all()
|
|
|
|
if not fixtures:
|
|
print("No fixtures found.")
|
|
return
|
|
|
|
table = Table("Name", "Contexts", "Dependencies")
|
|
|
|
for fixture in fixtures:
|
|
contexts = ", ".join(fixture.contexts)
|
|
deps = ", ".join(fixture.depends_on) if fixture.depends_on else "-"
|
|
table.add_row(fixture.name, contexts, deps)
|
|
|
|
console.print(table)
|
|
print(f"\nTotal: {len(fixtures)} fixture(s)")
|
|
|
|
|
|
@fixture_cli.command("load")
|
|
@async_command
|
|
async def load(
|
|
ctx: typer.Context,
|
|
contexts: Annotated[
|
|
list[str] | None,
|
|
typer.Argument(
|
|
help="Contexts to load (base, production, development, testing)."
|
|
),
|
|
] = None,
|
|
strategy: Annotated[
|
|
str,
|
|
typer.Option(
|
|
"--strategy", "-s", help="Load strategy: merge, insert, skip_existing."
|
|
),
|
|
] = "merge",
|
|
dry_run: Annotated[
|
|
bool,
|
|
typer.Option(
|
|
"--dry-run", "-n", help="Show what would be loaded without loading."
|
|
),
|
|
] = False,
|
|
) -> None:
|
|
"""Load fixtures into the database."""
|
|
config = _get_config(ctx)
|
|
registry = config.get_fixtures_registry()
|
|
get_db_context = config.get_db_context()
|
|
|
|
context_list = contexts if contexts else [Context.BASE]
|
|
|
|
try:
|
|
load_strategy = LoadStrategy(strategy)
|
|
except ValueError:
|
|
typer.echo(
|
|
f"Invalid strategy: {strategy}. Use: merge, insert, skip_existing", err=True
|
|
)
|
|
raise typer.Exit(1)
|
|
|
|
ordered = registry.resolve_context_dependencies(*context_list)
|
|
|
|
if not ordered:
|
|
print("No fixtures to load for the specified context(s).")
|
|
return
|
|
|
|
print(f"\nFixtures to load ({load_strategy.value} strategy):")
|
|
for name in ordered:
|
|
fixture = registry.get(name)
|
|
instances = list(fixture.func())
|
|
model_name = type(instances[0]).__name__ if instances else "?"
|
|
print(f" - {name}: {len(instances)} {model_name}(s)")
|
|
|
|
if dry_run:
|
|
print("\n[Dry run - no changes made]")
|
|
return
|
|
|
|
async with get_db_context() as session:
|
|
result = await load_fixtures_by_context(
|
|
session, registry, *context_list, strategy=load_strategy
|
|
)
|
|
|
|
total = sum(len(items) for items in result.values())
|
|
print(f"\nLoaded {total} record(s) successfully.")
|