diff alveo_item_list_importer.py @ 0:bfe39bd252df draft

planemo upload commit 5de43e6a614de2a1b2065bc63823ecc9854ebb32-dirty

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/alveo_item_list_importer.py	Mon Jul 18 23:49:40 2016 -0400
@@ -0,0 +1,40 @@
+from __future__ import print_function
+import json
+import argparse
+import pyalveo
+import sys
+
+API_URL = 'https://app.alveo.edu.au' # TODO: export constants to a separate module
+
+
+def parser():
+    parser = argparse.ArgumentParser(description="Retrieves Alveo Item Lists")
+    parser.add_argument('--api_key', required=True, action="store", type=str, help="Alveo API key")
+    parser.add_argument('--output', required=True, action="store", type=str, help="Path to output file")
+    return parser.parse_args()
+
+# TODO: export common function to helper module
+def get_item_lists(api_key):
+    client = pyalveo.Client(api_key=api_key, api_url=API_URL)
+    return client.get_item_lists()
+
+def write_table(item_lists, filename):
+    with open(filename, 'w') as outfile:
+        for list_set in item_lists.itervalues():
+            for item_list in list_set:
+                outfile.write("%s (%d)\t%s\n" % (item_list['name'], item_list['num_items'], item_list['item_list_url']))
+
+def main():
+    args = parser()
+    try:
+        api_key = open(args.api_key, 'r').read().strip()
+        item_lists = get_item_lists(api_key)
+        if item_lists:
+            write_table(item_lists, args.output)
+    except Exception as e:
+        print("ERROR: " + str(e), file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    main()