mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-22 15:36:44 +00:00
eed418477f
Previously, all templates were named with the .html extension, which caused problems for editors to recognize the Jinja2 syntax used in them. In this commit, all templates have been renamed to use the .jinja2 extension to better reflect their use of Jinja2 and to improve editor recognition.
134 lines
5.5 KiB
Python
134 lines
5.5 KiB
Python
import random
|
|
from typing import Optional
|
|
|
|
from genshin import Client, GenshinException
|
|
from genshin.models import GenshinUserStats
|
|
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.constants import ChatAction
|
|
from telegram.ext import CallbackContext, filters
|
|
from telegram.helpers import create_deep_linked_url
|
|
|
|
from core.plugin import Plugin, handler
|
|
from core.services.cookies.error import TooManyRequestPublicCookies
|
|
from core.services.template.models import RenderResult
|
|
from core.services.template.services import TemplateService
|
|
from plugins.tools.genshin import GenshinHelper, PlayerNotFoundError, CookiesNotFoundError
|
|
from utils.log import logger
|
|
|
|
__all__ = ("PlayerStatsPlugins",)
|
|
|
|
|
|
class PlayerStatsPlugins(Plugin):
|
|
"""玩家统计查询"""
|
|
|
|
def __init__(
|
|
self,
|
|
template: TemplateService,
|
|
helper: GenshinHelper,
|
|
):
|
|
self.template_service = template
|
|
self.helper = helper
|
|
|
|
@handler.command("stats", block=False)
|
|
@handler.message(filters.Regex("^玩家统计查询(.*)"), block=False)
|
|
async def command_start(self, update: Update, context: CallbackContext) -> Optional[int]:
|
|
user = update.effective_user
|
|
message = update.effective_message
|
|
logger.info("用户 %s[%s] 查询游戏用户命令请求", user.full_name, user.id)
|
|
uid: Optional[int] = None
|
|
try:
|
|
args = context.args
|
|
if args is not None and len(args) >= 1:
|
|
uid = int(args[0])
|
|
except ValueError as exc:
|
|
logger.warning("获取 uid 发生错误！ 错误信息为 %s", str(exc))
|
|
await message.reply_text("输入错误")
|
|
return
|
|
try:
|
|
try:
|
|
client = await self.helper.get_genshin_client(user.id)
|
|
except CookiesNotFoundError:
|
|
client, uid = await self.helper.get_public_genshin_client(user.id)
|
|
render_result = await self.render(client, uid)
|
|
except PlayerNotFoundError:
|
|
buttons = [[InlineKeyboardButton("点我绑定账号", url=create_deep_linked_url(context.bot.username, "set_cookie"))]]
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
reply_message = await message.reply_text(
|
|
"未查询到您所绑定的账号信息，请先私聊派蒙绑定账号", reply_markup=InlineKeyboardMarkup(buttons)
|
|
)
|
|
self.add_delete_message_job(reply_message, delay=30)
|
|
self.add_delete_message_job(message, delay=30)
|
|
else:
|
|
await message.reply_text("未查询到您所绑定的账号信息，请先绑定账号", reply_markup=InlineKeyboardMarkup(buttons))
|
|
return
|
|
except GenshinException as exc:
|
|
if exc.retcode == 1034 and uid:
|
|
await message.reply_text("出错了呜呜呜 ~ 请稍后重试")
|
|
return
|
|
raise exc
|
|
except TooManyRequestPublicCookies:
|
|
await message.reply_text("用户查询次数过多 请稍后重试")
|
|
return
|
|
except AttributeError as exc:
|
|
logger.error("角色数据有误")
|
|
logger.exception(exc)
|
|
await message.reply_text("角色数据有误 估计是派蒙晕了")
|
|
return
|
|
await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
|
|
await render_result.reply_photo(message, filename=f"{client.uid}.png", allow_sending_without_reply=True)
|
|
|
|
async def render(self, client: Client, uid: Optional[int] = None) -> RenderResult:
|
|
if uid is None:
|
|
uid = client.uid
|
|
|
|
user_info = await client.get_genshin_user(uid)
|
|
logger.debug(user_info)
|
|
|
|
# 因为需要替换线上图片地址为本地地址，先克隆数据，避免修改原数据
|
|
user_info = user_info.copy(deep=True)
|
|
|
|
data = {
|
|
"uid": uid,
|
|
"info": user_info.info,
|
|
"stats": user_info.stats,
|
|
"explorations": user_info.explorations,
|
|
"teapot": user_info.teapot,
|
|
"stats_labels": [
|
|
("活跃天数", "days_active"),
|
|
("成就达成数", "achievements"),
|
|
("获取角色数", "characters"),
|
|
("深境螺旋", "spiral_abyss"),
|
|
("解锁传送点", "unlocked_waypoints"),
|
|
("解锁秘境", "unlocked_domains"),
|
|
("奇馈宝箱数", "remarkable_chests"),
|
|
("华丽宝箱数", "luxurious_chests"),
|
|
("珍贵宝箱数", "precious_chests"),
|
|
("精致宝箱数", "exquisite_chests"),
|
|
("普通宝箱数", "common_chests"),
|
|
("风神瞳", "anemoculi"),
|
|
("岩神瞳", "geoculi"),
|
|
("雷神瞳", "electroculi"),
|
|
("草神瞳", "dendroculi"),
|
|
],
|
|
"style": random.choice(["mondstadt", "liyue"]), # nosec
|
|
}
|
|
|
|
await self.cache_images(user_info)
|
|
|
|
return await self.template_service.render(
|
|
"genshin/stats/stats.jinja2",
|
|
data,
|
|
{"width": 650, "height": 800},
|
|
full_page=True,
|
|
)
|
|
|
|
async def cache_images(self, data: GenshinUserStats) -> None:
|
|
"""缓存所有图片到本地"""
|
|
# TODO: 并发下载所有资源
|
|
|
|
# 探索地区
|
|
for item in data.explorations:
|
|
item.__config__.allow_mutation = True
|
|
item.icon = await self.download_resource(item.icon)
|
|
item.cover = await self.download_resource(item.cover)
|