PaiGram/plugins/uid.py

import os
import random

import genshin
from genshin import DataNotPublic, GenshinException
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.constants import ChatAction
from telegram.ext import CallbackContext, CommandHandler, MessageHandler, ConversationHandler, filters, \
    CallbackQueryHandler

from logger import Log
from manager import listener_plugins_class
from model.base import ServiceEnum
from model.helpers import url_to_file
from plugins.base import BasePlugins, restricts
from plugins.errorhandler import conversation_error_handler
from service import BaseService
from service.base import UserInfoData


class UidCommandData:
    user_info: UserInfoData = UserInfoData()


@listener_plugins_class(need_service=True)
class Uid(BasePlugins):
    """
    玩家查询
    """

    COMMAND_RESULT, = range(10200, 10201)

    def __init__(self, service: BaseService):
        self.service = service
        self.current_dir = os.getcwd()

    @classmethod
    def create_handlers(cls, service: BaseService):
        uid = cls(service)
        uid_handler = ConversationHandler(
            entry_points=[CommandHandler('uid', uid.command_start, block=True),
                          MessageHandler(filters.Regex(r"^玩家查询(.*)"), uid.command_start, block=True)],
            states={
                uid.COMMAND_RESULT: [CallbackQueryHandler(uid.command_result, block=True)]
            },
            fallbacks=[CommandHandler('cancel', uid.cancel, block=True)]
        )
        return [uid_handler]

    async def _start_get_user_info(self, user_info_data: UserInfoData, service: ServiceEnum, uid: int = -1) -> bytes:
        if service == ServiceEnum.HYPERION:
            client = genshin.ChineseClient(cookies=user_info_data.mihoyo_cookie)
            if uid <= 0:
                _uid = user_info_data.mihoyo_game_uid
            else:
                _uid = uid
        else:
            client = genshin.GenshinClient(cookies=user_info_data.hoyoverse_cookie, lang="zh-cn")
            if uid <= 0:
                _uid = user_info_data.hoyoverse_game_uid
            else:
                _uid = uid
        try:
            user_info = await client.get_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.service.template.render('genshin/info', "info.html", user_data,
                                                      {"width": 1024, "height": 1024})
        return png_data

    @conversation_error_handler
    @restricts(return_data=ConversationHandler.END)
    async def command_start(self, update: Update, context: CallbackContext) -> int:
        user = update.effective_user
        message = update.message
        Log.info(f"用户 {user.full_name}[{user.id}] 查询游戏用户命令请求")
        uid_command_data: UidCommandData = context.chat_data.get("uid_command_data")
        if uid_command_data is None:
            uid_command_data = UidCommandData()
            context.chat_data["uid_command_data"] = uid_command_data
        user_info = await self.service.user_service_db.get_user_info(user.id)
        if user_info.user_id == 0:
            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, 300)
                self._add_delete_message_job(context, message.chat_id, message.message_id, 300)
            return ConversationHandler.END
        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("获取 chat_id 发生错误！ 错误信息为", error)
            await message.reply_text("输入错误")
            return ConversationHandler.END
        if user_info.service == ServiceEnum.NULL:
            reply_text = "请选择你要查询的类别"
            keyboard = [
                [
                    InlineKeyboardButton("米游社", callback_data="uid|米游社"),
                    InlineKeyboardButton("HoYoLab", callback_data="uid|HoYoLab")
                ]
            ]
            uid_command_data.user_info = user_info
            await message.reply_text(reply_text, reply_markup=InlineKeyboardMarkup(keyboard))
            return self.COMMAND_RESULT
        else:
            await message.reply_chat_action(ChatAction.TYPING)
            try:
                png_data = await self._start_get_user_info(user_info, user_info.service, uid)
            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"{user_info.user_id}.png", allow_sending_without_reply=True)

        return ConversationHandler.END

    @conversation_error_handler
    async def command_result(self, update: Update, context: CallbackContext) -> int:
        get_user_command_data: UidCommandData = context.chat_data["uid_command_data"]
        query = update.callback_query
        message = query.message
        await query.answer()
        await query.delete_message()
        if query.data == "uid|米游社":
            service = ServiceEnum.HYPERION
        elif query.data == "uid|HoYoLab":
            service = ServiceEnum.HOYOLAB
        else:
            return ConversationHandler.END
        try:
            png_data = await self._start_get_user_info(get_user_command_data.user_info, service)
        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"{get_user_command_data.user_info.user_id}.png",
                                  allow_sending_without_reply=True)
        return ConversationHandler.END