Antoni Nuñez Romeu
118 lines
3.1 KiB
Python
118 lines
3.1 KiB
Python
"""CLI entrypoints for the PIP platform."""
|
|
import asyncio
|
|
import sys
|
|
|
|
import click
|
|
|
|
from src.core.logging import configure_logging, get_logger
|
|
from src.infrastructure.cache.redis_cache import redis_cache
|
|
from src.infrastructure.config.settings import settings
|
|
from src.infrastructure.database.session import async_session_factory, engine
|
|
from src.infrastructure.messaging.rabbitmq import rabbitmq_client
|
|
from src.services.auth.auth_service import AuthService
|
|
from src.domain.entities import UserRole
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
"""PIP platform CLI."""
|
|
configure_logging()
|
|
|
|
|
|
@cli.command()
|
|
def run():
|
|
"""Start the API server."""
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"src.main:app",
|
|
host="0.0.0.0",
|
|
port=8000,
|
|
reload=True if settings.DEBUG else False,
|
|
log_level=settings.LOG_LEVEL.lower(),
|
|
)
|
|
|
|
|
|
@cli.command()
|
|
def migrate_upgrade():
|
|
"""Apply database migrations."""
|
|
from alembic.config import Config
|
|
from alembic import command
|
|
|
|
cfg = Config("alembic.ini")
|
|
command.upgrade(cfg, "head")
|
|
click.echo("Migrations applied.")
|
|
|
|
|
|
@cli.command()
|
|
def migrate_downgrade():
|
|
"""Roll back the last database migration."""
|
|
from alembic.config import Config
|
|
from alembic import command
|
|
|
|
cfg = Config("alembic.ini")
|
|
command.downgrade(cfg, "-1")
|
|
click.echo("Rollback complete.")
|
|
|
|
|
|
@cli.command()
|
|
def migrate_revision():
|
|
"""Create a new database revision."""
|
|
from alembic.config import Config
|
|
from alembic import command
|
|
|
|
message = click.prompt("Revision message")
|
|
cfg = Config("alembic.ini")
|
|
command.revision(cfg, autogenerate=True, message=message)
|
|
click.echo("Revision created.")
|
|
|
|
|
|
@cli.command()
|
|
@click.option("--username", required=True)
|
|
@click.option("--password", required=True)
|
|
@click.option("--email", default=None)
|
|
@click.option("--role", default="admin", type=click.Choice([r.value for r in UserRole]))
|
|
def create_user(username: str, password: str, email: str | None, role: str):
|
|
"""Create a technical user."""
|
|
asyncio.run(_create_user(username, password, email, role))
|
|
|
|
|
|
async def _create_user(username: str, password: str, email: str | None, role: str):
|
|
async with async_session_factory() as session:
|
|
service = AuthService(session)
|
|
user = await service.register(
|
|
username=username,
|
|
password=password,
|
|
email=email,
|
|
role=UserRole(role),
|
|
)
|
|
click.echo(f"Created user '{user.username}' with role '{user.role.value}'.")
|
|
|
|
|
|
@cli.command()
|
|
def health():
|
|
"""Check platform health across services."""
|
|
asyncio.run(_health())
|
|
|
|
|
|
async def _health():
|
|
logger = get_logger("cli.health")
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(lambda s: None)
|
|
click.echo("Database: OK")
|
|
|
|
await redis_cache.connect()
|
|
ok = await redis_cache.health_check()
|
|
click.echo(f"Redis: {'OK' if ok else 'FAIL'}")
|
|
await redis_cache.disconnect()
|
|
|
|
await rabbitmq_client.connect()
|
|
ok = await rabbitmq_client.health_check()
|
|
click.echo(f"RabbitMQ: {'OK' if ok else 'FAIL'}")
|
|
await rabbitmq_client.disconnect()
|
|
|
|
await engine.dispose()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|