(original) (raw)

import pickle import pickletools def get_obj(): import collections return {'a': collections.OrderedDict(), 'b': 1} if __name__ == '__main__': import sys if sys.argv[1] == 'dump': p = pickle.dumps(get_obj(), protocol=2) pickletools.dis(p) with open('test.pickle', 'w+b') as fp: fp.write(p) elif sys.argv[1] == 'load': with open('test.pickle', 'r+b') as fp: pickle.load(fp) print('load succeeded') else: print("run with either load or dump as an argument")