mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-21 04:46:26 +00:00
98 lines
2.3 KiB
Python
Executable File
98 lines
2.3 KiB
Python
Executable File
# -*- coding: utf-8 -*-
|
|
|
|
import uvicorn
|
|
import typer
|
|
from fastapi import 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
|
|
)
|
|
from app.core.logger import logger
|
|
|
|
|
|
shell_app = typer.Typer()
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
# 创建FastAPI应用
|
|
app = FastAPI(**settings.get_backend_app_attributes, 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():
|
|
logger.info(settings.BANNER)
|
|
# 启动uvicorn服务
|
|
uvicorn.run(
|
|
app='main:create_app',
|
|
host=settings.SERVER_HOST,
|
|
port=settings.SERVER_PORT,
|
|
reload=settings.RELOAD,
|
|
log_config=settings.LOGGING_CONFIG,
|
|
workers=settings.WORKERS,
|
|
limit_concurrency=settings.LIMIT_CONCURRENCY,
|
|
backlog=settings.BACKLOG,
|
|
limit_max_requests=settings.LIMIT_MAX_REQUESTS,
|
|
timeout_keep_alive=settings.TIMEOUT_KEEP_ALIVE,
|
|
lifespan=settings.LIFESPAN,
|
|
factory=settings.FACTORY,
|
|
)
|
|
|
|
|
|
"""
|
|
初始化数据命令。
|
|
|
|
该命令用于初始化项目数据。通过调用 `InitializeData` 类的 `run` 方法，并使用 `asyncio.run` 来异步执行数据初始化过程。
|
|
"""
|
|
|
|
|
|
@shell_app.command()
|
|
def init():
|
|
import asyncio
|
|
from app.scripts.initialize import InitializeData
|
|
# 初始化数据
|
|
data = InitializeData()
|
|
# 使用asyncio.run来运行异步函数
|
|
asyncio.run(data.run())
|
|
|
|
# @shell_app.command()
|
|
# def migrate(env: Environment = Environment.pro):
|
|
# """
|
|
# 将模型迁移到数据库，更新数据库表结构
|
|
|
|
# :param env: 数据库环境
|
|
# """
|
|
# print("开始更新数据库表")
|
|
# data = InitializeData()
|
|
# data.migrate_model(env)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 启动服务
|
|
# 方式一：
|
|
# uvicorn main:create_app --host 0.0.0.0 --port 8000 --factory --workers 4
|
|
# 方式二：
|
|
# python main.py run # 启动服务
|
|
# python main.py init # 初始化数据
|
|
# 方式三：
|
|
# gunicorn -c gunicorn.py main:create_app
|
|
shell_app()
|