执行_run_script添加try

This commit is contained in:
HibiKier 2023-02-19 12:29:51 +08:00
parent c97f1bd750
commit 876cfb421a
3 changed files with 8 additions and 4 deletions

View File

@ -500,4 +500,4 @@ class WordBank(Model):
@classmethod
async def _run_script(cls):
await cls.raw("ALTER TABLE word_bank2 ADD to_me varchar(255);")
"""允许 user_join_time 为空"""
"""添加 to_me 字段"""

View File

@ -27,7 +27,7 @@ class Model(Model_):
MODELS.append(cls.__module__)
if func := getattr(cls, "_run_script", None):
SCRIPT_METHOD.append(func)
SCRIPT_METHOD.append((cls.__module__, func))
class TestSQL(Model):
@ -53,8 +53,11 @@ async def init():
except Exception as e:
raise Exception(f"数据库连接错误.... {type(e)}: {e}")
if SCRIPT_METHOD:
for func in SCRIPT_METHOD:
await func()
for module, func in SCRIPT_METHOD:
try:
await func()
except Exception as e:
logger.debug(f"{module} 执行SQL", e=e)
async def disconnect():

View File

@ -109,6 +109,7 @@ class logger:
user_id: Optional[int] = None,
group_id: Optional[int] = None,
target: Optional[Any] = None,
e: Optional[Exception] = None,
):
template = cls.__parser_template(info, command, user_id, group_id, target)
logger_.opt(colors=True).debug(template)