Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

🚨 auto fix by pre-commit hooks

Browse Source
This commit is contained in:
molanp 2025-07-14 18:07:03 +08:00
parent 2e664c387e
commit 05ca69bd4c
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
zhenxun/models/group_member_info.py
View File

@ -26,6 +26,7 @@ class GroupInfoUser(Model):
table = "group_info_users" table = "group_info_users"
table_description = "群员信息数据表" table_description = "群员信息数据表"
unique_together = ("user_id", "group_id") unique_together = ("user_id", "group_id")
_unicode_safe_fields = ["user_name", "nickname"] # noqa: RUF012 _unicode_safe_fields = ["user_name", "nickname"] # noqa: RUF012
@classmethod @classmethod

5
zhenxun/services/db_context.py
View File

@ -3,7 +3,8 @@ import re
import nonebot import nonebot
from nonebot.utils import is_coroutine_callable 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.connection import connections
from tortoise.models import Model as Model_ from tortoise.models import Model as Model_
@ -76,7 +77,7 @@ class UnicodeSafeMixin(Model_):
return instance return instance
@classmethod @classmethod
def filter(cls, *args, **kwargs): # pyright: ignore[reportIncompatibleMethodOverride] def filter(cls, *args, **kwargs): # pyright: ignore[reportIncompatibleMethodOverride]
for field in cls._unicode_safe_fields: for field in cls._unicode_safe_fields:
if field in kwargs and isinstance(kwargs[field], str): if field in kwargs and isinstance(kwargs[field], str):
kwargs[field] = unicode_escape(kwargs[field]) kwargs[field] = unicode_escape(kwargs[field])