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

Merge pull request #52 from mizhexiaoxiao/dev

Browse Source 
Update
This commit is contained in:
mizhexiaoxiao 2024-09-23 10:49:52 +08:00 committed by GitHub
commit 28492e0fd9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
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()
await command.migrate() try:
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)