94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from datetime import datetime
|
|
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⏰ {datetime.now().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)
|