109 lines
3.3 KiB
Python
109 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Creaet Time : 2021/10/19 15:47
|
||
# @File : main.py
|
||
# @IDE : PyCharm
|
||
# @desc : 主程序入口
|
||
|
||
"""
|
||
FastApi 更新文档：https://github.com/tiangolo/fastapi/releases
|
||
FastApi Github：https://github.com/tiangolo/fastapi
|
||
"""
|
||
|
||
from fastapi import FastAPI
|
||
import uvicorn
|
||
from starlette.middleware.cors import CORSMiddleware
|
||
from application import settings
|
||
from application import urls
|
||
from starlette.staticfiles import StaticFiles # 依赖安装：pip install aiofiles
|
||
import importlib
|
||
from core.logger import logger
|
||
from core.exception import register_exception
|
||
import typer
|
||
from scripts.initialize.initialize import InitializeData
|
||
import asyncio
|
||
|
||
|
||
shell_app = typer.Typer()
|
||
|
||
|
||
def create_app():
|
||
"""
|
||
启动项目
|
||
|
||
docs_url：配置交互文档的路由地址，如果禁用则为None，默认为 /docs
|
||
redoc_url： 配置 Redoc 文档的路由地址，如果禁用则为None，默认为 /redoc
|
||
openapi_url：配置接口文件json数据文件路由地址，如果禁用则为None，默认为/openapi.json
|
||
"""
|
||
app = FastAPI(
|
||
title="KInit",
|
||
description="本项目基于Fastapi与Vue3+Typescript+Vite3+element-plus的基础项目 前端基于vue-element-plus-admin框架开发",
|
||
version="1.0.0"
|
||
)
|
||
|
||
def import_module(modules: list, desc: str):
|
||
for module in modules:
|
||
if not module:
|
||
continue
|
||
try:
|
||
# 动态导入模块
|
||
module_pag = importlib.import_module(module[0:module.rindex(".")])
|
||
getattr(module_pag, module[module.rindex(".") + 1:])(app)
|
||
except ModuleNotFoundError:
|
||
logger.error(f"AttributeError：导入{desc}失败，未找到该模块：{module}")
|
||
except AttributeError:
|
||
logger.error(f"ModuleNotFoundError：导入{desc}失败，未找到该模块下的方法：{module}")
|
||
|
||
import_module(settings.MIDDLEWARES, "中间件")
|
||
import_module(settings.EVENTS, "全局事件")
|
||
# 全局异常捕捉处理
|
||
register_exception(app)
|
||
# 跨域解决
|
||
if settings.CORS_ORIGIN_ENABLE:
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=settings.ALLOW_ORIGINS,
|
||
allow_credentials=settings.ALLOW_CREDENTIALS,
|
||
allow_methods=settings.ALLOW_METHODS,
|
||
allow_headers=settings.ALLOW_HEADERS)
|
||
# 挂在静态目录
|
||
if settings.STATIC_ENABLE:
|
||
app.mount(settings.STATIC_URL, app=StaticFiles(directory=settings.STATIC_ROOT))
|
||
if settings.TEMP_ENABLE:
|
||
app.mount(settings.TEMP_URL, app=StaticFiles(directory=settings.TEMP_DIR))
|
||
# 引入应用中的路由
|
||
for url in urls.urlpatterns:
|
||
app.include_router(url["ApiRouter"], prefix=url["prefix"], tags=url["tags"])
|
||
return app
|
||
|
||
|
||
@shell_app.command()
|
||
def run():
|
||
"""
|
||
启动项目
|
||
"""
|
||
uvicorn.run(app='main:create_app', host="0.0.0.0", port=9000)
|
||
|
||
|
||
@shell_app.command()
|
||
def init():
|
||
"""
|
||
初始化数据
|
||
"""
|
||
print("开始初始化数据")
|
||
data = InitializeData()
|
||
asyncio.run(data.run())
|
||
|
||
|
||
@shell_app.command()
|
||
def migrate():
|
||
"""
|
||
将模型迁移到数据库，更新数据库表结构
|
||
"""
|
||
print("开始更新数据库表")
|
||
InitializeData().migrate_model()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
shell_app()
|