diff options
Diffstat (limited to 'Lib/pickle.py')
| -rw-r--r-- | Lib/pickle.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Lib/pickle.py b/Lib/pickle.py index efcdcbe..4fa3632 100644 --- a/Lib/pickle.py +++ b/Lib/pickle.py @@ -1907,7 +1907,7 @@ except ImportError: dump, dumps, load, loads = _dump, _dumps, _load, _loads -if __name__ == "__main__": +def _main(args=None): import argparse import pprint parser = argparse.ArgumentParser( @@ -1915,7 +1915,7 @@ if __name__ == "__main__": parser.add_argument( 'pickle_file', nargs='+', help='the pickle file') - args = parser.parse_args() + args = parser.parse_args(args) for fn in args.pickle_file: if fn == '-': obj = load(sys.stdin.buffer) @@ -1923,3 +1923,7 @@ if __name__ == "__main__": with open(fn, 'rb') as f: obj = load(f) pprint.pprint(obj) + + +if __name__ == "__main__": + _main() |