From 8c65caf35f68522bc0c8b95b5c27396f56bb8d98 Mon Sep 17 00:00:00 2001 From: yajiwa <839790708@qq.com> Date: Sat, 22 Apr 2023 21:14:51 +0800 Subject: [PATCH] =?UTF-8?q?:ambulance:=20=E8=AF=AD=E6=B3=95=E9=80=82?= =?UTF-8?q?=E9=85=8D=E5=88=B0python3.8=E4=BB=A5=E4=B8=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/bag_user.py | 16 ++++++++-------- models/ban_user.py | 12 ++++++------ models/chat_history.py | 4 ++-- models/friend_user.py | 8 ++++---- models/goods_info.py | 6 +++--- models/group_member_info.py | 12 ++++++------ models/level_user.py | 12 ++++++------ 7 files changed, 35 insertions(+), 35 deletions(-) diff --git a/models/bag_user.py b/models/bag_user.py index 07f219dc..283a49f2 100755 --- a/models/bag_user.py +++ b/models/bag_user.py @@ -1,4 +1,4 @@ -from typing import Dict +from typing import Dict, Union from tortoise import fields @@ -34,7 +34,7 @@ class BagUser(Model): unique_together = ("user_id", "group_id") @classmethod - async def get_user_total_gold(cls, user_id: str | int, group_id: str | int) -> str: + async def get_user_total_gold(cls, user_id: Union[int, str], group_id: Union[int, str]) -> str: """ 说明: 获取金币概况 @@ -50,7 +50,7 @@ class BagUser(Model): ) @classmethod - async def get_gold(cls, user_id: str | int, group_id: str | int) -> int: + async def get_gold(cls, user_id: Union[int, str], group_id: Union[int, str]) -> int: """ 说明: 获取当前金币 @@ -63,7 +63,7 @@ class BagUser(Model): @classmethod async def get_property( - cls, user_id: str | int, group_id: str | int, only_active: bool = False + cls, user_id: Union[int, str], group_id: Union[int, str], only_active: bool = False ) -> Dict[str, int]: """ 说明: @@ -87,7 +87,7 @@ class BagUser(Model): return user.property @classmethod - async def add_gold(cls, user_id: str | int, group_id: str | int, num: int): + async def add_gold(cls, user_id: Union[int, str], group_id: Union[int, str], num: int): """ 说明: 增加金币 @@ -103,7 +103,7 @@ class BagUser(Model): await user.save(update_fields=["gold", "get_today_gold", "get_total_gold"]) @classmethod - async def spend_gold(cls, user_id: str | int, group_id: str | int, num: int): + async def spend_gold(cls, user_id: Union[int, str], group_id: Union[int, str], num: int): """ 说明: 花费金币 @@ -119,7 +119,7 @@ class BagUser(Model): await user.save(update_fields=["gold", "spend_total_gold", "spend_today_gold"]) @classmethod - async def add_property(cls, user_id: str | int, group_id: str | int, name: str, num: int = 1): + async def add_property(cls, user_id: Union[int, str], group_id: Union[int, str], name: str, num: int = 1): """ 说明: 增加道具 @@ -139,7 +139,7 @@ class BagUser(Model): @classmethod async def delete_property( - cls, user_id: str | int, group_id: str | int, name: str, num: int = 1 + cls, user_id: Union[int, str], group_id: Union[int, str], name: str, num: int = 1 ) -> bool: """ 说明: diff --git a/models/ban_user.py b/models/ban_user.py index be2269b9..e75c9552 100755 --- a/models/ban_user.py +++ b/models/ban_user.py @@ -23,7 +23,7 @@ class BanUser(Model): table_description = ".ban/b了 封禁人员数据表" @classmethod - async def check_ban_level(cls, user_id: str | int, level: int) -> bool: + async def check_ban_level(cls, user_id: Union[int, str], level: int) -> bool: """ 说明: 检测ban掉目标的用户与unban用户的权限等级大小 @@ -41,7 +41,7 @@ class BanUser(Model): return False @classmethod - async def check_ban_time(cls, user_id: str | int) -> Union[str, int]: + async def check_ban_time(cls, user_id: Union[int, str]) -> Union[str, int]: """ 说明: 检测用户被ban时长 @@ -61,7 +61,7 @@ class BanUser(Model): return "" @classmethod - async def is_ban(cls, user_id: str | int) -> bool: + async def is_ban(cls, user_id: Union[int, str]) -> bool: """ 说明: 判断用户是否被ban @@ -76,7 +76,7 @@ class BanUser(Model): return False @classmethod - async def is_super_ban(cls, user_id: str | int) -> bool: + async def is_super_ban(cls, user_id: Union[int, str]) -> bool: """ 说明: 判断用户是否被超级用户ban / b了 @@ -90,7 +90,7 @@ class BanUser(Model): return False @classmethod - async def ban(cls, user_id: str | int, ban_level: int, duration: int): + async def ban(cls, user_id: Union[int, str], ban_level: int, duration: int): """ 说明: ban掉目标用户 @@ -110,7 +110,7 @@ class BanUser(Model): ) @classmethod - async def unban(cls, user_id: str | int) -> bool: + async def unban(cls, user_id: Union[int, str]) -> bool: """ 说明: unban用户 diff --git a/models/chat_history.py b/models/chat_history.py index 52597b2a..dcdaf570 100644 --- a/models/chat_history.py +++ b/models/chat_history.py @@ -29,7 +29,7 @@ class ChatHistory(Model): @classmethod async def get_group_msg_rank( cls, - gid: str | int, + gid: Union[int, str], limit: int = 10, order: str = "DESC", date_scope: Optional[Tuple[datetime, datetime]] = None, @@ -56,7 +56,7 @@ class ChatHistory(Model): ) # type: ignore @classmethod - async def get_group_first_msg_datetime(cls, group_id: str | int) -> Optional[datetime]: + async def get_group_first_msg_datetime(cls, group_id: Union[int, str]) -> Optional[datetime]: """ 说明: 获取群第一条记录消息时间 diff --git a/models/friend_user.py b/models/friend_user.py index 90cf279b..9de696c9 100755 --- a/models/friend_user.py +++ b/models/friend_user.py @@ -1,5 +1,5 @@ from tortoise import fields - +from typing import Union from configs.config import Config from services.db_context import Model @@ -20,7 +20,7 @@ class FriendUser(Model): table_description = "好友信息数据表" @classmethod - async def get_user_name(cls, user_id: str | int) -> str: + async def get_user_name(cls, user_id: Union[int, str]) -> str: """ 说明: 获取好友用户名称 @@ -32,7 +32,7 @@ class FriendUser(Model): return "" @classmethod - async def get_user_nickname(cls, user_id: str | int) -> str: + async def get_user_nickname(cls, user_id: Union[int, str]) -> str: """ 说明: 获取用户昵称 @@ -49,7 +49,7 @@ class FriendUser(Model): return "" @classmethod - async def set_user_nickname(cls, user_id: str | int, nickname: str): + async def set_user_nickname(cls, user_id: Union[int, str], nickname: str): """ 说明: 设置用户昵称 diff --git a/models/goods_info.py b/models/goods_info.py index bd73e1c3..d0afcef2 100644 --- a/models/goods_info.py +++ b/models/goods_info.py @@ -1,4 +1,4 @@ -from typing import Dict, List, Optional, Tuple +from typing import Dict, List, Optional, Tuple, Union from tortoise import fields @@ -145,7 +145,7 @@ class GoodsInfo(Model): @classmethod async def add_user_daily_purchase( - cls, goods: "GoodsInfo", user_id_: int, group_id_: int, num: int = 1 + cls, goods: "GoodsInfo", user_id_: Union[int, str], group_id_: Union[int, str], num: int = 1 ): """ 说明: @@ -168,7 +168,7 @@ class GoodsInfo(Model): @classmethod async def check_user_daily_purchase( - cls, goods: "GoodsInfo", user_id_: int, group_id_: int, num: int = 1 + cls, goods: "GoodsInfo", user_id_: Union[int, str], group_id_: Union[int, str], num: int = 1 ) -> Tuple[bool, int]: """ 说明: diff --git a/models/group_member_info.py b/models/group_member_info.py index 8e6527b8..fc43a98d 100755 --- a/models/group_member_info.py +++ b/models/group_member_info.py @@ -1,5 +1,5 @@ from datetime import datetime -from typing import List, Optional, Set +from typing import List, Optional, Set, Union from tortoise import fields @@ -31,7 +31,7 @@ class GroupInfoUser(Model): unique_together = ("user_id", "group_id") @classmethod - async def get_group_member_id_list(cls, group_id: str | int) -> Set[int]: + async def get_group_member_id_list(cls, group_id: Union[int, str]) -> Set[int]: """ 说明: 获取该群所有用户id @@ -43,7 +43,7 @@ class GroupInfoUser(Model): ) # type: ignore @classmethod - async def set_user_nickname(cls, user_id: str | int, group_id: str | int, nickname: str): + async def set_user_nickname(cls, user_id: Union[int, str], group_id: Union[int, str], nickname: str): """ 说明: 设置群员在该群内的昵称 @@ -59,7 +59,7 @@ class GroupInfoUser(Model): ) @classmethod - async def get_user_all_group(cls, user_id: str | int) -> List[int]: + async def get_user_all_group(cls, user_id: Union[int, str]) -> List[int]: """ 说明: 获取该用户所在的所有群聊 @@ -71,7 +71,7 @@ class GroupInfoUser(Model): ) # type: ignore @classmethod - async def get_user_nickname(cls, user_id: str | int, group_id: str | int) -> str: + async def get_user_nickname(cls, user_id: Union[int, str], group_id: Union[int, str]) -> str: """ 说明: 获取用户在该群的昵称 @@ -90,7 +90,7 @@ class GroupInfoUser(Model): return "" @classmethod - async def get_group_member_uid(cls, user_id: str | int, group_id: str | int) -> Optional[int]: + async def get_group_member_uid(cls, user_id: Union[int, str], group_id: Union[int, str]) -> Optional[int]: logger.debug( f"GroupInfoUser 尝试获取 用户[{user_id}] 群聊[{group_id}] UID" ) diff --git a/models/level_user.py b/models/level_user.py index c74e41e4..0c67d8ac 100755 --- a/models/level_user.py +++ b/models/level_user.py @@ -1,7 +1,7 @@ from tortoise import fields from services.db_context import Model - +from typing import Union class LevelUser(Model): @@ -22,7 +22,7 @@ class LevelUser(Model): unique_together = ("user_id", "group_id") @classmethod - async def get_user_level(cls, user_id: str | int, group_id: str | int) -> int: + async def get_user_level(cls, user_id: Union[int, str], group_id: Union[int, str]) -> int: """ 说明: 获取用户在群内的等级 @@ -36,7 +36,7 @@ class LevelUser(Model): @classmethod async def set_level( - cls, user_id: str | int, group_id: str | int, level: int, group_flag: int = 0 + cls, user_id: Union[int, str], group_id: Union[int, str], level: int, group_flag: int = 0 ): """ 说明: @@ -54,7 +54,7 @@ class LevelUser(Model): ) @classmethod - async def delete_level(cls, user_id: str | int, group_id: str | int) -> bool: + async def delete_level(cls, user_id: Union[int, str], group_id: Union[int, str]) -> bool: """ 说明: 删除用户权限 @@ -68,7 +68,7 @@ class LevelUser(Model): return False @classmethod - async def check_level(cls, user_id: str | int, group_id: str | int, level: int) -> bool: + async def check_level(cls, user_id: Union[int, str], group_id: Union[int, str], level: int) -> bool: """ 说明: 检查用户权限等级是否大于 level @@ -87,7 +87,7 @@ class LevelUser(Model): return False @classmethod - async def is_group_flag(cls, user_id: str | int, group_id: str | int) -> bool: + async def is_group_flag(cls, user_id: Union[int, str], group_id: Union[int, str]) -> bool: """ 说明: 检测是否会被自动更新刷新权限