updates
This commit is contained in:
rUv
502
src/cli.py
Normal file
502
src/cli.py
Normal file
@@ -0,0 +1,502 @@
|
||||
"""
|
||||
Command-line interface for WiFi-DensePose API
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import click
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from src.config.settings import get_settings
|
||||
from src.logger import setup_logging, get_logger
|
||||
from src.commands.start import start_command
|
||||
from src.commands.stop import stop_command
|
||||
from src.commands.status import status_command
|
||||
|
||||
# Setup logging for CLI
|
||||
setup_logging()
|
||||
logger = get_logger(__name__)
|
||||
|
||||
|
||||
@click.group()
|
||||
@click.option(
|
||||
'--config',
|
||||
'-c',
|
||||
type=click.Path(exists=True),
|
||||
help='Path to configuration file'
|
||||
)
|
||||
@click.option(
|
||||
'--verbose',
|
||||
'-v',
|
||||
is_flag=True,
|
||||
help='Enable verbose logging'
|
||||
)
|
||||
@click.option(
|
||||
'--debug',
|
||||
is_flag=True,
|
||||
help='Enable debug mode'
|
||||
)
|
||||
@click.pass_context
|
||||
def cli(ctx, config: Optional[str], verbose: bool, debug: bool):
|
||||
"""WiFi-DensePose API Command Line Interface."""
|
||||
|
||||
# Ensure context object exists
|
||||
ctx.ensure_object(dict)
|
||||
|
||||
# Store CLI options in context
|
||||
ctx.obj['config_file'] = config
|
||||
ctx.obj['verbose'] = verbose
|
||||
ctx.obj['debug'] = debug
|
||||
|
||||
# Setup logging level
|
||||
if debug:
|
||||
import logging
|
||||
logging.getLogger().setLevel(logging.DEBUG)
|
||||
logger.debug("Debug mode enabled")
|
||||
elif verbose:
|
||||
import logging
|
||||
logging.getLogger().setLevel(logging.INFO)
|
||||
logger.info("Verbose mode enabled")
|
||||
|
||||
|
||||
@cli.command()
|
||||
@click.option(
|
||||
'--host',
|
||||
default='0.0.0.0',
|
||||
help='Host to bind to (default: 0.0.0.0)'
|
||||
)
|
||||
@click.option(
|
||||
'--port',
|
||||
default=8000,
|
||||
type=int,
|
||||
help='Port to bind to (default: 8000)'
|
||||
)
|
||||
@click.option(
|
||||
'--workers',
|
||||
default=1,
|
||||
type=int,
|
||||
help='Number of worker processes (default: 1)'
|
||||
)
|
||||
@click.option(
|
||||
'--reload',
|
||||
is_flag=True,
|
||||
help='Enable auto-reload for development'
|
||||
)
|
||||
@click.option(
|
||||
'--daemon',
|
||||
'-d',
|
||||
is_flag=True,
|
||||
help='Run as daemon (background process)'
|
||||
)
|
||||
@click.pass_context
|
||||
def start(ctx, host: str, port: int, workers: int, reload: bool, daemon: bool):
|
||||
"""Start the WiFi-DensePose API server."""
|
||||
|
||||
try:
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Override settings with CLI options
|
||||
if ctx.obj.get('debug'):
|
||||
settings.debug = True
|
||||
|
||||
# Run start command
|
||||
asyncio.run(start_command(
|
||||
settings=settings,
|
||||
host=host,
|
||||
port=port,
|
||||
workers=workers,
|
||||
reload=reload,
|
||||
daemon=daemon
|
||||
))
|
||||
|
||||
except KeyboardInterrupt:
|
||||
logger.info("Received interrupt signal, shutting down...")
|
||||
sys.exit(0)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to start server: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.command()
|
||||
@click.option(
|
||||
'--force',
|
||||
'-f',
|
||||
is_flag=True,
|
||||
help='Force stop without graceful shutdown'
|
||||
)
|
||||
@click.option(
|
||||
'--timeout',
|
||||
default=30,
|
||||
type=int,
|
||||
help='Timeout for graceful shutdown (default: 30 seconds)'
|
||||
)
|
||||
@click.pass_context
|
||||
def stop(ctx, force: bool, timeout: int):
|
||||
"""Stop the WiFi-DensePose API server."""
|
||||
|
||||
try:
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Run stop command
|
||||
asyncio.run(stop_command(
|
||||
settings=settings,
|
||||
force=force,
|
||||
timeout=timeout
|
||||
))
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to stop server: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.command()
|
||||
@click.option(
|
||||
'--format',
|
||||
type=click.Choice(['text', 'json']),
|
||||
default='text',
|
||||
help='Output format (default: text)'
|
||||
)
|
||||
@click.option(
|
||||
'--detailed',
|
||||
is_flag=True,
|
||||
help='Show detailed status information'
|
||||
)
|
||||
@click.pass_context
|
||||
def status(ctx, format: str, detailed: bool):
|
||||
"""Show the status of the WiFi-DensePose API server."""
|
||||
|
||||
try:
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Run status command
|
||||
asyncio.run(status_command(
|
||||
settings=settings,
|
||||
output_format=format,
|
||||
detailed=detailed
|
||||
))
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get status: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.group()
|
||||
def db():
|
||||
"""Database management commands."""
|
||||
pass
|
||||
|
||||
|
||||
@db.command()
|
||||
@click.option(
|
||||
'--url',
|
||||
help='Database URL (overrides config)'
|
||||
)
|
||||
@click.pass_context
|
||||
def init(ctx, url: Optional[str]):
|
||||
"""Initialize the database schema."""
|
||||
|
||||
try:
|
||||
from src.database.connection import get_database_manager
|
||||
from alembic.config import Config
|
||||
from alembic import command
|
||||
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
if url:
|
||||
settings.database_url = url
|
||||
|
||||
# Initialize database
|
||||
db_manager = get_database_manager(settings)
|
||||
|
||||
async def init_db():
|
||||
await db_manager.initialize()
|
||||
logger.info("Database initialized successfully")
|
||||
|
||||
asyncio.run(init_db())
|
||||
|
||||
# Run migrations
|
||||
alembic_cfg = Config("alembic.ini")
|
||||
command.upgrade(alembic_cfg, "head")
|
||||
logger.info("Database migrations applied successfully")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to initialize database: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@db.command()
|
||||
@click.option(
|
||||
'--revision',
|
||||
default='head',
|
||||
help='Target revision (default: head)'
|
||||
)
|
||||
@click.pass_context
|
||||
def migrate(ctx, revision: str):
|
||||
"""Run database migrations."""
|
||||
|
||||
try:
|
||||
from alembic.config import Config
|
||||
from alembic import command
|
||||
|
||||
# Run migrations
|
||||
alembic_cfg = Config("alembic.ini")
|
||||
command.upgrade(alembic_cfg, revision)
|
||||
logger.info(f"Database migrated to revision: {revision}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to run migrations: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@db.command()
|
||||
@click.option(
|
||||
'--steps',
|
||||
default=1,
|
||||
type=int,
|
||||
help='Number of steps to rollback (default: 1)'
|
||||
)
|
||||
@click.pass_context
|
||||
def rollback(ctx, steps: int):
|
||||
"""Rollback database migrations."""
|
||||
|
||||
try:
|
||||
from alembic.config import Config
|
||||
from alembic import command
|
||||
|
||||
# Rollback migrations
|
||||
alembic_cfg = Config("alembic.ini")
|
||||
command.downgrade(alembic_cfg, f"-{steps}")
|
||||
logger.info(f"Database rolled back {steps} step(s)")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to rollback database: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.group()
|
||||
def tasks():
|
||||
"""Background task management commands."""
|
||||
pass
|
||||
|
||||
|
||||
@tasks.command()
|
||||
@click.option(
|
||||
'--task',
|
||||
type=click.Choice(['cleanup', 'monitoring', 'backup']),
|
||||
help='Specific task to run'
|
||||
)
|
||||
@click.pass_context
|
||||
def run(ctx, task: Optional[str]):
|
||||
"""Run background tasks."""
|
||||
|
||||
try:
|
||||
from src.tasks.cleanup import get_cleanup_manager
|
||||
from src.tasks.monitoring import get_monitoring_manager
|
||||
from src.tasks.backup import get_backup_manager
|
||||
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
async def run_tasks():
|
||||
if task == 'cleanup' or task is None:
|
||||
cleanup_manager = get_cleanup_manager(settings)
|
||||
result = await cleanup_manager.run_all_tasks()
|
||||
logger.info(f"Cleanup result: {result}")
|
||||
|
||||
if task == 'monitoring' or task is None:
|
||||
monitoring_manager = get_monitoring_manager(settings)
|
||||
result = await monitoring_manager.run_all_tasks()
|
||||
logger.info(f"Monitoring result: {result}")
|
||||
|
||||
if task == 'backup' or task is None:
|
||||
backup_manager = get_backup_manager(settings)
|
||||
result = await backup_manager.run_all_tasks()
|
||||
logger.info(f"Backup result: {result}")
|
||||
|
||||
asyncio.run(run_tasks())
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to run tasks: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@tasks.command()
|
||||
@click.pass_context
|
||||
def status(ctx):
|
||||
"""Show background task status."""
|
||||
|
||||
try:
|
||||
from src.tasks.cleanup import get_cleanup_manager
|
||||
from src.tasks.monitoring import get_monitoring_manager
|
||||
from src.tasks.backup import get_backup_manager
|
||||
import json
|
||||
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Get task managers
|
||||
cleanup_manager = get_cleanup_manager(settings)
|
||||
monitoring_manager = get_monitoring_manager(settings)
|
||||
backup_manager = get_backup_manager(settings)
|
||||
|
||||
# Collect status
|
||||
status_data = {
|
||||
"cleanup": cleanup_manager.get_stats(),
|
||||
"monitoring": monitoring_manager.get_stats(),
|
||||
"backup": backup_manager.get_stats(),
|
||||
}
|
||||
|
||||
# Print status
|
||||
click.echo(json.dumps(status_data, indent=2))
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get task status: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.group()
|
||||
def config():
|
||||
"""Configuration management commands."""
|
||||
pass
|
||||
|
||||
|
||||
@config.command()
|
||||
@click.pass_context
|
||||
def show(ctx):
|
||||
"""Show current configuration."""
|
||||
|
||||
try:
|
||||
import json
|
||||
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Convert settings to dict (excluding sensitive data)
|
||||
config_dict = {
|
||||
"environment": settings.environment,
|
||||
"debug": settings.debug,
|
||||
"api_version": settings.api_version,
|
||||
"host": settings.host,
|
||||
"port": settings.port,
|
||||
"database": {
|
||||
"host": settings.db_host,
|
||||
"port": settings.db_port,
|
||||
"name": settings.db_name,
|
||||
"pool_size": settings.db_pool_size,
|
||||
},
|
||||
"redis": {
|
||||
"enabled": settings.redis_enabled,
|
||||
"host": settings.redis_host,
|
||||
"port": settings.redis_port,
|
||||
"db": settings.redis_db,
|
||||
},
|
||||
"monitoring": {
|
||||
"interval_seconds": settings.monitoring_interval_seconds,
|
||||
"cleanup_interval_seconds": settings.cleanup_interval_seconds,
|
||||
"backup_interval_seconds": settings.backup_interval_seconds,
|
||||
},
|
||||
"retention": {
|
||||
"csi_data_days": settings.csi_data_retention_days,
|
||||
"pose_detection_days": settings.pose_detection_retention_days,
|
||||
"metrics_days": settings.metrics_retention_days,
|
||||
"audit_log_days": settings.audit_log_retention_days,
|
||||
}
|
||||
}
|
||||
|
||||
click.echo(json.dumps(config_dict, indent=2))
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to show configuration: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@config.command()
|
||||
@click.pass_context
|
||||
def validate(ctx):
|
||||
"""Validate configuration."""
|
||||
|
||||
try:
|
||||
# Get settings
|
||||
settings = get_settings(config_file=ctx.obj.get('config_file'))
|
||||
|
||||
# Validate database connection
|
||||
from src.database.connection import get_database_manager
|
||||
|
||||
async def validate_config():
|
||||
db_manager = get_database_manager(settings)
|
||||
|
||||
try:
|
||||
await db_manager.test_connection()
|
||||
click.echo("✓ Database connection: OK")
|
||||
except Exception as e:
|
||||
click.echo(f"✗ Database connection: FAILED - {e}")
|
||||
return False
|
||||
|
||||
# Validate Redis connection (if enabled)
|
||||
if settings.redis_enabled:
|
||||
try:
|
||||
redis_stats = await db_manager.get_connection_stats()
|
||||
if "redis" in redis_stats and not redis_stats["redis"].get("error"):
|
||||
click.echo("✓ Redis connection: OK")
|
||||
else:
|
||||
click.echo("✗ Redis connection: FAILED")
|
||||
return False
|
||||
except Exception as e:
|
||||
click.echo(f"✗ Redis connection: FAILED - {e}")
|
||||
return False
|
||||
else:
|
||||
click.echo("- Redis connection: DISABLED")
|
||||
|
||||
# Validate directories
|
||||
from pathlib import Path
|
||||
|
||||
directories = [
|
||||
("Log directory", settings.log_directory),
|
||||
("Backup directory", settings.backup_directory),
|
||||
]
|
||||
|
||||
for name, directory in directories:
|
||||
path = Path(directory)
|
||||
if path.exists() and path.is_dir():
|
||||
click.echo(f"✓ {name}: OK")
|
||||
else:
|
||||
click.echo(f"✗ {name}: NOT FOUND - {directory}")
|
||||
return False
|
||||
|
||||
click.echo("\n✓ Configuration validation passed")
|
||||
return True
|
||||
|
||||
result = asyncio.run(validate_config())
|
||||
if not result:
|
||||
sys.exit(1)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to validate configuration: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
@cli.command()
|
||||
def version():
|
||||
"""Show version information."""
|
||||
|
||||
try:
|
||||
from src.config.settings import get_settings
|
||||
|
||||
settings = get_settings()
|
||||
|
||||
click.echo(f"WiFi-DensePose API v{settings.api_version}")
|
||||
click.echo(f"Environment: {settings.environment}")
|
||||
click.echo(f"Python: {sys.version}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to get version: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
cli()
|
||||
Reference in New Issue
Block a user