david-mccullars/ruby-splitta

View on GitHub
bin/pickle2json

Summary

Maintainability
Test Coverage
#!/usr/bin/env python
import gzip
import json
import pickle
import sys

if len(sys.argv) < 2:
  sys.exit('Usage: pickle2json <GZIP PICKLE FILE>')

with gzip.open(sys.argv[1], 'rb') as f:
    data = pickle.load(f)
if type(data) is dict and type(next(iter(data))) is tuple:
    data = [[[k[0], k[1]], v] for k, v in data.items()]
print(json.dumps(data))