Mercurial > repos > bebatut > merge_metaphlan_tables
comparison transform_json_to_pkl.py @ 0:b35463d95657 draft
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/metaphlan2/ commit b'3deaa3adf381c1f3f6134ef6a36074d443f7e3fc\n'-dirty
| author | bebatut |
|---|---|
| date | Tue, 28 Feb 2017 09:56:05 -0500 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:b35463d95657 |
|---|---|
| 1 #!/usr/bin/env python | |
| 2 # -*- coding: utf-8 -*- | |
| 3 | |
| 4 import argparse | |
| 5 import bz2 | |
| 6 import cPickle as pickle | |
| 7 import json | |
| 8 | |
| 9 | |
| 10 def transform_json_to_pkl(args): | |
| 11 with open(args.json_input, 'r') as json_file: | |
| 12 json_str = json_file.read() | |
| 13 metadata = json.loads(json_str) | |
| 14 | |
| 15 for marker in metadata["markers"]: | |
| 16 a_set = set(metadata["markers"][marker]["ext"]) | |
| 17 metadata["markers"][marker]["ext"] = a_set | |
| 18 | |
| 19 pkl_output = bz2.BZ2File(args.pkl_output, 'w') | |
| 20 pickle.dump(metadata, pkl_output, pickle.HIGHEST_PROTOCOL) | |
| 21 pkl_output.close() | |
| 22 | |
| 23 | |
| 24 if __name__ == '__main__': | |
| 25 parser = argparse.ArgumentParser() | |
| 26 parser.add_argument('--json_input', required=True) | |
| 27 parser.add_argument('--pkl_output', required=True) | |
| 28 args = parser.parse_args() | |
| 29 | |
| 30 transform_json_to_pkl(args) |