mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
309 lines
12 KiB
Python
309 lines
12 KiB
Python
"""虚构叙事数据查询"""
|
||
import asyncio
|
||
import re
|
||
from functools import lru_cache
|
||
from typing import Any, List, Optional, Tuple, Union, TYPE_CHECKING
|
||
|
||
from arkowrapper import ArkoWrapper
|
||
from pytz import timezone
|
||
from telegram import Message, Update
|
||
from telegram.constants import ChatAction, ParseMode
|
||
from telegram.ext import CallbackContext, filters
|
||
|
||
from core.dependence.assets import AssetsService
|
||
from core.plugin import Plugin, handler
|
||
from core.services.cookies.error import TooManyRequestPublicCookies
|
||
from core.services.template.models import RenderGroupResult, RenderResult
|
||
from core.services.template.services import TemplateService
|
||
from plugins.tools.genshin import GenshinHelper
|
||
from utils.log import logger
|
||
from utils.uid import mask_number
|
||
|
||
try:
|
||
import ujson as jsonlib
|
||
|
||
except ImportError:
|
||
import json as jsonlib
|
||
|
||
if TYPE_CHECKING:
|
||
from simnet import StarRailClient
|
||
from simnet.models.starrail.chronicle.challenge_story import StarRailChallengeStory
|
||
|
||
|
||
TZ = timezone("Asia/Shanghai")
|
||
cmd_pattern = r"(?i)^/challenge_story(?:@[\w]+)?\s*((?:\d+)|(?:all))?\s*(pre)?"
|
||
msg_pattern = r"^虚构叙事数据((?:查询)|(?:总览))(上期)?\D?(\d*)?.*?$"
|
||
MAX_FLOOR = 4
|
||
|
||
|
||
@lru_cache
|
||
def get_args(text: str) -> Tuple[int, bool, bool]:
|
||
if text.startswith("/"):
|
||
result = re.match(cmd_pattern, text).groups()
|
||
try:
|
||
floor = int(result[0] or 0)
|
||
if floor > 100:
|
||
floor = 0
|
||
except ValueError:
|
||
floor = 0
|
||
return floor, result[0] == "all", bool(result[1])
|
||
result = re.match(msg_pattern, text).groups()
|
||
return int(result[2] or 0), result[0] == "总览", result[1] == "上期"
|
||
|
||
|
||
class AbyssUnlocked(Exception):
|
||
"""根本没动"""
|
||
|
||
|
||
class AbyssFastPassed(Exception):
|
||
"""快速通过，无数据"""
|
||
|
||
|
||
class ChallengeStoryPlugin(Plugin):
|
||
"""虚构叙事数据查询"""
|
||
|
||
def __init__(
|
||
self,
|
||
template: TemplateService,
|
||
helper: GenshinHelper,
|
||
assets_service: AssetsService,
|
||
):
|
||
self.template_service = template
|
||
self.helper = helper
|
||
self.assets_service = assets_service
|
||
|
||
async def get_uid(self, user_id: int, args: List[str], reply: Optional[Message]) -> int:
|
||
"""通过消息获取 uid，优先级：args > reply > self"""
|
||
uid, user_id_ = None, user_id
|
||
if args:
|
||
for i in args:
|
||
if i is not None and i.isdigit() and len(i) == 9:
|
||
uid = int(i)
|
||
if reply:
|
||
try:
|
||
user_id_ = reply.from_user.id
|
||
except AttributeError:
|
||
pass
|
||
if not uid:
|
||
player_info = await self.helper.players_service.get_player(user_id_)
|
||
if player_info is not None:
|
||
uid = player_info.player_id
|
||
if (not uid) and (user_id_ != user_id):
|
||
player_info = await self.helper.players_service.get_player(user_id)
|
||
if player_info is not None:
|
||
uid = player_info.player_id
|
||
return uid
|
||
|
||
@handler.command("challenge_story", block=False)
|
||
@handler.message(filters.Regex(msg_pattern), block=False)
|
||
async def command_start(self, update: Update, context: CallbackContext) -> None:
|
||
user_id = await self.get_real_user_id(update)
|
||
message = update.effective_message
|
||
uid: int = await self.get_uid(user_id, context.args, message.reply_to_message)
|
||
|
||
# 若查询帮助
|
||
if (message.text.startswith("/") and "help" in message.text) or "帮助" in message.text:
|
||
await message.reply_text(
|
||
"<b>虚构叙事数据</b>功能使用帮助（中括号表示可选参数）\n\n"
|
||
"指令格式：\n<code>/challenge_story + [层数/all] + [pre]</code>\n（<code>pre</code>表示上期）\n\n"
|
||
"文本格式：\n<code>虚构叙事数据 + 查询/总览 + [上期] + [层数]</code> \n\n"
|
||
"例如以下指令都正确：\n"
|
||
"<code>/challenge_story</code>\n<code>/challenge_story 1 pre</code>\n<code>/challenge_story all pre</code>\n"
|
||
"<code>虚构叙事数据查询</code>\n<code>虚构叙事数据查询上期第1层</code>\n<code>虚构叙事数据总览上期</code>",
|
||
parse_mode=ParseMode.HTML,
|
||
)
|
||
self.log_user(update, logger.info, "查询[bold]虚构叙事数据[/bold]帮助", extra={"markup": True})
|
||
return
|
||
|
||
# 解析参数
|
||
floor, total, previous = get_args(message.text)
|
||
|
||
if floor > MAX_FLOOR or floor < 0:
|
||
reply_msg = await message.reply_text(f"虚构叙事层数输入错误，请重新输入。支持的参数为： 1-{MAX_FLOOR} 或 all")
|
||
if filters.ChatType.GROUPS.filter(message):
|
||
self.add_delete_message_job(reply_msg)
|
||
self.add_delete_message_job(message)
|
||
return
|
||
|
||
self.log_user(
|
||
update,
|
||
logger.info,
|
||
"[bold]虚构叙事挑战数据[/bold]请求: uid=%s floor=%s total=%s previous=%s",
|
||
uid,
|
||
floor,
|
||
total,
|
||
previous,
|
||
extra={"markup": True},
|
||
)
|
||
|
||
async def reply_message_func(content: str) -> None:
|
||
_reply_msg = await message.reply_text(f"开拓者 (<code>{uid}</code>) {content}", parse_mode=ParseMode.HTML)
|
||
|
||
reply_text: Optional[Message] = None
|
||
|
||
try:
|
||
async with self.helper.genshin_or_public(user_id, uid=uid) as client:
|
||
if total:
|
||
reply_text = await message.reply_text("彦卿需要时间整理虚构叙事数据，还请耐心等待哦~")
|
||
await message.reply_chat_action(ChatAction.TYPING)
|
||
images = await self.get_rendered_pic(client, uid, floor, total, previous)
|
||
except TooManyRequestPublicCookies:
|
||
reply_message = await message.reply_text("查询次数太多，请您稍后重试")
|
||
if filters.ChatType.GROUPS.filter(message):
|
||
self.add_delete_message_job(reply_message)
|
||
self.add_delete_message_job(message)
|
||
return
|
||
except AbyssUnlocked: # 若虚构叙事未解锁
|
||
await reply_message_func("还未解锁虚构叙事哦~")
|
||
return
|
||
except AbyssFastPassed: # 若虚构叙事已快速通过
|
||
await reply_message_func("本层已被快速通过，无详细数据~")
|
||
return
|
||
except IndexError: # 若虚构叙事为挑战此层
|
||
await reply_message_func("还没有挑战本层呢，咕咕咕~")
|
||
return
|
||
if images is None:
|
||
await reply_message_func(f"还没有第 {floor} 层的挑战数据")
|
||
return
|
||
|
||
await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
|
||
|
||
for group in ArkoWrapper(images).group(10): # 每 10 张图片分一个组
|
||
await RenderGroupResult(results=group).reply_media_group(
|
||
message, allow_sending_without_reply=True, write_timeout=60
|
||
)
|
||
|
||
if reply_text is not None:
|
||
await reply_text.delete()
|
||
|
||
self.log_user(update, logger.info, "[bold]虚构叙事挑战数据[/bold]: 成功发送图片", extra={"markup": True})
|
||
|
||
@staticmethod
|
||
def get_floor_data(abyss_data: "StarRailChallengeStory", floor: int):
|
||
try:
|
||
floor_data = abyss_data.floors[-floor]
|
||
except IndexError:
|
||
floor_data = None
|
||
if not floor_data:
|
||
raise AbyssUnlocked()
|
||
if floor_data.is_fast or floor_data.round_num == 0:
|
||
raise AbyssFastPassed()
|
||
render_data = {
|
||
"floor": floor_data,
|
||
"floor_time": floor_data.node_1.challenge_time.datetime.astimezone(TZ).strftime("%Y-%m-%d %H:%M:%S"),
|
||
"floor_nodes": [floor_data.node_1, floor_data.node_2],
|
||
"floor_num": floor,
|
||
}
|
||
return render_data
|
||
|
||
async def get_rendered_pic(
|
||
self, client: "StarRailClient", uid: int, floor: int, total: bool, previous: bool
|
||
) -> Union[
|
||
Tuple[
|
||
Union[BaseException, Any],
|
||
Union[BaseException, Any],
|
||
Union[BaseException, Any],
|
||
Union[BaseException, Any],
|
||
Union[BaseException, Any],
|
||
],
|
||
List[RenderResult],
|
||
None,
|
||
]:
|
||
"""
|
||
获取渲染后的图片
|
||
|
||
Args:
|
||
client (Client): 获取 genshin 数据的 client
|
||
uid (int): 需要查询的 uid
|
||
floor (int): 层数
|
||
total (bool): 是否为总览
|
||
previous (bool): 是否为上期
|
||
|
||
Returns:
|
||
bytes格式的图片
|
||
"""
|
||
|
||
abyss_data = await client.get_starrail_challenge_story(uid, previous=previous, lang="zh-cn")
|
||
if not abyss_data.has_data:
|
||
raise AbyssUnlocked()
|
||
if not abyss_data.groups:
|
||
raise AbyssUnlocked()
|
||
season = abyss_data.groups[0]
|
||
start_time = season.begin_time.datetime.astimezone(TZ).strftime("%m月%d日 %H:%M")
|
||
end_time = season.end_time.datetime.astimezone(TZ).strftime("%m月%d日 %H:%M")
|
||
total_stars = f"{abyss_data.total_stars}"
|
||
|
||
render_data = {
|
||
"title": "虚构叙事",
|
||
"start_time": start_time,
|
||
"end_time": end_time,
|
||
"stars": total_stars,
|
||
"uid": mask_number(uid),
|
||
"max_floor": abyss_data.max_floor,
|
||
"total_battles": abyss_data.total_battles,
|
||
"floor_colors": {
|
||
1: "#374952",
|
||
2: "#374952",
|
||
3: "#55464B",
|
||
4: "#55464B",
|
||
5: "#55464B",
|
||
6: "#1D2A5D",
|
||
7: "#1D2A5D",
|
||
8: "#1D2A5D",
|
||
9: "#292B58",
|
||
10: "#382024",
|
||
11: "#252550",
|
||
12: "#1D2A4A",
|
||
},
|
||
}
|
||
if total:
|
||
|
||
def floor_task(floor_index: int):
|
||
_abyss_data = self.get_floor_data(abyss_data, floor_index)
|
||
return (
|
||
floor_index,
|
||
self.template_service.render(
|
||
"starrail/abyss/floor_story.html",
|
||
{
|
||
**render_data,
|
||
**_abyss_data,
|
||
},
|
||
viewport={"width": 690, "height": 500},
|
||
full_page=True,
|
||
ttl=15 * 24 * 60 * 60,
|
||
),
|
||
)
|
||
|
||
render_inputs = []
|
||
floors = abyss_data.floors[::-1]
|
||
for i in range(len(floors)):
|
||
try:
|
||
render_inputs.append(floor_task(i + 1))
|
||
except AbyssFastPassed:
|
||
pass
|
||
|
||
render_group_inputs = list(map(lambda x: x[1], sorted(render_inputs, key=lambda x: x[0])))
|
||
|
||
return await asyncio.gather(*render_group_inputs)
|
||
|
||
if floor < 1:
|
||
return [
|
||
await self.template_service.render(
|
||
"starrail/abyss/overview.html", render_data, viewport={"width": 750, "height": 250}
|
||
)
|
||
]
|
||
try:
|
||
floor_data = abyss_data.floors[-floor]
|
||
except IndexError:
|
||
return None
|
||
if not floor_data:
|
||
return None
|
||
if floor_data.is_fast or floor_data.round_num == 0:
|
||
raise AbyssFastPassed()
|
||
render_data.update(self.get_floor_data(abyss_data, floor))
|
||
return [
|
||
await self.template_service.render(
|
||
"starrail/abyss/floor_story.html", render_data, viewport={"width": 690, "height": 500}
|
||
)
|
||
]
|