fix
This commit is contained in:
@ -45,4 +45,4 @@ def update_db():
|
||||
alembic_cfg.set_main_option('sqlalchemy.url', f"sqlite:///{db_location}")
|
||||
upgrade(alembic_cfg, 'head')
|
||||
except Exception as e:
|
||||
logger.error(f'数据库更新失败:{e}')
|
||||
logger.error(f'数据库更新失败:{str(e)}')
|
||||
|
@ -16,7 +16,12 @@ def db_persist(func):
|
||||
|
||||
def wrapper(*args, **kwargs):
|
||||
with DBLock:
|
||||
db: Session = kwargs.get("db") or args[1]
|
||||
db: Session = kwargs.get("db")
|
||||
if not db:
|
||||
for arg in args:
|
||||
if isinstance(arg, Session):
|
||||
db = arg
|
||||
break
|
||||
try:
|
||||
if db:
|
||||
db.close()
|
||||
|
Reference in New Issue
Block a user