diff --git a/scripts/pickle_converter/readme.md b/scripts/pickle_converter/readme.md new file mode 100644 index 0000000..aed92e1 --- /dev/null +++ b/scripts/pickle_converter/readme.md @@ -0,0 +1,6 @@ +# .pickle to .txt or .json +This script converts .pickle files to .txt or .json files. + +## Usage +* Dependency: + * *None* diff --git a/scripts/pickle_converter/script.py b/scripts/pickle_converter/script.py new file mode 100644 index 0000000..4b23236 --- /dev/null +++ b/scripts/pickle_converter/script.py @@ -0,0 +1,20 @@ +import pickle +import pprint +import json + + +def convert_pickle(pickle_file="sample.pickle", target_file='txt'): + obj = pickle.load(open(pickle_file, "rb")) + + if target_file == 'txt': + with open("out.txt", "a") as f: + pprint.pprint(obj, stream=f) + + elif target_file == 'json': + json_obj = json.loads(json.dumps(obj, default=str)) + + with open('out.json', 'w', encoding='utf-8') as outfile: + json.dump(json_obj, outfile, ensure_ascii=False, indent=2) + + else: + print("please enter a valid doc type: 'txt', 'json'")