PaiGram/plugins/genshin/abyss.py

from genshin import Client
from telegram import Update
from telegram.constants import ChatAction
from telegram.ext import CommandHandler, MessageHandler, filters, CallbackContext

from core.cookies.services import CookiesService
from core.template.services import TemplateService
from core.user import UserService
from core.user.repositories import UserNotFoundError
from logger import Log
from plugins.base import BasePlugins
from utils.decorators.error import error_callable
from utils.decorators.restricts import restricts
from utils.helpers import get_genshin_client, url_to_file
from utils.plugins.manager import listener_plugins_class
from utils.service.inject import inject


class AbyssUnlocked(Exception):
    """根本没动"""
    pass


class NoMostKills(Exception):
    """挑战了但是数据没刷新"""
    pass


@listener_plugins_class()
class Abyss(BasePlugins):
    """深渊数据查询"""

    @inject
    def __init__(self, user_service: UserService = None, cookies_service: CookiesService = None,
                 template_service: TemplateService = None):
        self.template_service = template_service
        self.cookies_service = cookies_service
        self.user_service = user_service

    @classmethod
    def create_handlers(cls) -> list:
        abyss = cls()
        return [
            CommandHandler("abyss", abyss.command_start, block=False),
            MessageHandler(filters.Regex(r"^深渊数据查询(.*)"), abyss.command_start, block=True)
        ]

    @staticmethod
    def _get_role_star_bg(value: int):
        if value == 4:
            return "./background/roleStarBg4.png"
        elif value == 5:
            return "./background/roleStarBg5.png"
        else:
            raise ValueError("错误的数据")

    async def _get_abyss_data(self, client: Client) -> dict:
        uid = client.uid
        await client.get_record_cards()
        spiral_abyss_info = await client.get_spiral_abyss(uid)
        if not spiral_abyss_info.unlocked:
            raise AbyssUnlocked
        ranks = spiral_abyss_info.ranks
        if len(spiral_abyss_info.ranks.most_kills) == 0:
            raise NoMostKills
        abyss_data = {
            "uid": uid,
            "max_floor": spiral_abyss_info.max_floor,
            "total_battles": spiral_abyss_info.total_battles,
            "total_stars": spiral_abyss_info.total_stars,
            "most_played_list": [],
            "most_kills": {
                "icon": await url_to_file(ranks.most_kills[0].icon),
                "value": ranks.most_kills[0].value,
            },
            "strongest_strike": {
                "icon": await url_to_file(ranks.strongest_strike[0].icon),
                "value": ranks.strongest_strike[0].value
            },
            "most_damage_taken": {
                "icon": await url_to_file(ranks.most_damage_taken[0].icon),
                "value": ranks.most_damage_taken[0].value
            },
            "most_bursts_used": {
                "icon": await url_to_file(ranks.most_bursts_used[0].icon),
                "value": ranks.most_bursts_used[0].value
            },
            "most_skills_used": {
                "icon": await url_to_file(ranks.most_skills_used[0].icon),
                "value": ranks.most_skills_used[0].value
            }
        }
        # most_kills
        most_played_list = ranks.most_played
        for most_played in most_played_list:
            temp = {
                "icon": await url_to_file(most_played.icon),
                "value": most_played.value,
                "background": self._get_role_star_bg(most_played.rarity)
            }
            abyss_data["most_played_list"].append(temp)
        return abyss_data

    @restricts()
    @error_callable
    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}] 查深渊挑战命令请求")
        await message.reply_chat_action(ChatAction.TYPING)
        try:
            client = await get_genshin_client(user.id, self.user_service, self.cookies_service)
            abyss_data = await self._get_abyss_data(client)
        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, 10)
                self._add_delete_message_job(context, message.chat_id, message.message_id, 10)
            return
        except (AbyssUnlocked, NoMostKills):
            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, 10)
                self._add_delete_message_job(context, message.chat_id, message.message_id, 10)
            return
        await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
        png_data = await self.template_service.render('genshin/abyss', "abyss.html", abyss_data,
                                                      {"width": 690, "height": 504}, full_page=False)
        await message.reply_photo(png_data, filename=f"abyss_{user.id}.png",
                                  allow_sending_without_reply=True)
        return