diff --git a/scripts/ML/JSONL.py b/scripts/ML/JSONL.py index d191ee5255d709b2f90ae4ee6ead8edc86a7e9ef..07b2aafb8075e0f12a1332b9bebaf1b9abcefbcb 100644 --- a/scripts/ML/JSONL.py +++ b/scripts/ML/JSONL.py @@ -1,6 +1,22 @@ import json def load(file_path): - with open(file_path, 'r') as input_file: - for line in input_file.readlines(): + if type(file_path) == str: + with open(file_path, 'r') as input_file: + for line in input_file.readlines(): + yield json.loads(line) + else: + for line in file_path.readlines(): yield json.loads(line) + +def save(file_path, objects): + if type(file_path) == str: + with open(file_path, 'w') as output_file: + saveObjects(output_file, objects) + else: + saveObjects(file_path, objects) + +def saveObjects(output_file, objects): + for obj in objects: + json.dump(obj, output_file) + print(file=output_file)