aury-boot 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

aury/boot/commands/migrate/__init__.py

@@ -0,0 +1,37 @@
+"""迁移命令行工具。
+
+提供数据库迁移管理的命令行接口。
+"""
+
+from .app import app, get_manager
+from .commands import (
+    check,
+    down,
+    history,
+    make,
+    merge,
+    show,
+    status,
+    up,
+)
+
+
+def main() -> None:
+    """命令行入口函数。"""
+    app()
+
+
+__all__ = [
+    "app",
+    "check",
+    "down",
+    "get_manager",
+    "history",
+    "main",
+    "make",
+    "merge",
+    "show",
+    "status",
+    "up",
+]
+

aury/boot/commands/migrate/app.py

@@ -0,0 +1,54 @@
+"""迁移 CLI 应用定义。
+
+定义 Typer 应用和辅助函数。
+"""
+
+from __future__ import annotations
+
+import typer
+
+from aury.boot.application.config import BaseConfig
+from aury.boot.application.migrations import MigrationManager
+
+# 创建 Typer 应用
+app = typer.Typer(
+    name="migrate",
+    help="数据库迁移管理工具（类似 Django 的 migrate 命令）",
+    add_completion=False,
+)
+
+
+def get_manager(
+    config_override: str | None = None,
+) -> MigrationManager:
+    """获取迁移管理器。
+    
+    从应用配置中获取所有必要参数并创建迁移管理器。
+    
+    Args:
+        config_override: 覆盖配置文件路径（可选）
+        
+    Returns:
+        MigrationManager: 迁移管理器实例
+    """
+    # 加载应用配置
+    app_config = BaseConfig()
+    
+    # 从配置中提取参数
+    migration_settings = app_config.migration
+    
+    # 创建迁移管理器（将配置转换为参数传递）
+    return MigrationManager(
+        database_url=app_config.database.url,
+        config_path=config_override or migration_settings.config_path,
+        script_location=migration_settings.script_location,
+        model_modules=migration_settings.model_modules,
+        auto_create=migration_settings.auto_create,
+    )
+
+
+__all__ = [
+    "app",
+    "get_manager",
+]
+

aury/boot/commands/migrate/commands.py

