xtaodada/Enka_Panel_Bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
fc253c8422
Enka_Panel_Bot/gspanel/__utils__.py

222 lines
9.0 KiB
Python
Raw Normal View History

update
2022-09-03 14:50:36 +00:00
import asyncio
import json
from pathlib import Path
from typing import Optional, Union
from httpx import AsyncClient
from playwright.async_api import Browser, async_playwright
GROW_VALUE = { # 词条成长值
"暴击率": 3.89,
"暴击伤害": 7.77,
"元素精通": 23.31,
"攻击力百分比": 5.83,
"生命值百分比": 5.83,
"防御力百分比": 7.29,
"元素充能效率": 6.48,
"元素伤害加成": 5.825,
"物理伤害加成": 7.288,
"治疗加成": 4.487,
}
MAIN_AFFIXS = { # 可能的主词条
"3": "攻击力百分比,防御力百分比,生命值百分比,元素精通,元素充能效率".split(","), # EQUIP_SHOES
"4": "攻击力百分比,防御力百分比,生命值百分比,元素精通,元素伤害加成,物理伤害加成".split(","), # EQUIP_RING
"5": "攻击力百分比,防御力百分比,生命值百分比,元素精通,治疗加成,暴击率,暴击伤害".split(","), # EQUIP_DRESS
}
SUB_AFFIXS = "攻击力,攻击力百分比,防御力,防御力百分比,生命值,生命值百分比,元素精通,元素充能效率,暴击率,暴击伤害".split(",")
# STAR = {"QUALITY_ORANGE": 5, "QUALITY_PURPLE": 4}
ELEM = {
"Fire": "火",
"Water": "水",
"Wind": "风",
"Electric": "雷",
"Grass": "草",
"Ice": "冰",
"Rock": "岩",
}
POS = ["EQUIP_BRACER", "EQUIP_NECKLACE", "EQUIP_SHOES", "EQUIP_RING", "EQUIP_DRESS"]
SKILL = {"1": "a", "2": "e", "9": "q"}
DMG = {
"40": "火",
"41": "雷",
"42": "水",
"43": "草",
"44": "风",
"45": "岩",
"46": "冰",
}
PROP = {
"FIGHT_PROP_BASE_ATTACK": "基础攻击力",
"FIGHT_PROP_HP": "生命值",
"FIGHT_PROP_ATTACK": "攻击力",
"FIGHT_PROP_DEFENSE": "防御力",
"FIGHT_PROP_HP_PERCENT": "生命值百分比",
"FIGHT_PROP_ATTACK_PERCENT": "攻击力百分比",
"FIGHT_PROP_DEFENSE_PERCENT": "防御力百分比",
"FIGHT_PROP_CRITICAL": "暴击率",
"FIGHT_PROP_CRITICAL_HURT": "暴击伤害",
"FIGHT_PROP_CHARGE_EFFICIENCY": "元素充能效率",
"FIGHT_PROP_HEAL_ADD": "治疗加成",
"FIGHT_PROP_ELEMENT_MASTERY": "元素精通",
"FIGHT_PROP_PHYSICAL_ADD_HURT": "物理伤害加成",
"FIGHT_PROP_FIRE_ADD_HURT": "火元素伤害加成",
"FIGHT_PROP_ELEC_ADD_HURT": "雷元素伤害加成",
"FIGHT_PROP_WATER_ADD_HURT": "水元素伤害加成",
"FIGHT_PROP_WIND_ADD_HURT": "风元素伤害加成",
"FIGHT_PROP_ICE_ADD_HURT": "冰元素伤害加成",
"FIGHT_PROP_ROCK_ADD_HURT": "岩元素伤害加成",
}
_browser: Optional[Browser] = None
EXPIRE_SEC = 60 * 5
LOCAL_DIR = Path("resources")
if not (LOCAL_DIR / "cache").exists():
(LOCAL_DIR / "cache").mkdir(parents=True, exist_ok=True)
def kStr(prop: str) -> str:
"""转换词条名称为简短形式"""
return (
prop.replace("百分比", "")
.replace("元素充能", "充能")
.replace("元素伤害", "伤")
.replace("物理伤害", "物伤")
)
def vStr(prop: str, value: Union[int, float]) -> str:
"""转换词条数值为字符串形式"""
if prop in {"生命值", "攻击力", "防御力", "元素精通"}:
return str(value)
else:
return f"{str(round(value, 1))}%"
async def initBrowser(**kwargs) -> Optional[Browser]:
global _browser
browser = await async_playwright().start()
try:
_browser = await browser.chromium.launch(**kwargs)
return _browser
except Exception as e:
print(f"启动 Chromium 发生错误 {type(e)}：{e}")
return None
async def getBrowser(**kwargs) -> Optional[Browser]:
return _browser or await initBrowser(**kwargs)
async def fetchInitRes() -> None:
"""
插件初始化资源下载，通过阿里云 CDN 获取 Enka.Network API 提供的 JSON 文件、HTML 模板资源文件、角色词条权重配置等
- https://raw.githubusercontent.com/EnkaNetwork/API-docs/master/store/loc.json
- https://raw.githubusercontent.com/Dimbreath/GenshinData/master/TextMap/TextMapCHS.json
- https://raw.githubusercontent.com/EnkaNetwork/API-docs/master/store/characters.json
- https://raw.githubusercontent.com/monsterxcn/nonebot_plugin_gspanel/main/data/gspanel/template.json
- https://raw.githubusercontent.com/monsterxcn/nonebot_plugin_gspanel/main/data/gspanel/calc-rule.json
"""
print("正在检查面板插件所需资源...")
# 仅首次启用插件下载的文件
initRes = [
"https://cdn.monsterx.cn/bot/gspanel/font/华文中宋.TTF",
"https://cdn.monsterx.cn/bot/gspanel/font/HYWH-65W.ttf",
"https://cdn.monsterx.cn/bot/gspanel/font/NZBZ.ttf",
"https://cdn.monsterx.cn/bot/gspanel/font/tttgbnumber.ttf",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-anemo.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-cryo.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-dendro.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-electro.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-geo.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-hydro.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/bg-pyro.jpg",
"https://cdn.monsterx.cn/bot/gspanel/imgs/card-bg.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/star.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-anemo.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-cryo.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-dendro.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-electro.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-geo.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-hydro.png",
"https://cdn.monsterx.cn/bot/gspanel/imgs/talent-pyro.png",
"https://cdn.monsterx.cn/bot/gspanel/style.css",
"https://cdn.monsterx.cn/bot/gspanel/tpl.html",
]
tasks = []
for r in initRes:
d = r.replace("https://cdn.monsterx.cn/bot/gspanel/", "").split("/")[0]
tasks.append(download(r, local=("" if "." in d else d)))
await asyncio.gather(*tasks)
tasks.clear()
print("首次启用所需资源检查完毕..")
# 总是尝试更新的文件
urls = [
# "https://cdn.monsterx.cn/bot/gapanel/loc.json", # 仅包含 zh-CN 语言
"https://cdn.monsterx.cn/bot/gspanel/TextMapCHS.json",
"https://cdn.monsterx.cn/bot/gspanel/characters.json",
"https://cdn.monsterx.cn/bot/gspanel/calc-rule.json",
]
tmp = {"0": {}, "1": {}, "2": {}}
async with AsyncClient(verify=False) as client:
for idx, url in enumerate(urls):
tmp[str(idx)] = (await client.get(url)).json()
(LOCAL_DIR / url.split("/")[-1]).write_text(
json.dumps(tmp[str(idx)], ensure_ascii=False, indent=2),
encoding="utf-8",
)
# 额外生成一份 {"中文名": "8 位角色 ID", ...} 配置
name2id, unknownCnt = {}, 1
for charId in tmp["1"]: # characters.json
if not tmp["1"][charId].get("NameTextMapHash"):
# 10000005-502 10000005-503 10000005-505
# 10000007-702 10000007-703 10000007-705
continue
nameTextMapHash = tmp["1"][charId]["NameTextMapHash"] # type: int
nameCn = tmp["0"].get(str(nameTextMapHash), f"未知角色{unknownCnt}")
name2id[nameCn] = charId
if nameCn == f"未知角色{unknownCnt}":
unknownCnt += 1
(LOCAL_DIR / "name2id.json").write_text(
json.dumps(name2id, ensure_ascii=False, indent=2), encoding="utf-8"
)
print("面板插件所需资源检查完毕！")
async def download(url: str, local: Union[Path, str] = "") -> Union[Path, None]:
"""
一般文件下载，通常是即用即下的角色命座图片、技能图片、抽卡大图、圣遗物图片等
* ``param url: str`` 指定下载链接
* ``param local: Union[Path, str] = ""`` 指定本地目标路径，传入类型为 ``Path`` 时视为保存文件完整路径，传入类型为 ``str`` 时视为保存文件子文件夹名（默认下载至插件资源根目录）
- ``return: Union[Path, None]`` 本地文件地址，出错时返回空
"""
if not isinstance(local, Path):
d = (LOCAL_DIR / local) if local else LOCAL_DIR
if not d.exists():
d.mkdir(parents=True, exist_ok=True)
f = d / url.split("/")[-1]
else:
if not local.parent.exists():
local.parent.mkdir(parents=True, exist_ok=True)
f = local
# 本地文件存在时便不再下载，JSON 文件除外
if f.exists() and ".json" not in f.name:
return f
client = AsyncClient()
retryCnt = 3
while retryCnt:
try:
async with client.stream(
"GET", url, headers={"user-agent": "NoneBot-GsPanel"}
) as res:
with open(f, "wb") as fb:
async for chunk in res.aiter_bytes():
fb.write(chunk)
return f
except Exception as e:
print(f"面板资源 {f.name} 下载出错 {type(e)}：{e}")
retryCnt -= 1
await asyncio.sleep(2)
return None
Reference in New Issue Copy Permalink