Genshin-bots/StarRailUID
1
0
Fork 0
mirror of https://github.com/baiqwerdvd/StarRailUID.git synced 2025-05-06 11:43:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

⚡ 缓存图片

Browse Source
This commit is contained in:
MingxuanGame 2023-05-03 09:05:42 +08:00
parent 739e94ee51
commit 03a8b6b3bd
No known key found for this signature in database
GPG Key ID: 90C7EFA11DC3C2FF
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
StarRailUID/starrailuid_roleinfo/utils.py
View File

@ -3,9 +3,13 @@ from typing import List, TypeVar, Generator
from PIL import Image
from aiohttp import ClientSession
from gsuid_core.data_store import get_res_path
T = TypeVar("T")
ROLEINFO_PATH = get_res_path() / 'StarRailUID' / 'roleinfo'
ROLEINFO_PATH.mkdir(parents=True, exist_ok=True)
def wrap_list(lst: List[T], n: int) -> Generator[List[T], None, None]:
for i in range(0, len(lst), n):
@ -13,6 +17,14 @@ def wrap_list(lst: List[T], n: int) -> Generator[List[T], None, None]:
async def get_icon(url: str) -> Image.Image:
async with ClientSession() as client:
async with client.get(url) as resp:
return Image.open(BytesIO(await resp.read())).convert("RGBA")
name = url.split('/')[-1]
path = ROLEINFO_PATH / name
if (path).exists():
content = path.read_bytes()
else:
async with ClientSession() as client:
async with client.get(url) as resp:
content = await resp.read()
with open(path, "wb") as f:
f.write(content)
return Image.open(BytesIO(content)).convert("RGBA")