Files
FastapiAdmin/backend/main.py
T
zhangtao e1d7d27967 refactor(devops): 重构 Docker 配置和目录结构
- 移除旧的 `Dockerfile` 和 `nginx.conf`,新增独立的 `frontend` 和 `backend` Dockerfile
- 更新 `docker-compose.yaml`,简化服务配置并优化网络和卷挂载
- 移除 `start.sh` 和 `stop.sh` 脚本,简化部署流程
- 将环境配置文件移动到 `backend/env` 目录,并移除旧的 `.env` 文件
- 更新 `backend/app/config/setting.py` 以适配新的环境配置文件路径
2025-05-17 01:34:52 +08:00

110 lines
3.1 KiB
Python
Executable File

# -*- coding: utf-8 -*-
import os
import uvicorn
import typer
from fastapi import FastAPI
from alembic import command
from alembic.config import Config
from app.common.enums import EnvironmentEnum
shell_app = typer.Typer()
# 初始化 Alembic 配置
alembic_cfg = Config("alembic.ini")
def create_app() -> FastAPI:
from app.config.setting import settings
from app.plugin.init_app import (
register_middlewares,
register_exceptions,
register_routers,
register_files,
reset_api_docs,
lifespan
)
# 创建FastAPI应用
app = FastAPI(**settings.FASTAPI_CONFIG, lifespan=lifespan)
# 注册异常处理器
register_exceptions(app)
# 注册中间件
register_middlewares(app)
# 注册路由
register_routers(app)
# 注册静态文件
register_files(app)
# 重设API文档
reset_api_docs(app)
return app
@shell_app.command()
def run(env: EnvironmentEnum = typer.Option(EnvironmentEnum.DEV, "--env", help="运行环境 (dev, prod)")):
typer.echo("项目启动中..")
# 设置环境变量
os.environ["ENVIRONMENT"] = env.value
from app.config.setting import settings
# 启动uvicorn服务
uvicorn.run(
app='main:create_app',
**settings.UVICORN_CONFIG
)
@shell_app.command()
def init(env: EnvironmentEnum = typer.Option(EnvironmentEnum.DEV, "--env", help="运行环境 (dev, test, prod)")):
typer.echo("项目初始化中...")
import asyncio
from app.scripts.initialize import InitializeData
# 设置环境变量
os.environ["ENVIRONMENT"] = env.value
# 初始化数据
data = InitializeData()
# 使用asyncio.run来运行异步函数,
loop = asyncio.get_event_loop()
loop.run_until_complete(data.run())
# asyncio.run(data.run())
@shell_app.command()
def revision(message: str, env: EnvironmentEnum = typer.Option(EnvironmentEnum.DEV, "--env", help="运行环境 (dev, test, prod)")):
"""
生成新的 Al
embic 迁移脚本。
"""
os.environ["ENVIRONMENT"] = env.value
command.revision(alembic_cfg, message=message, autogenerate=True)
typer.echo(f"迁移脚本已生成: {message}")
@shell_app.command()
def upgrade(env: EnvironmentEnum = typer.Option(EnvironmentEnum.DEV, "--env", help="运行环境 (dev, test, prod)")):
"""
应用最新的 Alembic 迁移。
"""
os.environ["ENVIRONMENT"] = env.value
command.upgrade(alembic_cfg, "head")
typer.echo("所有迁移已应用。")
if __name__ == '__main__':
# 启动服务
# 方式一:
# uvicorn main:create_app --host 0.0.0.0 --port 8000 --factory --workers 4
# 方式二:
# python main.py run # 启动服务
# python3 main.py run --env=dev(不加默认为dev)
# python main.py init # 初始化数据
# 方式三:(linux部署使用)
# gunicorn -c gunicorn.py main:create_app
# 修改了模型后需要:重新生成迁移文件,然后应用迁移
# 生成迁移
# python main.py revision "初始化迁移" --env=dev(不加默认为dev)
# 应用迁移
# python main.py upgrade --env=dev(不加默认为dev)
shell_app()