mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-30 01:14:03 +00:00
264 lines
12 KiB
Python
264 lines
12 KiB
Python
"""练度统计"""
|
|
import asyncio
|
|
from typing import Iterable, List, Optional, Sequence
|
|
|
|
from arkowrapper import ArkoWrapper
|
|
from enkanetwork import Assets as EnkaAssets, EnkaNetworkAPI
|
|
from genshin import Client, GenshinException
|
|
from genshin.models import CalculatorCharacterDetails, CalculatorTalent, Character
|
|
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Message, Update, User
|
|
from telegram.constants import ChatAction, ParseMode
|
|
from telegram.ext import CallbackContext, filters
|
|
|
|
from core.base.assets import AssetsService
|
|
from core.baseplugin import BasePlugin
|
|
from core.config import config
|
|
from core.cookies.error import CookiesNotFoundError
|
|
from core.cookies.services import CookiesService
|
|
from core.plugin import Plugin, handler
|
|
from core.template import TemplateService
|
|
from core.template.models import FileType
|
|
from core.user.error import UserNotFoundError
|
|
from metadata.genshin import AVATAR_DATA, NAMECARD_DATA
|
|
from modules.wiki.base import Model
|
|
from utils.decorators.error import error_callable
|
|
from utils.decorators.restricts import restricts
|
|
from utils.helpers import get_genshin_client
|
|
from utils.log import logger
|
|
|
|
|
|
class AvatarListPlugin(Plugin, BasePlugin):
|
|
def __init__(
|
|
self, cookies_service: CookiesService, assets_service: AssetsService, template_service: TemplateService
|
|
) -> None:
|
|
self.cookies_service = cookies_service
|
|
self.assets_service = assets_service
|
|
self.template_service = template_service
|
|
self.enka_client = EnkaNetworkAPI(lang="chs", agent=config.enka_network_api_agent)
|
|
self.enka_assets = EnkaAssets(lang="chs")
|
|
|
|
async def get_user_client(self, user: User, message: Message, context: CallbackContext) -> Optional[Client]:
|
|
try:
|
|
return await get_genshin_client(user.id)
|
|
except UserNotFoundError: # 若未找到账号
|
|
buttons = [[InlineKeyboardButton("点我绑定账号", url=f"https://t.me/{context.bot.username}?start=set_cookie")]]
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
reply_message = await message.reply_text(
|
|
"未查询到您所绑定的账号信息，请先私聊派蒙绑定账号", reply_markup=InlineKeyboardMarkup(buttons)
|
|
)
|
|
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)
|
|
else:
|
|
await message.reply_text("未查询到您所绑定的账号信息，请先绑定账号", reply_markup=InlineKeyboardMarkup(buttons))
|
|
except CookiesNotFoundError:
|
|
buttons = [[InlineKeyboardButton("点我绑定账号", url=f"https://t.me/{context.bot.username}?start=set_cookie")]]
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
reply_msg = await message.reply_text(
|
|
"此功能需要绑定<code>cookie</code>后使用，请先私聊派蒙绑定账号",
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
parse_mode=ParseMode.HTML,
|
|
)
|
|
self._add_delete_message_job(context, reply_msg.chat_id, reply_msg.message_id, 30)
|
|
self._add_delete_message_job(context, message.chat_id, message.message_id, 30)
|
|
else:
|
|
await message.reply_text(
|
|
"此功能需要绑定<code>cookie</code>后使用，请先私聊派蒙进行绑定",
|
|
parse_mode=ParseMode.HTML,
|
|
reply_markup=InlineKeyboardMarkup(buttons),
|
|
)
|
|
|
|
async def get_avatar_data(self, character: Character, client: Client) -> Optional["AvatarData"]:
|
|
detail = None
|
|
for _ in range(5):
|
|
try:
|
|
detail = await client.get_character_details(character)
|
|
except Exception as e: # pylint: disable=W0703
|
|
if isinstance(e, GenshinException) and "Too Many Requests" in e.msg:
|
|
await asyncio.sleep(0.2)
|
|
continue
|
|
if character.name == "旅行者":
|
|
logger.debug(f"解析旅行者数据时遇到了错误：{e}")
|
|
return None
|
|
raise e
|
|
else:
|
|
break
|
|
else:
|
|
logger.warning(f"解析[bold]{character.name}[/]的数据时遇到了 Too Many Requests 错误", extra={"markup": True})
|
|
return None
|
|
if character.id == 10000005: # 针对男草主
|
|
talents = []
|
|
for talent in detail.talents:
|
|
if "普通攻击" in talent.name:
|
|
talent.Config.allow_mutation = True
|
|
# noinspection Pydantic
|
|
talent.group_id = 1131
|
|
if talent.type in ["attack", "skill", "burst"]:
|
|
talents.append(talent)
|
|
else:
|
|
talents = [t for t in detail.talents if t.type in ["attack", "skill", "burst"]]
|
|
buffed_talents = []
|
|
for constellation in filter(lambda x: x.pos in [3, 5], character.constellations[: character.constellation]):
|
|
if result := list(
|
|
filter(lambda x: all([x.name in constellation.effect]), talents) # pylint: disable=W0640
|
|
):
|
|
buffed_talents.append(result[0].type)
|
|
return AvatarData(
|
|
avatar=character,
|
|
detail=detail,
|
|
icon=(await self.assets_service.avatar(character.id).side()).as_uri(),
|
|
weapon=(
|
|
await self.assets_service.weapon(character.weapon.id).__getattr__(
|
|
"icon" if character.weapon.ascension < 2 else "awaken"
|
|
)()
|
|
).as_uri(),
|
|
skills=[
|
|
SkillData(skill=s, buffed=s.type in buffed_talents)
|
|
for s in sorted(talents, key=lambda x: ["attack", "skill", "burst"].index(x.type))
|
|
],
|
|
)
|
|
|
|
async def get_avatars_data(
|
|
self, characters: Sequence[Character], client: Client, max_length: int = None
|
|
) -> List["AvatarData"]:
|
|
async def _task(c, n):
|
|
return n, await self.get_avatar_data(c, client)
|
|
|
|
task_results = await asyncio.gather(
|
|
*[_task(character, num) for num, character in enumerate(characters[:max_length])]
|
|
)
|
|
|
|
return list(filter(lambda x: x, map(lambda x: x[1], sorted(task_results, key=lambda x: x[0]))))
|
|
|
|
async def get_final_data(self, client: Client, characters: Sequence[Character], update: Update):
|
|
try:
|
|
response = await self.enka_client.fetch_user(client.uid)
|
|
namecard = (await self.assets_service.namecard(response.player.namecard.id).navbar()).as_uri()
|
|
avatar = (await self.assets_service.avatar(response.player.icon.id).icon()).as_uri()
|
|
nickname = response.player.nickname
|
|
if response.player.icon.id in [10000005, 10000007]:
|
|
rarity = 5
|
|
else:
|
|
rarity = {k: v["rank"] for k, v in AVATAR_DATA.items()}[str(response.player.icon.id)]
|
|
except Exception as e: # pylint: disable=W0703
|
|
logger.debug(f"enka 请求失败: {e}")
|
|
choices = ArkoWrapper(characters).filter(lambda x: x.friendship == 10) # 筛选出好感满了的角色
|
|
if choices.length == 0: # 若没有满好感角色、则以好感等级排序
|
|
choices = ArkoWrapper(characters).sort(lambda x: x.friendship, reverse=True)
|
|
namecard_choices = ( # 找到与角色对应的满好感名片ID
|
|
ArkoWrapper(choices)
|
|
.map(lambda x: next(filter(lambda y: y["name"].split(".")[0] == x.name, NAMECARD_DATA.values()), None))
|
|
.filter(lambda x: x)
|
|
.map(lambda x: x["id"])
|
|
)
|
|
namecard = (await self.assets_service.namecard(namecard_choices[0]).navbar()).as_uri()
|
|
avatar = (await self.assets_service.avatar(cid := choices[0].id).icon()).as_uri()
|
|
nickname = update.effective_user.full_name
|
|
if cid in [10000005, 10000007]:
|
|
rarity = 5
|
|
else:
|
|
rarity = {k: v["rank"] for k, v in AVATAR_DATA.items()}[str(cid)]
|
|
return namecard, avatar, nickname, rarity
|
|
|
|
async def get_default_final_data(self, characters: Sequence[Character], update: Update):
|
|
nickname = update.effective_user.full_name
|
|
rarity = 5
|
|
# 须弥·正明
|
|
namecard = (await self.assets_service.namecard(210132).navbar()).as_uri()
|
|
if traveller := next(filter(lambda x: x.id in [10000005, 10000007], characters), None):
|
|
avatar = (await self.assets_service.avatar(traveller.id).icon()).as_uri()
|
|
else:
|
|
avatar = (await self.assets_service.avatar(10000005).icon()).as_uri()
|
|
return namecard, avatar, nickname, rarity
|
|
|
|
@handler.command("avatars", filters.Regex(r"^/avatars\s*(?:(\d+)|(all))?$"), block=False)
|
|
@handler.message(filters.Regex(r"^(全部)?练度统计$"), block=False)
|
|
@restricts(30)
|
|
@error_callable
|
|
async def avatar_list(self, update: Update, context: CallbackContext):
|
|
user = update.effective_user
|
|
message = update.effective_message
|
|
|
|
args = [i.lower() for i in context.match.groups() if i]
|
|
|
|
all_avatars = any(["all" in args, "全部" in args]) # 是否发送全部角色
|
|
|
|
logger.info(f"用户 {user.full_name}[{user.id}] [bold]练度统计[/bold]: all={all_avatars}", extra={"markup": True})
|
|
|
|
client = await self.get_user_client(user, message, context)
|
|
if not client:
|
|
return
|
|
|
|
notice = await message.reply_text("派蒙需要收集整理数据，还请耐心等待哦~")
|
|
await message.reply_chat_action(ChatAction.TYPING)
|
|
|
|
characters = await client.get_genshin_characters(client.uid)
|
|
|
|
try:
|
|
avatar_datas: List[AvatarData] = await self.get_avatars_data(
|
|
characters, client, None if all_avatars else 20
|
|
)
|
|
except GenshinException as e:
|
|
if e.retcode == -502002:
|
|
self._add_delete_message_job(context, notice.chat_id, notice.message_id, 5)
|
|
notice = await message.reply_text("请先在米游社中使用一次<b>养成计算器</b>后再使用此功能~", parse_mode=ParseMode.HTML)
|
|
self._add_delete_message_job(context, notice.chat_id, notice.message_id, 20)
|
|
return
|
|
raise e
|
|
|
|
try:
|
|
namecard, avatar, nickname, rarity = await self.get_final_data(client, characters, update)
|
|
except Exception as e:
|
|
logger.debug(f"卡片信息请求失败: {e}")
|
|
namecard, avatar, nickname, rarity = await self.get_default_final_data(characters, update)
|
|
|
|
render_data = {
|
|
"uid": client.uid, # 玩家uid
|
|
"nickname": nickname, # 玩家昵称
|
|
"avatar": avatar, # 玩家头像
|
|
"rarity": rarity, # 玩家头像对应的角色星级
|
|
"namecard": namecard, # 玩家名片
|
|
"avatar_datas": avatar_datas, # 角色数据
|
|
"has_more": len(characters) != len(avatar_datas), # 是否显示了全部角色
|
|
}
|
|
|
|
as_document = all_avatars and len(characters) > 20
|
|
|
|
await message.reply_chat_action(ChatAction.UPLOAD_DOCUMENT if as_document else ChatAction.UPLOAD_PHOTO)
|
|
|
|
image = await self.template_service.render(
|
|
"genshin/avatar_list/main.html",
|
|
render_data,
|
|
viewport={"width": 1040, "height": 500},
|
|
full_page=True,
|
|
query_selector=".container",
|
|
file_type=FileType.DOCUMENT if as_document else FileType.PHOTO,
|
|
ttl=30 * 24 * 60 * 60,
|
|
)
|
|
self._add_delete_message_job(context, notice.chat_id, notice.message_id, 5)
|
|
if as_document:
|
|
await image.reply_document(message, filename="练度统计.png")
|
|
else:
|
|
await image.reply_photo(message)
|
|
|
|
logger.info(
|
|
f"用户 {user.full_name}[{user.id}] [bold]练度统计[/bold]发送{'文件' if all_avatars else '图片'}成功",
|
|
extra={"markup": True},
|
|
)
|
|
|
|
|
|
class SkillData(Model):
|
|
"""天赋数据"""
|
|
|
|
skill: CalculatorTalent
|
|
buffed: bool = False
|
|
"""是否得到了命座加成"""
|
|
|
|
|
|
class AvatarData(Model):
|
|
avatar: Character
|
|
detail: CalculatorCharacterDetails
|
|
icon: str
|
|
weapon: Optional[str]
|
|
skills: Iterable[SkillData]
|