@@ -0,0 +1,303 @@
+"""迁移命令实现。
+
+实现所有迁移相关的命令行命令。
+"""
+
+from __future__ import annotations
+
+import asyncio
+import traceback
+
+from rich.console import Console
+from rich.table import Table
+import typer
+
+from .app import app, get_manager
+
+console = Console()
+
+
+def _handle_exception(e: Exception, operation: str) -> None:
+    """统一处理异常，打印完整堆栈信息。
+    
+    Args:
+        e: 异常对象
+        operation: 操作名称（用于错误消息）
+    """
+    typer.echo(f"\n❌ {operation}失败: {e}", err=True)
+    typer.echo("\n📋 异常堆栈:", err=True)
+    typer.echo(traceback.format_exc(), err=True)
+    raise typer.Exit(1)
+
+
+@app.command()
+def make(
+    message: str = typer.Option(..., "-m", "--message", help="迁移消息"),
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+    autogenerate: bool = typer.Option(True, "--autogenerate/--no-autogenerate", help="是否自动生成"),
+    dry_run: bool = typer.Option(False, "--dry-run", help="干运行（只检测变更，不生成文件）"),
+) -> None:
+    """生成迁移文件（类似 Django 的 makemigrations）。
+    
+    示例:
+        aury migrate make -m "add user table"
+        aury migrate make -m "update schema" --no-autogenerate
+        aury migrate make -m "check changes" --dry-run
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _make():
+            result = await manager.make_migrations(
+                message=message,
+                autogenerate=autogenerate,
+                dry_run=dry_run,
+            )
+            
+            if dry_run:
+                changes = result.get("changes", [])
+                if changes:
+                    typer.echo(f"\n📝 检测到 {len(changes)} 个变更:")
+                    for change in changes:
+                        typer.echo(f"  - {change['type']}: {change['description']}")
+                else:
+                    typer.echo("✅ 没有检测到模型变更")
+            else:
+                typer.echo(f"✅ 迁移文件已生成: {result.get('path', '')}")
+                changes = result.get("changes", [])
+                if changes:
+                    typer.echo(f"📝 包含 {len(changes)} 个变更")
+        
+        asyncio.run(_make())
+    except Exception as e:
+        _handle_exception(e, "生成迁移")
+
+
+@app.command()
+def up(
+    revision: str = typer.Option("head", "-r", "--revision", help="目标版本（默认 head）"),
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+    dry_run: bool = typer.Option(False, "--dry-run", help="干运行（只显示会执行的迁移，不实际执行）"),
+) -> None:
+    """执行迁移（类似 Django 的 migrate）。
+    
+    示例:
+        aury migrate up
+        aury migrate up -r "abc123"
+        aury migrate up --dry-run
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _upgrade():
+            await manager.upgrade(revision=revision, dry_run=dry_run)
+            if not dry_run:
+                typer.echo(f"✅ 迁移已执行到版本: {revision}")
+        
+        asyncio.run(_upgrade())
+    except Exception as e:
+        _handle_exception(e, "执行迁移")
+
+
+@app.command()
+def down(
+    revision: str = typer.Argument(..., help="目标版本（如 previous, -1, 或具体版本号）"),
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+    dry_run: bool = typer.Option(False, "--dry-run", help="干运行（只显示会回滚的迁移，不实际执行）"),
+) -> None:
+    """回滚迁移。
+    
+    示例:
+        aury migrate down previous
+        aury migrate down -1
+        aury migrate down abc123 --dry-run
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _downgrade():
+            await manager.downgrade(revision=revision, dry_run=dry_run)
+            if not dry_run:
+                typer.echo(f"✅ 迁移已回滚到版本: {revision}")
+        
+        asyncio.run(_downgrade())
+    except Exception as e:
+        _handle_exception(e, "回滚迁移")
+
+
+@app.command()
+def status(
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+) -> None:
+    """查看迁移状态（类似 Django 的 showmigrations）。
+    
+    示例:
+        aury migrate status
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _status():
+            status_info = await manager.status()
+            
+            # 格式化输出
+            typer.echo("\n📊 迁移状态:")
+            typer.echo(f"  当前版本: {status_info.get('current', 'None')}")
+            typer.echo(f"  最新版本: {status_info.get('head', 'None')}")
+            
+            pending = status_info.get('pending', [])
+            applied = status_info.get('applied', [])
+            
+            if pending:
+                typer.echo(f"\n⏳ 待执行迁移 ({len(pending)}):")
+                for rev in pending:
+                    typer.echo(f"  - {rev}")
+            else:
+                typer.echo("\n✅ 所有迁移已执行")
+            
+            if applied:
+                typer.echo(f"\n✅ 已执行迁移 ({len(applied)}):")
+                for rev in applied:
+                    typer.echo(f"  - {rev}")
+        
+        asyncio.run(_status())
+    except Exception as e:
+        _handle_exception(e, "查看状态")
+
+
+@app.command()
+def show(
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+) -> None:
+    """显示所有迁移（类似 Django 的 showmigrations）。
+    
+    示例:
+        aury migrate show
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _show():
+            migrations = await manager.show()
+            
+            if not migrations:
+                typer.echo("📝 没有找到迁移文件")
+                return
+            
+            # 使用 Rich 表格显示
+            table = Table(title="📝 所有迁移", show_header=True, header_style="bold magenta")
+            table.add_column("版本", style="cyan", width=15)
+            table.add_column("父版本", style="yellow", width=15)
+            table.add_column("消息", style="green")
+            
+            for mig in migrations:
+                revision = mig.get('revision', '')[:12]
+                down_revision = mig.get('down_revision', '')[:12] if mig.get('down_revision') else '-'
+                message = mig.get('message', '')
+                table.add_row(revision, down_revision, message)
+            
+            console.print(table)
+        
+        asyncio.run(_show())
+    except Exception as e:
+        _handle_exception(e, "显示迁移")
+
+
+@app.command()
+def check(
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+) -> None:
+    """检查迁移（类似 Django 的 check）。
+    
+    检查迁移文件是否有问题。
+    
+    示例:
+        aury migrate check
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _check():
+            result = await manager.check()
+            
+            if result["valid"]:
+                typer.echo("✅ 迁移检查通过")
+            else:
+                typer.echo("❌ 发现迁移问题:")
+                for issue in result["issues"]:
+                    typer.echo(f"  - {issue}")
+            
+            if result["warnings"]:
+                typer.echo("\n⚠️  警告:")
+                for warning in result["warnings"]:
+                    typer.echo(f"  - {warning}")
+            
+            typer.echo("\n📊 统计:")
+            typer.echo(f"  迁移总数: {result['revision_count']}")
+            typer.echo(f"  Head 数量: {result['head_count']}")
+        
+        asyncio.run(_check())
+    except Exception as e:
+        _handle_exception(e, "检查")
+
+
+@app.command()
+def merge(
+    revisions: str = typer.Argument(..., help="要合并的版本（逗号分隔）"),
+    message: str | None = typer.Option(None, "-m", "--message", help="合并消息"),
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+) -> None:
+    """合并迁移（类似 Django 的迁移合并）。
+    
+    当有多个分支时，创建合并迁移。
+    
+    示例:
+        aury migrate merge "abc123,def456"
+        aury migrate merge "abc123,def456" -m "merge branches"
+    """
+    try:
+        manager = get_manager(config_override=config)
+        revision_list = [r.strip() for r in revisions.split(",")]
+        
+        async def _merge():
+            result = await manager.merge(revisions=revision_list, message=message)
+            typer.echo(f"✅ 迁移已合并: {result}")
+        
+        asyncio.run(_merge())
+    except Exception as e:
+        _handle_exception(e, "合并迁移")
+
+
+@app.command()
+def history(
+    verbose: bool = typer.Option(False, "-v", "--verbose", help="显示详细信息"),
+    config: str | None = typer.Option(None, "--config", help="Alembic 配置文件路径（覆盖默认配置）"),
+) -> None:
+    """显示迁移历史。
+    
+    示例:
+        aury migrate history
+        aury migrate history --verbose
+    """
+    try:
+        manager = get_manager(config_override=config)
+        
+        async def _history():
+            await manager.history(verbose=verbose)
+        
+        asyncio.run(_history())
+    except Exception as e:
+        _handle_exception(e, "显示历史")
+
+
+__all__ = [
+    "check",
+    "down",
+    "history",
+    "make",
+    "merge",
+    "show",
+    "status",
+    "up",
+]
+

