343 lines
14 KiB
Python
343 lines
14 KiB
Python
import asyncio
|
||
import json
|
||
from copy import deepcopy
|
||
from time import time
|
||
from typing import Dict, List, Literal, Union
|
||
|
||
from httpx import AsyncClient, HTTPError
|
||
|
||
# from nonebot import require
|
||
# from nonebot.log import logger
|
||
from ci import logger
|
||
|
||
# require("nonebot_plugin_htmlrender")
|
||
# from nonebot_plugin_htmlrender import template_to_pic
|
||
from defs.browser import template_to_pic
|
||
|
||
from gspanel.__utils__ import LOCAL_DIR, SCALE_FACTOR, TPL_VERSION, download
|
||
from gspanel.data_convert import (
|
||
simplDamageRes,
|
||
simplFightProp,
|
||
simplTeamDamageRes,
|
||
transFromEnka,
|
||
transToTeyvat,
|
||
)
|
||
|
||
|
||
async def queryPanelApi(uid: str, source: Literal["enka", "mgg"] = "enka") -> Dict:
|
||
"""
|
||
原神游戏内角色展柜数据请求
|
||
|
||
* ``param uid: str`` 查询用户 UID
|
||
* ``param source: Literal["enka", "mgg"] = "enka"`` 查询接口
|
||
- ``return: Dict`` 查询结果，出错时返回 ``{"error": "错误信息"}``
|
||
"""
|
||
enkaMirrors = [
|
||
"https://enka.network",
|
||
"https://enka.minigg.cn",
|
||
"https://enka.microgg.cn",
|
||
]
|
||
async with AsyncClient() as client:
|
||
resJson = {}
|
||
for idx, root in enumerate(enkaMirrors):
|
||
try:
|
||
res = await client.get(
|
||
url=f"{root}/u/{uid}/__data.json",
|
||
headers={
|
||
"Accept": "application/json",
|
||
"Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8,en-US;q=0.7",
|
||
"Cache-Control": "no-cache",
|
||
"Cookie": "locale=zh-CN",
|
||
"Referer": "https://enka.network/",
|
||
"User-Agent": (
|
||
"Mozilla/5.0 (Linux; Android 12; Nexus 5) "
|
||
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
||
"Chrome/102.0.0.0 Mobile Safari/537.36"
|
||
),
|
||
},
|
||
timeout=20.0,
|
||
)
|
||
resJson = res.json()
|
||
break
|
||
except (HTTPError, json.decoder.JSONDecodeError) as e:
|
||
if idx == len(enkaMirrors) - 1:
|
||
logger.opt(exception=e).error("面板数据接口无法访问或返回错误")
|
||
return {"error": f"[{e.__class__.__name__}] 暂时无法访问面板数据接口.."}
|
||
else:
|
||
logger.info(f"从 {root} 获取面板失败，正在自动切换镜像重试...")
|
||
if not resJson.get("playerInfo"):
|
||
return {"error": f"玩家 {uid} 返回信息不全，接口可能正在维护.."}
|
||
if not resJson.get("avatarInfoList"):
|
||
return {"error": f"玩家 {uid} 的角色展柜详细数据已隐藏！"}
|
||
if not resJson["playerInfo"].get("showAvatarInfoList"):
|
||
return {"error": f"玩家 {uid} 的角色展柜内还没有角色哦！"}
|
||
return resJson
|
||
|
||
|
||
async def queryDamageApi(body: Dict, mode: Literal["single", "team"] = "single") -> Dict:
|
||
"""
|
||
角色伤害计算数据请求（提瓦特小助手）
|
||
|
||
* ``param body: Dict`` 查询角色数据
|
||
* ``param mode: Literal["single", "team"] = "single"`` 查询接口类型，默认请求角色伤害接口，传入 ``"team"`` 请求队伍伤害接口
|
||
- ``return: Dict`` 查询结果，出错时返回 ``{}``
|
||
"""
|
||
apiMap = {
|
||
"single": "https://api.lelaer.com/ys/getDamageResult.php",
|
||
"team": "https://api.lelaer.com/ys/getTeamResult.php",
|
||
}
|
||
async with AsyncClient() as client:
|
||
try:
|
||
res = await client.post(
|
||
apiMap[mode],
|
||
json=body,
|
||
headers={
|
||
"referer": "https://servicewechat.com/wx2ac9dce11213c3a8/192/page-frame.html",
|
||
"user-agent": (
|
||
"Mozilla/5.0 (Linux; Android 12; SM-G977N Build/SP1A.210812.016; wv) "
|
||
"AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/86.0.4240.99 "
|
||
"XWEB/4375 MMWEBSDK/20221011 Mobile Safari/537.36 MMWEBID/4357 "
|
||
"MicroMessenger/8.0.30.2244(0x28001E44) WeChat/arm64 Weixin GPVersion/1 "
|
||
"NetType/WIFI Language/zh_CN ABI/arm64 MiniProgramEnv/android"
|
||
),
|
||
},
|
||
)
|
||
return res.json()
|
||
except (HTTPError, json.decoder.JSONDecodeError) as e:
|
||
logger.opt(exception=e).error("提瓦特小助手接口无法访问或返回错误")
|
||
return {}
|
||
except Exception as e:
|
||
logger.opt(exception=e).error("提瓦特小助手接口错误")
|
||
return {}
|
||
|
||
|
||
async def getAvatarData(uid: str, char: str = "全部") -> Dict:
|
||
"""
|
||
角色数据获取（内部格式）
|
||
|
||
* ``param uid: str`` 查询用户 UID
|
||
* ``param char: str = "全部"`` 查询角色名
|
||
- ``return: Dict`` 查询结果。出错时返回 ``{"error": "错误信息"}``
|
||
"""
|
||
# 总是先读取一遍缓存
|
||
cache = LOCAL_DIR / "cache" / f"{uid}.json"
|
||
if cache.exists():
|
||
cacheData = json.loads(cache.read_text(encoding="utf-8"))
|
||
nextQueryTime = cacheData["next"]
|
||
else:
|
||
cacheData, nextQueryTime = {}, 0
|
||
|
||
if int(time()) <= nextQueryTime:
|
||
logger.info("UID{} 的角色展柜数据刷新冷却还有 {} 秒！".format(uid, nextQueryTime - int(time())))
|
||
else:
|
||
logger.info(f"UID{uid} 的角色展柜数据正在刷新！")
|
||
newData = await queryPanelApi(uid)
|
||
if not cacheData and newData.get("error"):
|
||
return newData
|
||
elif not newData.get("error"):
|
||
avatarsCache = {str(x["id"]): x for x in cacheData.get("avatars", [])}
|
||
now, wait4Dmg, avatars, avatarIdsNew = int(time()), {}, [], []
|
||
for newAvatar in newData["avatarInfoList"]:
|
||
if newAvatar["avatarId"] in [10000005, 10000007]:
|
||
logger.info("旅行者面板查询暂未支持！")
|
||
continue
|
||
tmp, gotDmg = await transFromEnka(newAvatar, now), False
|
||
|
||
if str(tmp["id"]) in avatarsCache:
|
||
# 保留旧的伤害计算数据
|
||
avatarsCache[str(tmp["id"])].pop("time")
|
||
cacheDmg = avatarsCache[str(tmp["id"])].pop("damage")
|
||
nowStat = {
|
||
k: v for k, v in tmp.items() if k not in ["damage", "time"]
|
||
}
|
||
if cacheDmg and avatarsCache[str(tmp["id"])] == nowStat:
|
||
logger.info(f"UID{uid} 的 {tmp['name']} 伤害计算结果无需刷新！")
|
||
tmp["damage"], gotDmg = cacheDmg, True
|
||
else:
|
||
logger.info(
|
||
"UID{} 的 {} 数据发生变化：\n{}\n{}".format(
|
||
uid, tmp["name"], avatarsCache[str(tmp["id"])], nowStat
|
||
)
|
||
)
|
||
avatarIdsNew.append(tmp["id"])
|
||
avatars.append(tmp)
|
||
if not gotDmg:
|
||
wait4Dmg[str(len(avatars) - 1)] = tmp
|
||
|
||
if wait4Dmg:
|
||
logger.info(
|
||
"正在为 UID{} 的 {} 重新请求伤害计算接口".format(
|
||
uid, "/".join(f"[{aI}]{a['name']}" for aI, a in wait4Dmg.items())
|
||
)
|
||
)
|
||
# 深拷贝避免转换对上下文中的 avatars 产生影响
|
||
wtf = deepcopy([a for _, a in wait4Dmg.items()])
|
||
teyvatBody = await transToTeyvat(wtf, uid)
|
||
teyvatRaw = await queryDamageApi(teyvatBody)
|
||
if teyvatRaw.get("code", "x") != 200 or len(wait4Dmg) != len(
|
||
teyvatRaw.get("result", [])
|
||
):
|
||
logger.error(
|
||
(
|
||
f"UID{uid} 的 {len(wait4Dmg)} 位角色伤害计算请求失败！"
|
||
f"\n>>>> [提瓦特返回] {teyvatRaw}"
|
||
)
|
||
)
|
||
else:
|
||
for dmgIdx, dmgData in enumerate(teyvatRaw.get("result", [])):
|
||
aIdx = int(list(wait4Dmg.keys())[dmgIdx])
|
||
avatars[aIdx]["damage"] = await simplDamageRes(dmgData)
|
||
|
||
cacheData["avatars"] = [
|
||
*avatars,
|
||
*[
|
||
aData
|
||
for _, aData in avatarsCache.items()
|
||
if aData["id"] not in avatarIdsNew
|
||
],
|
||
]
|
||
cacheData["next"] = now + newData["ttl"]
|
||
cache.write_text(
|
||
json.dumps(cacheData, ensure_ascii=False, indent=2), encoding="utf-8"
|
||
)
|
||
|
||
# 获取所需角色数据
|
||
if char == "全部":
|
||
return cacheData
|
||
searchRes = [x for x in cacheData["avatars"] if x["name"] == char]
|
||
return (
|
||
{
|
||
"error": "玩家 {} 游戏内展柜中的 {} 位角色中没有 {}！".format(
|
||
uid, len(cacheData["avatars"]), char
|
||
)
|
||
}
|
||
if not searchRes
|
||
else searchRes[0]
|
||
)
|
||
|
||
|
||
async def getPanel(uid: str, char: str = "全部") -> Union[bytes, str]:
|
||
"""
|
||
原神游戏内角色展柜消息生成入口
|
||
|
||
* ``param uid: str`` 查询用户 UID
|
||
* ``param char: str = "全部"`` 查询角色
|
||
- ``return: Union[bytes, str]`` 查询结果。一般返回图片字节，出错时返回错误信息字符串
|
||
"""
|
||
# 获取面板数据
|
||
data = await getAvatarData(uid, char)
|
||
if data.get("error"):
|
||
return data["error"]
|
||
if char == "全部":
|
||
return "成功获取了 UID{} 的{}等 {} 位角色数据！".format(
|
||
uid, "、".join(a["name"] for a in data["avatars"]), len(data["avatars"])
|
||
)
|
||
|
||
# 图标下载任务
|
||
dlTasks = [
|
||
download(data["icon"], local=char),
|
||
download(data["gachaAvatarImg"], local=char),
|
||
*[download(sData["icon"], local=char) for _, sData in data["skills"].items()],
|
||
*[download(conData["icon"], local=char) for conData in data["consts"]],
|
||
download(data["weapon"]["icon"], local="weapon"),
|
||
*[download(relicData["icon"], local="artifacts") for relicData in data["relics"]],
|
||
]
|
||
await asyncio.gather(*dlTasks)
|
||
dlTasks.clear()
|
||
|
||
# 渲染截图
|
||
data["fightProp"] = await simplFightProp(
|
||
data["fightProp"], data["baseProp"], char, data["element"]
|
||
)
|
||
htmlBase = str(LOCAL_DIR.resolve())
|
||
return await template_to_pic(
|
||
template_path=htmlBase,
|
||
template_name=f"panel-{TPL_VERSION}.html",
|
||
templates={"css": TPL_VERSION, "uid": uid, "data": data},
|
||
pages={
|
||
"device_scale_factor": SCALE_FACTOR,
|
||
"viewport": {"width": 600, "height": 300},
|
||
"base_url": f"file://{htmlBase}",
|
||
},
|
||
wait=2,
|
||
)
|
||
|
||
|
||
async def getTeam(uid: str, chars: List[str] = []) -> Union[bytes, str]:
|
||
"""
|
||
队伍伤害消息生成入口
|
||
|
||
* ``param uid: str`` 查询用户 UID
|
||
* ``param chars: List[str] = []`` 查询角色，为空默认数据中前四个
|
||
- ``return: Union[bytes, str]`` 查询结果。一般返回图片字节，出错时返回错误信息字符串
|
||
"""
|
||
# 获取面板数据
|
||
data = await getAvatarData(uid, "全部")
|
||
if data.get("error"):
|
||
return data["error"]
|
||
|
||
if chars:
|
||
extract = [a for a in data["avatars"] if a["name"] in chars]
|
||
if len(extract) != len(chars):
|
||
gotThis = [a["name"] for a in extract]
|
||
return "玩家 {} 的最新数据中未发现{}！".format(
|
||
uid, "、".join(c for c in chars if c not in gotThis)
|
||
)
|
||
elif len(data["avatars"]) >= 4:
|
||
extract = data["avatars"][:4]
|
||
logger.info(
|
||
"UID{} 未指定队伍，自动选择面板中前 4 位进行计算：{} ...".format(
|
||
uid, "、".join(a["name"] for a in extract)
|
||
)
|
||
)
|
||
else:
|
||
return f"玩家 {uid} 的面板数据甚至不足以组成一支队伍呢！"
|
||
|
||
# 图片下载任务
|
||
for tmp in extract:
|
||
dlTasks = [
|
||
download(tmp["icon"], local=tmp["name"]),
|
||
*[
|
||
download(sData["icon"], local=tmp["name"])
|
||
for _, sData in tmp["skills"].items()
|
||
],
|
||
download(tmp["weapon"]["icon"], local="weapon"),
|
||
*[
|
||
download(
|
||
f"UI_RelicIcon_{relicData['icon'].split('_')[-2]}_4",
|
||
local="artifacts",
|
||
)
|
||
for relicData in tmp["relics"]
|
||
],
|
||
]
|
||
await asyncio.gather(*dlTasks)
|
||
dlTasks.clear()
|
||
|
||
teyvatBody = await transToTeyvat(deepcopy(extract), uid)
|
||
teyvatRaw = await queryDamageApi(teyvatBody, "team")
|
||
if teyvatRaw.get("code", "x") != 200 or not teyvatRaw.get("result"):
|
||
logger.error(
|
||
(f"UID{uid} 的 {len(extract)} 位角色队伍伤害计算请求失败！" f"\n>>>> [提瓦特返回] {teyvatRaw}")
|
||
)
|
||
return f"玩家 {uid} 队伍伤害计算失败，接口可能发生变动！" if teyvatRaw else "啊哦，队伍伤害计算小程序状态异常！"
|
||
try:
|
||
data = await simplTeamDamageRes(
|
||
teyvatRaw["result"], {a["name"]: a for a in extract}
|
||
)
|
||
except Exception as e:
|
||
logger.opt(exception=e).error("队伍伤害数据解析出错")
|
||
return f"[{e.__class__.__name__}] 队伍伤害数据解析出错咯"
|
||
|
||
htmlBase = str(LOCAL_DIR.resolve())
|
||
return await template_to_pic(
|
||
template_path=htmlBase,
|
||
template_name=f"team-{TPL_VERSION}.html",
|
||
templates={"css": TPL_VERSION, "data": data},
|
||
pages={
|
||
"device_scale_factor": SCALE_FACTOR,
|
||
"viewport": {"width": 600, "height": 300},
|
||
"base_url": f"file://{htmlBase}",
|
||
},
|
||
wait=2,
|
||
)
|