mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import json
|
|
from pathlib import Path
|
|
from typing import Union, Optional, Tuple
|
|
from ssl import SSLCertVerificationError
|
|
from PIL import Image
|
|
from utils import aiorequests
|
|
|
|
|
|
def load_image(
|
|
path: Union[Path, str],
|
|
*,
|
|
size: Optional[Union[Tuple[int, int], float]] = None,
|
|
crop: Optional[Tuple[int, int, int, int]] = None,
|
|
mode: Optional[str] = None,
|
|
):
|
|
img = Image.open(path)
|
|
if size:
|
|
if isinstance(size, float):
|
|
img = img.resize((int(img.size[0] * size), int(img.size[1] * size)), Image.ANTIALIAS)
|
|
elif isinstance(size, tuple):
|
|
img = img.resize(size, Image.ANTIALIAS)
|
|
if crop:
|
|
img = img.crop(crop)
|
|
if mode:
|
|
img = img.convert(mode)
|
|
return img
|
|
|
|
|
|
def load_json(file: str = None, path: Union[Path, str] = None, encoding: str = 'utf-8'):
|
|
if file and not path:
|
|
path = Path() / 'data' / 'LittlePaimon' / file
|
|
elif path:
|
|
if isinstance(path, str):
|
|
path = Path(path)
|
|
if not path.exists():
|
|
save_json({}, file, path, encoding)
|
|
return json.load(path.open('r', encoding=encoding))
|
|
|
|
|
|
async def load_json_from_url(url: str, encoding: str = 'utf-8', refresh: bool = False):
|
|
if 'static.cherishmoon.fun' in url:
|
|
url_path = Path() / 'data' / url.split('static.cherishmoon.fun/')[1]
|
|
if refresh or not url_path.exists():
|
|
try:
|
|
resp = await aiorequests.get(url)
|
|
except SSLCertVerificationError:
|
|
resp = await aiorequests.get(url.replace('https', 'http'))
|
|
save_json(resp.json(), path=url_path, encoding=encoding)
|
|
return resp.json()
|
|
else:
|
|
return load_json(path=url_path, encoding=encoding)
|
|
|
|
|
|
def save_json(data, file: str = None, path: Union[Path, str] = None, encoding: str = 'utf-8'):
|
|
if file and not path:
|
|
path = Path() / 'data' / 'LittlePaimon' / file
|
|
elif path:
|
|
if isinstance(path, str):
|
|
path = Path(path)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
json.dump(data, path.open('w', encoding=encoding), ensure_ascii=False, indent=4)
|