| 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 |
|
|
|
|
|
|
