upload project source code

后端源码/yifan.action-ai.cn/api/app/utils/console.py

@@ -0,0 +1,94 @@
+# -*- coding: utf-8 -*-
+
+from datetime import datetime
+from app.utils.time_util import TimeUtil
+from typing import Optional
+from rich import get_console
+from rich.panel import Panel
+from rich.text import Text
+from rich.console import Group
+
+from app.config.setting import settings
+
+console = get_console()
+
+
+def create_service_panel(
+    host: str, port: int, reload: bool, *, 
+    redis_ready: Optional[bool] = None, 
+    scheduler_jobs: Optional[int] = None, 
+    scheduler_status: Optional[str] = None,
+) -> Panel:
+    """创建简洁的服务启动信息面板"""
+    
+    url = f'http://{host}:{port}'
+    base_url = f'{url}{settings.ROOT_PATH}'
+    docs_url = base_url + settings.DOCS_URL
+    redoc_url = base_url + settings.REDOC_URL
+
+    # 核心服务信息
+    service_info = Text()
+    service_info.append(f"服务名称 {settings.TITLE} • 优雅 • 简洁 • 高效", style="bold magenta")
+    service_info.append(f"\n当前版本 v{settings.VERSION}" , style="bold green")
+    service_info.append(f"\n服务地址 {url}", style="bold blue")
+    service_info.append(f"\n运行环境 {settings.ENVIRONMENT.value if hasattr(settings.ENVIRONMENT, 'value') else settings.ENVIRONMENT}", style="bold red")
+    service_info.append(f"\n重载配置: {'✅ 开启' if reload else '❌ 关闭'}", style="bold italic")
+    service_info.append(f"\n调试模式: {'✅ 开启' if settings.DEBUG else '❌ 关闭'}", style="bold italic")
+    service_info.append(f"\n数据库类型: {settings.DATABASE_TYPE} 数据库", style="bold italic")
+    service_info.append(f"\nRedis: {'✅ 已连接' if redis_ready else '❌ 未连接'}", style="bold italic")
+    service_info.append(f"\n定时任务 {'✅ 运行中' if scheduler_status == 'running' else '⏸️ 暂停'} {scheduler_jobs}", style="bold italic")
+    
+    docs_info = Text()
+    docs_info.append("📖 文档", style="bold magenta")
+    docs_info.append(f"\n🔗 Swagger: {docs_url}", style="blue link")
+    docs_info.append(f"\n🔗 ReDoc: {redoc_url}", style="blue link")
+
+    final_content = Group(
+        service_info,
+        "\n" + "─" * 40,
+        docs_info,
+    )
+
+    return Panel(
+        renderable=final_content,
+        title="[bold purple]🚀 服务启动完成[/]",
+        border_style="green",
+        padding=(1, 2)
+    )
+
+
+def run(host: str, port: int, reload: bool, *, 
+        redis_ready: Optional[bool] = None, 
+        scheduler_jobs: Optional[int] = None, 
+        scheduler_status: Optional[str] = None
+    ) -> None:
+    """显示启动信息面板"""
+    
+    # 创建并显示启动面板
+    service_panel = create_service_panel(
+        host=host,
+        port=port,
+        reload=reload,
+        redis_ready=redis_ready,
+        scheduler_jobs=scheduler_jobs,
+        scheduler_status=scheduler_status,
+    )
+    
+    console.print(service_panel)
+
+def display_shutdown_info():
+    """显示关闭信息"""
+    shutdown_content = Text()
+    shutdown_content.append("🛑 ", style="bold red")
+    shutdown_content.append("FastapiAdmin 服务关闭")
+    shutdown_content.append(f"\n⏰ {TimeUtil.now_u8().strftime('%H:%M:%S')}")
+    shutdown_content.append("\n👋 感谢使用！", style="dim")
+    
+    shutdown_panel = Panel(
+        shutdown_content,
+        title="[bold red]服务关闭[/]",
+        border_style="red",
+        padding=(1, 2)
+    )
+    
+    console.print(shutdown_panel)