LittlePaimon/utils/file_handler.py

from pathlib import Path
from PIL import Image
from typing import Union, Optional, Tuple
import json


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] = 'RGB'
):
    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)
    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))


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)