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

aury/boot/commands/add.py

@@ -0,0 +1,76 @@
+"""给已有项目添加可选模块。
+
+用法：
+    aury add admin-console   # 添加 Admin Console 模块
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from rich.console import Console
+import typer
+
+from .config import get_project_config
+
+console = Console()
+
+app = typer.Typer(
+    name="add",
+    help="给已有项目添加可选模块",
+    no_args_is_help=True,
+)
+
+
+@app.command(name="admin-console")
+def add_admin_console(
+    force: bool = typer.Option(False, "--force", "-f", help="强制覆盖已有 admin_console.py"),
+    enable_env: bool = typer.Option(True, "--enable-env", help="尝试在 .env.example 中开启 ADMIN_* 示例配置"),
+) -> None:
+    """添加 Admin Console 模块到现有项目。
+
+    将在项目代码根（包根或平铺）下生成 `admin_console.py`，便于注册 SQLAdmin 的 ModelView / 自定义认证。
+
+    同时（可选）在 `.env.example` 中启用 ADMIN 示例配置，方便快速启动。
+    """
+    from .init import init_admin_console_module
+
+    base_path = Path.cwd()
+    config = get_project_config(base_path)
+    code_root = config.get_package_dir(base_path)
+    import_prefix = config.get_import_prefix()
+
+    result = init_admin_console_module(
+        base_path=base_path,
+        code_root=code_root,
+        import_prefix=import_prefix,
+        force=force,
+        enable_env=enable_env,
+    )
+
+    # 输出结果
+    admin_pkg = code_root / "admin_console"
+    if result["file_created"]:
+        console.print(f"[green]✅ 已生成: {admin_pkg.relative_to(base_path)}/[/green]")
+    elif result["file_existed"]:
+        console.print(f"[yellow]⚠️  已存在: {admin_pkg.relative_to(base_path)}/（使用 --force 覆盖）[/yellow]")
+
+    if enable_env:
+        if result["env_updated"]:
+            console.print("[green]✅ 已在 .env.example 中启用 ADMIN_* 示例配置[/green]")
+        elif (base_path / ".env.example").exists():
+            console.print("[dim]ℹ️  .env.example 已包含 ADMIN_* 配置示例[/dim]")
+        else:
+            console.print("[dim]ℹ️  未找到 .env.example，已跳过环境示例更新[/dim]")
+
+    console.print()
+    console.print("[bold]下一步：[/bold]")
+    console.print("  1. 安装扩展依赖（如未安装）：")
+    console.print('     [cyan]uv add "aury-boot[admin]"[/cyan]')
+    console.print("  2. 复制 .env.example 并按需修改：")
+    console.print("     [cyan]cp .env.example .env[/cyan]")
+    console.print("  3. 启动开发服务器并访问：")
+    console.print("     [cyan]aury server dev[/cyan] → http://127.0.0.1:8000/api/admin-console")
+
+
+__all__ = ["app"]

aury/boot/commands/app.py

@@ -0,0 +1,105 @@
+"""Aury Boot 统一命令行入口。
+
+提供统一的 CLI 入口，整合所有子命令：
+- aury init              项目脚手架初始化
+- aury generate          代码生成器
+- aury server            服务器管理
+- aury scheduler         独立运行调度器
+- aury worker            运行任务队列 Worker
+- aury migrate           数据库迁移
+- aury docker            Docker 配置生成
+- aury docs              生成/更新项目文档
+
+使用示例：
+    aury init                      # 初始化项目
+    aury generate crud user        # 生成 CRUD 代码
+    aury server dev                # 启动开发服务器
+    aury scheduler                 # 独立运行调度器
+    aury worker                    # 运行 Worker
+    aury migrate up                # 执行数据库迁移
+    aury docs all --force          # 更新所有文档
+"""
+
+from __future__ import annotations
+
+import typer
+
+app: typer.Typer | None = None
+_registered = False
+
+
+def _get_app() -> typer.Typer:
+    """获取并初始化 Typer 应用（延迟加载）。"""
+    global app, _registered
+    
+    if app is None:
+        app = typer.Typer(
+            name="aury",
+            help="🚀 Aury Boot CLI - 现代化微服务开发工具",
+            add_completion=True,
+            no_args_is_help=True,
+            rich_markup_mode="rich",
+        )
+        
+        @app.callback(invoke_without_command=True)
+        def callback(
+            ctx: typer.Context,
+            version: bool = typer.Option(
+                False,
+                "--version",
+                "-v",
+                help="显示版本信息",
+                is_eager=True,
+            ),
+        ) -> None:
+            """Aury Boot - 现代化微服务基础架构框架。"""
+            if version:
+                from rich.console import Console
+
+                from aury.boot import __version__
+                console = Console()
+                console.print(f"[bold cyan]Aury Boot[/bold cyan] v{__version__}")
+                raise typer.Exit()
+    
+    if not _registered:
+        _registered = True
+        # 延迟导入子命令
+        from .add import app as add_app
+        from .docker import app as docker_app
+        from .docs import app as docs_app
+        from .generate import app as generate_app
+        from .init import init
+        from .migrate import app as migrate_app
+        from .scheduler import app as scheduler_app
+        from .server import app as server_app
+        from .worker import app as worker_app
+
+        app.command(name="init", help="🎯 初始化项目脚手架")(init)
+        app.add_typer(add_app, name="add", help="➕ 添加可选模块")
+        app.add_typer(generate_app, name="generate", help="⚡ 代码生成器")
+        app.add_typer(server_app, name="server", help="🖥️  服务器管理")
+        app.add_typer(scheduler_app, name="scheduler", help="🕐 独立运行调度器")
+        app.add_typer(worker_app, name="worker", help="⚙️  运行任务队列 Worker")
+        app.add_typer(migrate_app, name="migrate", help="🗃️  数据库迁移")
+        app.add_typer(docker_app, name="docker", help="🐳 Docker 配置")
+        app.add_typer(docs_app, name="docs", help="📚 生成/更新项目文档")
+    
+    return app
+
+
+def main() -> None:
+    """CLI 入口点。"""
+    _get_app()()
+
+
+# 为了向后兼容，允许 `from .app import app`
+def __getattr__(name: str):
+    if name == "app":
+        return _get_app()
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "app",
+    "main",
+]

aury/boot/commands/config.py

@@ -0,0 +1,177 @@
+"""项目配置读取工具。
+
+读取和写入 pyproject.toml 中的 [tool.aury] 配置。
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+
+@dataclass
+class ProjectConfig:
+    """Aury 项目配置。"""
+
+    package: str | None = None  # 顶层包名，None 表示无顶层包
+    app: str | None = None      # 应用入口，形如 "main:app"
+
+    @property
+    def has_package(self) -> bool:
+        """是否有顶层包。"""
+        return self.package is not None and self.package != ""
+
+    def get_import_prefix(self) -> str:
+        """获取 import 前缀。
+
+        有包名返回 "mypackage."，无包名返回 ""。
+        """
+        return f"{self.package}." if self.has_package else ""
+
+    def get_package_dir(self, base_path: Path) -> Path:
+        """获取包目录路径。
+
+        有包名返回 base_path/mypackage，无包名返回 base_path。
+        """
+        return base_path / self.package if self.has_package else base_path
+
+
+def get_project_config(base_path: Path | None = None) -> ProjectConfig:
+    """读取项目配置。
+
+    从 pyproject.toml 的 [tool.aury] 读取配置。
+
+    Args:
+        base_path: 项目根目录，默认为当前目录
+
+    Returns:
+        ProjectConfig 实例
+    """
+    if base_path is None:
+        base_path = Path.cwd()
+
+    pyproject_path = base_path / "pyproject.toml"
+
+    if not pyproject_path.exists():
+        return ProjectConfig()
+
+    content = pyproject_path.read_text(encoding="utf-8")
+
+    # 简单解析 [tool.aury] 部分
+    package = _parse_toml_value(content, "tool.aury", "package")
+    app = _parse_toml_value(content, "tool.aury", "app")
+
+    return ProjectConfig(package=package, app=app)
+
+
+def save_project_config(config: ProjectConfig, base_path: Path | None = None) -> bool:
+    """保存项目配置到 pyproject.toml。
+
+    Args:
+        config: 项目配置
+        base_path: 项目根目录
+
+    Returns:
+        是否保存成功
+    """
+    if base_path is None:
+        base_path = Path.cwd()
+
+    pyproject_path = base_path / "pyproject.toml"
+
+    if not pyproject_path.exists():
+        return False
+
+    content = pyproject_path.read_text(encoding="utf-8")
+
+    # 检查是否已有 [tool.aury] 部分
+    if "[tool.aury]" in content:
+        # 更新现有配置
+        lines = content.split("\n")
+        new_lines = []
+        in_aury_section = False
+
+        for line in lines:
+            if line.strip() == "[tool.aury]":
+                in_aury_section = True
+                new_lines.append(line)
+                # 添加（或更新）配置项
+                if config.has_package:
+                    new_lines.append(f'package = "{config.package}"')
+                if config.app:
+                    new_lines.append(f'app = "{config.app}"')
+                continue
+
+            if in_aury_section:
+                # 跳过旧的 package/app 配置
+                if line.strip().startswith("package") or line.strip().startswith("app"):
+                    continue
+                # 遇到新的 section 结束
+                if line.strip().startswith("["):
+                    in_aury_section = False
+
+            new_lines.append(line)
+
+        content = "\n".join(new_lines)
+    else:
+        # 添加新的 [tool.aury] 部分
+        aury_section = "\n[tool.aury]\n"
+        if config.has_package:
+            aury_section += f'package = "{config.package}"\n'
+        if config.app:
+            aury_section += f'app = "{config.app}"\n'
+
+        content += aury_section
+
+    pyproject_path.write_text(content, encoding="utf-8")
+    return True
+
+
+def _parse_toml_value(content: str, section: str, key: str) -> str | None:
+    """简单解析 TOML 值。
+
+    注意：这是一个简单实现，不处理复杂情况。
+
+    Args:
+        content: TOML 文件内容
+        section: 节名（如 "tool.aury"）
+        key: 键名
+
+    Returns:
+        值字符串，找不到返回 None
+    """
+    import re
+
+    # 构建 section 匹配
+    section_pattern = r"\[" + re.escape(section) + r"\]"
+
+    # 查找 section
+    section_match = re.search(section_pattern, content)
+    if not section_match:
+        return None
+
+    # 从 section 开始查找 key
+    section_start = section_match.end()
+
+    # 查找下一个 section（或文件结尾）
+    next_section = re.search(r"\n\[", content[section_start:])
+    if next_section:
+        section_content = content[section_start : section_start + next_section.start()]
+    else:
+        section_content = content[section_start:]
+
+    # 查找 key = value
+    key_pattern = rf"^\s*{re.escape(key)}\s*=\s*[\"']?([^\"'\n]+)[\"']?\s*$"
+    key_match = re.search(key_pattern, section_content, re.MULTILINE)
+
+    if key_match:
+        return key_match.group(1).strip().strip("\"'")
+
+    return None
+
+
+__all__ = [
+    "ProjectConfig",
+    "get_project_config",
+    "save_project_config",
+]