zfc/guzhi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
mizhexiaoxiao 2024-09-23 10:49:22 +08:00
parent 060a7695c8
commit 6c7cecc7a3
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app/core/init_app.py
View File

@ -1,3 +1,5 @@
import shutil
from aerich import Command from aerich import Command
from fastapi import FastAPI from fastapi import FastAPI
from fastapi.middleware import Middleware from fastapi.middleware import Middleware
@ -19,6 +21,7 @@ from app.core.exceptions import (
ResponseValidationError, ResponseValidationError,
ResponseValidationHandle, ResponseValidationHandle,
) )
from app.log import logger
from app.models.admin import Api, Menu, Role from app.models.admin import Api, Menu, Role
from app.schemas.menus import MenuType from app.schemas.menus import MenuType
from app.settings.config import settings from app.settings.config import settings
@ -186,7 +189,13 @@ async def init_db():
pass pass
await command.init() await command.init()
try:
await command.migrate() await command.migrate()
except AttributeError:
logger.warning("unable to retrieve model history from database, model history will be created from scratch")
shutil.rmtree("migrations")
await command.init_db(safe=True)
await command.upgrade(run_in_transaction=True) await command.upgrade(run_in_transaction=True)