mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
✨ 添加玩家查询插件
This commit is contained in:
洛水居室
parent
5e93a31e12
commit
ea5d9ee1d2
163
plugins/uid.py
Normal file
163
plugins/uid.py
Normal file
@ -0,0 +1,163 @@
|
||||
import os
|
||||
import random
|
||||
|
||||
from genshin import DataNotPublic, GenshinException, Client
|
||||
from telegram import Update
|
||||
from telegram.constants import ChatAction
|
||||
from telegram.ext import CallbackContext, CommandHandler, MessageHandler, ConversationHandler, filters
|
||||
|
||||
from app.cookies.service import CookiesService
|
||||
from app.template import TemplateService
|
||||
from app.user import UserService
|
||||
from app.user.repositories import UserNotFoundError
|
||||
from logger import Log
|
||||
from plugins.base import BasePlugins
|
||||
from utils.app.inject import inject
|
||||
from utils.decorators.error import error_callable
|
||||
from utils.decorators.restricts import restricts
|
||||
from utils.helpers import url_to_file, get_genshin_client
|
||||
from utils.plugins.manager import listener_plugins_class
|
||||
|
||||
|
||||
@listener_plugins_class()
|
||||
class Uid(BasePlugins):
|
||||
"""玩家查询"""
|
||||
|
||||
COMMAND_RESULT, = range(10200, 10201)
|
||||
|
||||
@inject
|
||||
def __init__(self, user_service: UserService, cookies_service: CookiesService, template_service: TemplateService):
|
||||
self.template_service = template_service
|
||||
self.cookies_service = cookies_service
|
||||
self.user_service = user_service
|
||||
self.current_dir = os.getcwd()
|
||||
|
||||
@classmethod
|
||||
def create_handlers(cls):
|
||||
uid = cls()
|
||||
return [CommandHandler('uid', uid.command_start, block=True),
|
||||
MessageHandler(filters.Regex(r"^玩家查询(.*)"), uid.command_start, block=True)]
|
||||
|
||||
async def _start_get_user_info(self, client: Client, uid: int = -1) -> bytes:
|
||||
if uid == -1:
|
||||
uid = client.uid
|
||||
try:
|
||||
user_info = await client.get_genshin_user(uid)
|
||||
except GenshinException as error:
|
||||
Log.warning("get_record_card请求失败 \n", error)
|
||||
raise error
|
||||
if user_info.teapot is None:
|
||||
raise ValueError("洞庭湖未解锁")
|
||||
try:
|
||||
# 查询的UID如果是自己的，会返回DataNotPublic，自己查不了自己可还行......
|
||||
if uid > 0:
|
||||
record_card_info = await client.get_record_card(uid)
|
||||
else:
|
||||
record_card_info = await client.get_record_card()
|
||||
except DataNotPublic as error:
|
||||
Log.warning("get_record_card请求失败 查询的用户数据未公开 \n", error)
|
||||
nickname = uid
|
||||
user_uid = ""
|
||||
except GenshinException as error:
|
||||
Log.warning("get_record_card请求失败 \n", error)
|
||||
raise error
|
||||
else:
|
||||
nickname = record_card_info.nickname
|
||||
user_uid = record_card_info.uid
|
||||
user_avatar = user_info.characters[0].icon
|
||||
user_data = {
|
||||
"name": nickname,
|
||||
"uid": user_uid,
|
||||
"user_avatar": await url_to_file(user_avatar),
|
||||
"action_day_number": user_info.stats.days_active,
|
||||
"achievement_number": user_info.stats.achievements,
|
||||
"avatar_number": user_info.stats.characters,
|
||||
"spiral_abyss": user_info.stats.spiral_abyss,
|
||||
"way_point_number": user_info.stats.unlocked_waypoints,
|
||||
"domain_number": user_info.stats.unlocked_domains,
|
||||
"luxurious_number": user_info.stats.luxurious_chests,
|
||||
"precious_chest_number": user_info.stats.precious_chests,
|
||||
"exquisite_chest_number": user_info.stats.exquisite_chests,
|
||||
"common_chest_number": user_info.stats.common_chests,
|
||||
"magic_chest_number": user_info.stats.remarkable_chests,
|
||||
"anemoculus_number": user_info.stats.anemoculi,
|
||||
"geoculus_number": user_info.stats.geoculi,
|
||||
"electroculus_number": user_info.stats.electroculi,
|
||||
"world_exploration_list": [],
|
||||
"teapot_level": user_info.teapot.level,
|
||||
"teapot_comfort_num": user_info.teapot.comfort,
|
||||
"teapot_item_num": user_info.teapot.items,
|
||||
"teapot_visit_num": user_info.teapot.visitors,
|
||||
"teapot_list": []
|
||||
}
|
||||
for exploration in user_info.explorations:
|
||||
exploration_data = {
|
||||
"name": exploration.name,
|
||||
"exploration_percentage": exploration.explored,
|
||||
"offerings": [],
|
||||
"icon": await url_to_file(exploration.icon)
|
||||
}
|
||||
for offering in exploration.offerings:
|
||||
# 修复上游奇怪的问题
|
||||
offering_name = offering.name
|
||||
if offering_name == "Reputation":
|
||||
offering_name = "声望等级"
|
||||
offering_data = {
|
||||
"data": f"{offering_name}：{offering.level}级"
|
||||
}
|
||||
exploration_data["offerings"].append(offering_data)
|
||||
user_data["world_exploration_list"].append(exploration_data)
|
||||
for teapot in user_info.teapot.realms:
|
||||
teapot_icon = teapot.icon
|
||||
# 修复 国际服绘绮庭 图标 地址请求 为404
|
||||
if "UI_HomeworldModule_4_Pic.png" in teapot_icon:
|
||||
teapot_icon = "https://upload-bbs.mihoyo.com/game_record/genshin/home/UI_HomeworldModule_4_Pic.png"
|
||||
teapot_data = {
|
||||
"icon": await url_to_file(teapot_icon),
|
||||
"name": teapot.name
|
||||
}
|
||||
user_data["teapot_list"].append(teapot_data)
|
||||
background_image = random.choice(os.listdir(f"{self.current_dir}/resources/background/vertical"))
|
||||
user_data[
|
||||
"background_image"] = f"file://{self.current_dir}/resources/background/vertical/{background_image}"
|
||||
png_data = await self.template_service.render('genshin/info', "info.html", user_data,
|
||||
{"width": 1024, "height": 1024})
|
||||
return png_data
|
||||
|
||||
@error_callable
|
||||
@restricts(return_data=ConversationHandler.END)
|
||||
async def command_start(self, update: Update, context: CallbackContext) -> None:
|
||||
user = update.effective_user
|
||||
message = update.message
|
||||
Log.info(f"用户 {user.full_name}[{user.id}] 查询游戏用户命令请求")
|
||||
uid: int = -1
|
||||
try:
|
||||
args = context.args
|
||||
if args is not None:
|
||||
if len(args) >= 1:
|
||||
uid = int(args[0])
|
||||
except ValueError as error:
|
||||
Log.error("获取 uid 发生错误！ 错误信息为", error)
|
||||
await message.reply_text("输入错误")
|
||||
return ConversationHandler.END
|
||||
try:
|
||||
client = await get_genshin_client(user.id, self.user_service, self.cookies_service)
|
||||
png_data = await self._start_get_user_info(client, uid)
|
||||
except UserNotFoundError:
|
||||
reply_message = await message.reply_text("未查询到账号信息，请先私聊派蒙绑定账号")
|
||||
if filters.ChatType.GROUPS.filter(message):
|
||||
self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id, 30)
|
||||
self._add_delete_message_job(context, message.chat_id, message.message_id, 30)
|
||||
return
|
||||
except ValueError as exc:
|
||||
if "洞庭湖未解锁" in str(exc):
|
||||
await message.reply_text("角色尘歌壶未解锁 如果想要查看具体数据 嗯...... 咕咕咕~")
|
||||
return ConversationHandler.END
|
||||
else:
|
||||
raise exc
|
||||
except AttributeError as exc:
|
||||
Log.warning("角色数据有误", exc)
|
||||
await message.reply_text("角色数据有误 估计是派蒙晕了")
|
||||
return ConversationHandler.END
|
||||
await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
|
||||
await message.reply_photo(png_data, filename=f"{client.uid}.png", allow_sending_without_reply=True)
|
||||
Loading…
Reference in New Issue
Block a user