diff --git a/zhenxun/models/group_member_info.py b/zhenxun/models/group_member_info.py index 8248bd71..0ddee496 100644 --- a/zhenxun/models/group_member_info.py +++ b/zhenxun/models/group_member_info.py @@ -26,6 +26,7 @@ class GroupInfoUser(Model): table = "group_info_users" table_description = "群员信息数据表" unique_together = ("user_id", "group_id") + _unicode_safe_fields = ["user_name", "nickname"] # noqa: RUF012 @classmethod diff --git a/zhenxun/services/db_context.py b/zhenxun/services/db_context.py index 3a8e5063..605db41b 100644 --- a/zhenxun/services/db_context.py +++ b/zhenxun/services/db_context.py @@ -3,7 +3,8 @@ import re import nonebot from nonebot.utils import is_coroutine_callable -from tortoise import BaseDBAsyncClient, Tortoise +from tortoise import Tortoise +from tortoise.backends.base.client import BaseDBAsyncClient from tortoise.connection import connections from tortoise.models import Model as Model_ @@ -76,7 +77,7 @@ class UnicodeSafeMixin(Model_): return instance @classmethod - def filter(cls, *args, **kwargs): # pyright: ignore[reportIncompatibleMethodOverride] + def filter(cls, *args, **kwargs): # pyright: ignore[reportIncompatibleMethodOverride] for field in cls._unicode_safe_fields: if field in kwargs and isinstance(kwargs[field], str): kwargs[field] = unicode_escape(kwargs[field])