diff --git a/main.py b/main.py index decc4a5d37a55ecd49b465a6d5f55a7110ddf6b7..de17f6515a661db6e0a239efed9ef80bf7c367fd 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- from json import load from src.tree import TreeWalker, load_json, dump_json diff --git a/src/tree.py b/src/tree.py index 61b5b8950df71658adfa6243961eab5ca45a8da2..6a9e39fc518542f858f1f80635b35164b046e6ba 100644 --- a/src/tree.py +++ b/src/tree.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- import logging from genericpath import exists import json @@ -16,7 +17,7 @@ logger.addHandler(handler) def load_json(p): - with open(p, 'r') as f: + with open(p, 'r', encoding='utf-8') as f: return json.loads(f.read()) @@ -29,7 +30,7 @@ def dump_json(p, j, exist_ok=False, override=False): logger.error(f"{p} already exist") sys.exit(0) - with open(p, 'w+') as f: + with open(p, 'w+', encoding='utf-8') as f: f.write(json.dumps(j, indent=2, ensure_ascii=False))