X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=tools%2Fsaveload.py;fp=tools%2Fsaveload.py;h=7f43a433d884cebdf048f6d8dd8c6718a23b8f01;hp=0000000000000000000000000000000000000000;hb=6c0dfe752994312ee58d307b383948bfeb2d6e2e;hpb=667281b828c8515e995c0000706157cee180fa08;ds=sidebyside diff --git a/tools/saveload.py b/tools/saveload.py new file mode 100644 index 0000000..7f43a43 --- /dev/null +++ b/tools/saveload.py @@ -0,0 +1,16 @@ +import pickle + +def save_obj(obj, name ): + with open('obj/'+ name + '.pkl', 'wb') as f: + pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL) + +def load_obj(name ): + with open('obj/' + name + '.pkl', 'r') as f: + return pickle.load(f) + + +def load_dict(name): + try: + return load_obj(name) + except: + return {}