PaiGramTeam/MibooGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/MibooGram.git synced 2025-01-09 15:02:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix avatar icon assets

Browse Source
This commit is contained in:
xtaodada 2024-12-18 17:10:21 +08:00
parent 63d7a5d6b5
commit 585651f7f9
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
core/dependence/assets.py
View File

@ -26,6 +26,15 @@ DATA_MAP = {
} }
def choose_path_by_url(url: str, png_path: Path, webp_path: Path) -> Path:
ext = url.split(".")[-1].lower()
if ext == "png":
return png_path
if ext == "webp":
return webp_path
return png_path
class AssetsServiceError(Exception): class AssetsServiceError(Exception):
pass pass
@ -88,18 +97,15 @@ class _AvatarAssets(_AssetsService):
for icon in self.data: for icon in self.data:
base_path = self.path / f"{icon.id}" base_path = self.path / f"{icon.id}"
base_path.mkdir(exist_ok=True, parents=True) base_path.mkdir(exist_ok=True, parents=True)
gacha_path = base_path / "gacha.png"
icon_path = base_path / "icon.png" def _get_path(name: str) -> tuple[Path, Path]:
square_path = base_path / "square.png" path = base_path / f"{name}.png"
normal_path = base_path / "normal.png" return path, path.with_suffix(".webp")
if not gacha_path.exists():
tasks.append(self._download(icon.gacha, gacha_path)) for i in (("gacha", icon.gacha), ("icon", icon.icon_), ("square", icon.square), ("normal", icon.normal)):
if not icon_path.exists(): png_path, webp_path = _get_path(i[0])
tasks.append(self._download(icon.icon_, icon_path)) if not png_path.exists() and not webp_path.exists() and i[1]:
if not square_path.exists(): tasks.append(self._download(i[1], choose_path_by_url(i[1], png_path, webp_path)))
tasks.append(self._download(icon.square, square_path))
if not normal_path.exists():
tasks.append(self._download(icon.normal, normal_path))
if len(tasks) >= 100: if len(tasks) >= 100:
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
@ -108,10 +114,16 @@ class _AvatarAssets(_AssetsService):
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
logger.info("角色素材图标初始化完成") logger.info("角色素材图标初始化完成")
def get_path(self, icon: Avatar, name: str, ext: str = "png") -> Path: def get_path(self, icon: Avatar, name: str) -> Path:
path = self.path / f"{icon.id}" path = self.path / f"{icon.id}"
path.mkdir(exist_ok=True, parents=True) path.mkdir(exist_ok=True, parents=True)
return path / f"{name}.{ext}" p1 = path / f"{name}.png"
if p1.exists():
return p1
p2 = path / f"{name}.webp"
if p2.exists():
return p2
return p1
def get_by_id(self, id_: int) -> Optional[Avatar]: def get_by_id(self, id_: int) -> Optional[Avatar]:
return self.id_map.get(id_, None) return self.id_map.get(id_, None)