aury/boot/commands/scheduler.py

@@ -0,0 +1,124 @@
+"""调度器命令 - 独立运行调度器。
+
+使用示例：
+    aury scheduler              # 运行调度器
+    aury scheduler --app main:app  # 指定应用模块
+"""
+
+from __future__ import annotations
+
+import asyncio
+from pathlib import Path
+import sys
+
+from rich.console import Console
+import typer
+
+console = Console()
+
+app = typer.Typer(
+    name="scheduler",
+    help="🕐 独立运行调度器",
+    no_args_is_help=False,
+)
+
+
+def _detect_app_module() -> str:
+    """自动检测应用模块路径。"""
+    import os
+
+    # 1. 环境变量
+    if app_module := os.environ.get("APP_MODULE"):
+        return app_module
+
+    # 2. pyproject.toml 中的 [tool.aury] 配置
+    pyproject_path = Path.cwd() / "pyproject.toml"
+    if pyproject_path.exists():
+        try:
+            import tomllib
+            with open(pyproject_path, "rb") as f:
+                data = tomllib.load(f)
+            if package := data.get("tool", {}).get("aury", {}).get("package"):
+                return f"{package}.main:app"
+        except Exception:
+            pass
+
+    # 3. 默认
+    return "main:app"
+
+
+@app.callback(invoke_without_command=True)
+def run_scheduler(
+    ctx: typer.Context,
+    app_path: str | None = typer.Option(
+        None,
+        "--app",
+        "-a",
+        help="应用模块路径（默认自动检测，如 main:app）",
+    ),
+) -> None:
+    """独立运行调度器进程。
+
+    调度器会加载应用中注册的所有定时任务并执行。
+
+    示例：
+        aury scheduler                    # 自动检测应用
+        aury scheduler --app main:app     # 指定应用模块
+        aury scheduler -a myapp.main:app  # 指定包中的应用
+    """
+    if ctx.invoked_subcommand is not None:
+        return
+
+    # 确保当前目录在 Python 路径中
+    cwd = str(Path.cwd())
+    if cwd not in sys.path:
+        sys.path.insert(0, cwd)
+
+    app_module = app_path or _detect_app_module()
+    console.print("[bold cyan]🕐 启动独立调度器[/bold cyan]")
+    console.print(f"   应用: [green]{app_module}[/green]")
+
+    try:
+        # 导入应用
+        module_path, app_name = app_module.rsplit(":", 1)
+        module = __import__(module_path, fromlist=[app_name])
+        application = getattr(module, app_name)
+
+        # 设置日志上下文
+        from aury.boot.common.logging import set_service_context
+        set_service_context("scheduler")
+
+        # 获取调度器
+        from aury.boot.infrastructure.scheduler import SchedulerManager
+        scheduler = SchedulerManager.get_instance()
+
+        console.print("[bold green]✅ 调度器启动成功[/bold green]")
+        console.print("[dim]按 Ctrl+C 停止[/dim]")
+
+        # 运行调度器
+        async def _run():
+            await scheduler.initialize()
+            scheduler.start()
+            try:
+                # 保持运行
+                while True:
+                    await asyncio.sleep(1)
+            except asyncio.CancelledError:
+                pass
+            finally:
+                scheduler.shutdown()
+
+        asyncio.run(_run())
+
+    except KeyboardInterrupt:
+        console.print("\n[yellow]调度器已停止[/yellow]")
+    except ImportError as e:
+        console.print(f"[red]❌ 无法导入应用: {e}[/red]")
+        console.print("[dim]请确保应用模块路径正确，如 main:app[/dim]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]❌ 启动失败: {e}[/red]")
+        raise typer.Exit(1)
+
+
+__all__ = ["app"]

aury/boot/commands/server/__init__.py

@@ -0,0 +1,21 @@
+"""服务器管理命令。
+
+提供 CLI 接口来管理和运行应用服务器。
+
+使用示例:
+    python -m aury.boot.commands.server run \\
+        --host 0.0.0.0 \\
+        --port 8000 \\
+        --workers 4 \\
+        --reload
+"""
+
+from .app import app, server_cli
+
+__all__ = [
+    "app",
+    "server_cli",
+]
+
+
+