From 05ca69bd4cf731d47509fb2cc75671177bbc2aba Mon Sep 17 00:00:00 2001 From: molanp <104612722+molanp@users.noreply.github.com> Date: Mon, 14 Jul 2025 18:07:03 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A8=20auto=20fix=20by=20pre-commit=20h?= =?UTF-8?q?ooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- zhenxun/models/group_member_info.py | 1 + zhenxun/services/db_context.py | 5 +++-- 2 files changed, 4 insertions(+), 2 deletions(-) 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])