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:
"""
说明:
检测是否会被自动更新刷新权限