mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 13:12:24 +00:00
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from dataclasses import dataclass
|
|
from typing import Annotated
|
|
|
|
import uvicorn
|
|
|
|
from cappa import Arg, Subcommands, invoke
|
|
from rich.panel import Panel
|
|
from rich.progress import (
|
|
Progress,
|
|
SpinnerColumn,
|
|
TextColumn,
|
|
TimeElapsedColumn,
|
|
)
|
|
from rich.text import Text
|
|
|
|
from backend import console, get_version
|
|
from backend.core.conf import settings
|
|
from backend.plugin.tools import get_plugins, install_requirements
|
|
|
|
|
|
def run(host: str, port: int, reload: bool, workers: int | None) -> None:
|
|
console.print(Text('检测插件依赖...', style='bold cyan'))
|
|
|
|
plugins = get_plugins()
|
|
|
|
with Progress(
|
|
SpinnerColumn(finished_text='[bold green]插件依赖安装完成[/]'),
|
|
TextColumn('[green]{task.completed}/{task.total}[/]'),
|
|
TimeElapsedColumn(),
|
|
console=console,
|
|
) as progress:
|
|
task = progress.add_task('安装插件依赖...', total=len(plugins))
|
|
for i, plugin in enumerate(plugins):
|
|
install_requirements(plugin)
|
|
progress.advance(task)
|
|
|
|
url = f'http://{host}:{port}'
|
|
docs_url = url + settings.FASTAPI_DOCS_URL
|
|
redoc_url = url + settings.FASTAPI_REDOC_URL
|
|
openapi_url = url + settings.FASTAPI_OPENAPI_URL
|
|
|
|
console.print(Text('启动 fba 服务...', style='bold magenta'))
|
|
|
|
panel_content = Text()
|
|
panel_content.append(f'📝 Swagger 文档: {docs_url}\n', style='blue')
|
|
panel_content.append(f'📚 Redoc 文档: {redoc_url}\n', style='yellow')
|
|
panel_content.append(f'📡 OpenAPI JSON: {openapi_url}\n', style='green')
|
|
panel_content.append(
|
|
'🌍 fba 官方文档: https://fastapi-practices.github.io/fastapi_best_architecture_docs/',
|
|
style='cyan',
|
|
)
|
|
|
|
console.print(Panel(panel_content, title='fba 服务信息', border_style='purple', padding=(1, 2)))
|
|
uvicorn.run(app='backend.main:app', host=host, port=port, reload=reload, workers=workers)
|
|
|
|
|
|
@dataclass
|
|
class Run:
|
|
host: Annotated[
|
|
str,
|
|
Arg(
|
|
long=True,
|
|
default='127.0.0.1',
|
|
help='提供服务的主机 IP 地址，对于本地开发，请使用 `127.0.0.1`。'
|
|
'要启用公共访问，例如在局域网中，请使用 `0.0.0.0`',
|
|
),
|
|
]
|
|
port: Annotated[
|
|
int,
|
|
Arg(long=True, default=8000, help='提供服务的主机端口号'),
|
|
]
|
|
reload: Annotated[
|
|
bool,
|
|
Arg(long=True, default=True, help='启用在（代码）文件更改时自动重新加载服务器'),
|
|
]
|
|
workers: Annotated[
|
|
int | None,
|
|
Arg(long=True, default=None, help='使用多个工作进程。与 `--reload` 标志互斥'),
|
|
]
|
|
|
|
def __call__(self):
|
|
run(host=self.host, port=self.port, reload=self.reload, workers=self.workers)
|
|
|
|
|
|
@dataclass
|
|
class FbaCli:
|
|
version: Annotated[
|
|
bool,
|
|
Arg(short='-V', long=True, default=False, help='打印 fba 当前版本号'),
|
|
]
|
|
subcmd: Subcommands[Run | None] = None
|
|
|
|
def __call__(self):
|
|
if self.version:
|
|
get_version()
|
|
|
|
|
|
def main() -> None:
|
|
invoke(FbaCli)
|