from fractions import Fraction
from pathlib import Path
from json import JSONEncoder
from .utils import *


def rmtree(path: Path):
    if path.is_file():
        path.unlink()
    else:
        for ch in path.iterdir():
            rmtree(ch)
        path.rmdir()


def safe_save(fpath, data, save_fun, rewrite=False, error_msg='File {fpath} exists! To rewite it use `--rewrite` flag', **kwargs):
    if not fpath.is_file() or rewrite:
        save_fun(str(fpath), data, **kwargs)
    else:
        raise FileExistsError(error_msg.format(fpath=fpath))


class FractionJSONEncoder(JSONEncoder):
    def default(self, o):
        if isinstance(o, Fraction):
            return {'Fraction': [o.numerator, o.denominator]}
        else:
            return o.__dict__


def fraction_from_json(json_object):
    if 'Fraction' in json_object:
        return Fraction(*json_object['Fraction'])
    return json_object