diff data_manager/make_json.py @ 0:d5d803e654dd draft

Uploaded
author devteam
date Wed, 22 Apr 2015 09:11:37 -0400
parents
children 6f0294046f41
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/data_manager/make_json.py	Wed Apr 22 09:11:37 2015 -0400
@@ -0,0 +1,28 @@
+import json
+import argparse
+import os
+import shutil
+
+parser = argparse.ArgumentParser(description='Create data manager json.')
+parser.add_argument('--db', dest='database', action='store', help='Database name')
+parser.add_argument('--out', dest='output', action='store', help='JSON filename')
+
+args = parser.parse_args()
+
+def main(args):
+    data_manager_entry = {}
+    data_manager_entry['value'] = args.database.lower()
+    data_manager_entry['name'] = args.database
+    data_manager_entry['path'] = '.'
+    data_manager_json = dict(data_tables=dict(kraken_databases=data_manager_entry))
+    params = json.loads(open(args.output).read())
+    target_directory = params['output_data'][0]['extra_files_path']
+    os.mkdir(target_directory)
+    output_path = os.path.join(os.getcwd(), 'kraken-database')
+    for filename in os.listdir(output_path):
+        shutil.move(os.path.join(output_path, filename), target_directory)
+    file(args.output, 'w').write(json.dumps(data_manager_json))
+
+if __name__ == '__main__':
+    main(args)
+