✨ support: skill and eidolon icons

fix_data.py

@@ -3,6 +3,7 @@ import asyncio
 from res_func.avatar import fix_avatar_config, fetch_text_map
 from res_func.light_cone import fix_light_cone_config
 from res_func.relic import fetch_relic_config
+from res_func.honkai_gg.avatar import get_all_avatars
 
 
 async def main():
@@ -10,6 +11,7 @@ async def main():
     await fix_avatar_config(text_map_data)
     await fix_light_cone_config()
     await fetch_relic_config(text_map_data)
+    await get_all_avatars()
 
 
 if __name__ == '__main__':

func/fetch_src.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 src_dir = Path("src")
 data_dir = Path("data")
-pic_lists = ["avatar_gacha", "destiny", "element"]
+pic_lists = ["avatar_gacha", "destiny", "element", "skill"]
 
 
 async def move_files():

res_func/honkai_gg/avatar.py

@@ -0,0 +1,50 @@
+import asyncio
+import aiofiles
+import re
+from typing import List
+
+import ujson
+from bs4 import BeautifulSoup
+from res_func.client import client
+from res_func.url import avatar_honkai_url, avatar_icon_honkai_url
+
+avatar_data = {}
+
+
+async def get_all_avatar() -> List[str]:
+    req = await client.get(avatar_honkai_url)
+    html = req.text
+    pattern = re.compile(r'href="/characters/(.*?)">')
+    result = pattern.findall(html)
+    return [f"{avatar_honkai_url}/{i}" for i in result if i != ""]
+
+
+async def get_single_avatar(url: str) -> None:
+    req = await client.get(url)
+    html = req.text
+    soup = BeautifulSoup(html, "lxml")
+    div = soup.find_all("div", {"class": "character_skills__FL3Dn"})[-1]
+    pattern = re.compile(r'skillicons/(.*?)/skillicon(.*?).webp')
+    result = pattern.findall(str(div))
+    if len(result) != 6:
+        print(f"{url} 获取星魂图片失败")
+        return
+    urls = [f"{avatar_icon_honkai_url}/{i[0]}/skillicon{i[1]}.webp" for i in result]
+    avatar_data[result[0][0]] = urls
+
+
+async def dump_icons():
+    final_data = dict(sorted(avatar_data.items(), key=lambda x: x[0]))
+    async with aiofiles.open("data/avatar_eidolon_icons.json", "w", encoding="utf-8") as f:
+        await f.write(ujson.dumps(final_data, indent=4, ensure_ascii=False))
+
+
+async def get_all_avatars() -> None:
+    print("开始获取星魂图片")
+    urls = await get_all_avatar()
+    tasks = []
+    for url in urls:
+        tasks.append(get_single_avatar(url))
+    await asyncio.gather(*tasks)
+    await dump_icons()
+    print("获取星魂图片成功")

res_func/url.py

@@ -12,3 +12,6 @@ relic_set_config = f"{base_data_url}ExcelOutput/RelicSetConfig.json"
 base_station_url = "https://starrailstation.com"
 avatar_url = f"{base_station_url}/cn/characters"
 light_cone_url = f"{base_station_url}/cn/equipment"
+base_honkai_url = "https://honkai.gg"
+avatar_honkai_url = f"{base_honkai_url}/cn/characters"
+avatar_icon_honkai_url = f"{base_honkai_url}/images/spriteoutput/skillicons"