mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-21 21:58:04 +00:00
✨ 支持BOT管理员获取用户自动签到信息
This commit is contained in:
parent
b678695f94
commit
77ee577f7a
@ -37,6 +37,14 @@ class SignRepository:
|
||||
results = await session.exec(statement)
|
||||
return sign[0] if (sign := results.first()) else None
|
||||
|
||||
async def get_by_chat_id(self, chat_id: int) -> Optional[List[Sign]]:
|
||||
async with self.mysql.Session() as session:
|
||||
session = cast(AsyncSession, session)
|
||||
statement = select(Sign).where(Sign.chat_id == chat_id)
|
||||
results = await session.exec(statement)
|
||||
signs = results.all()
|
||||
return [sign[0] for sign in signs]
|
||||
|
||||
async def get_all(self) -> List[Sign]:
|
||||
async with self.mysql.Session() as session:
|
||||
query = select(Sign)
|
||||
|
@ -20,3 +20,6 @@ class SignServices:
|
||||
|
||||
async def get_by_user_id(self, user_id: int):
|
||||
return await self._repository.get_by_user_id(user_id)
|
||||
|
||||
async def get_by_chat_id(self, chat_id: int):
|
||||
return await self._repository.get_by_chat_id(chat_id)
|
||||
|
@ -8,6 +8,7 @@ from telegram.ext import CommandHandler, CallbackContext
|
||||
from core.cookies import CookiesService
|
||||
from core.cookies.error import CookiesNotFoundError
|
||||
from core.plugin import Plugin, handler
|
||||
from core.sign import SignServices
|
||||
from core.user import UserService
|
||||
from core.user.error import UserNotFoundError
|
||||
from modules.apihelper.gacha_log import GachaLog
|
||||
@ -19,15 +20,23 @@ from utils.models.base import RegionEnum
|
||||
|
||||
|
||||
class GetChat(Plugin):
|
||||
def __init__(self, user_service: UserService = None, cookies_service: CookiesService = None):
|
||||
def __init__(
|
||||
self,
|
||||
user_service: UserService = None,
|
||||
cookies_service: CookiesService = None,
|
||||
sign_service: SignServices = None,
|
||||
):
|
||||
self.cookies_service = cookies_service
|
||||
self.user_service = user_service
|
||||
self.sign_service = sign_service
|
||||
|
||||
@staticmethod
|
||||
def parse_group_chat(chat: Chat, admins: List[ChatMember]) -> str:
|
||||
async def parse_group_chat(self, chat: Chat, admins: List[ChatMember]) -> str:
|
||||
text = f"群 ID:<code>{chat.id}</code>\n" f"群名称:<code>{chat.title}</code>\n"
|
||||
if chat.username:
|
||||
text += f"群用户名:<code>{chat.username}</code>\n"
|
||||
text += f"群用户名:@{chat.username}\n"
|
||||
sign_info = await self.sign_service.get_by_chat_id(chat.id)
|
||||
if sign_info:
|
||||
text += f"自动签到推送人数:<code>{len(sign_info)}</code>\n"
|
||||
if chat.description:
|
||||
text += f"群简介:<code>{chat.description}</code>\n"
|
||||
if admins:
|
||||
@ -72,6 +81,17 @@ class GetChat(Plugin):
|
||||
except CookiesNotFoundError:
|
||||
temp = "UID 绑定"
|
||||
text += f"<code>{temp}</code>\n" f"游戏 ID:<code>{uid}</code>"
|
||||
sign_info = await self.sign_service.get_by_user_id(chat.id)
|
||||
if sign_info is not None:
|
||||
text += (
|
||||
f"\n自动签到:已开启"
|
||||
f"\n推送会话:<code>{sign_info.chat_id}</code>"
|
||||
f"\n开启时间:<code>{sign_info.time_created}</code>"
|
||||
f"\n更新时间:<code>{sign_info.time_updated}</code>"
|
||||
f"\n签到状态:<code>{sign_info.status.name}</code>"
|
||||
)
|
||||
else:
|
||||
text += f"\n自动签到:未开启"
|
||||
with contextlib.suppress(Exception):
|
||||
gacha_log, status = await GachaLog.load_history_info(str(chat.id), str(uid))
|
||||
if status:
|
||||
@ -102,7 +122,7 @@ class GetChat(Plugin):
|
||||
chat = await message.get_bot().get_chat(args[0])
|
||||
if chat_id < 0:
|
||||
admins = await chat.get_administrators() if chat_id < 0 else None
|
||||
text = self.parse_group_chat(chat, admins)
|
||||
text = await self.parse_group_chat(chat, admins)
|
||||
else:
|
||||
text = await self.parse_private_chat(chat)
|
||||
await message.reply_text(text, parse_mode="HTML")
|
||||
|
Loading…
Reference in New Issue
Block a user