Add xml output option and file editing example

Author: tmorrell

caltechdata_api/get_metadata.py

@@ -36,7 +36,8 @@ def get_metadata(idv,production=True):
     and returns DataCite-compatable metadata")
     parser.add_argument('ids', metavar='ID', type=int, nargs='+',\
     help='The CaltechDATA ID for each record of interest')
-    parser.add_argument('--test',dest='production', action='store_false')
+    parser.add_argument('-test',dest='production', action='store_false')
+    parser.add_argument('-xml',dest='save_xml', action='store_true')
 
     args = parser.parse_args()
 
@@ -48,3 +49,8 @@ def get_metadata(idv,production=True):
         outfile = open(str(idv)+'.json','w')
         outfile.write(json.dumps(metadata))
         outfile.close()
+        if args.save_xml == True:
+            xml = schema40.tostring(metadata)
+            outfile = open(str(idv)+'.xml','w',encoding='utf8')
+            outfile.write(xml)
+

edit.py

@@ -16,5 +16,7 @@
 metaf = open(args.json_file[0], 'r')
 metadata = json.load(metaf)
 
-response = caltechdata_edit(token, args.ids, metadata, args.fnames, {}, False)
+production = True
+
+response = caltechdata_edit(token, args.ids, metadata, args.fnames, {'pdf'}, production)
 print(response)

edit_files.py

@@ -0,0 +1,18 @@
+import argparse, os, json
+from caltechdata_api import caltechdata_edit
+
+parser = argparse.ArgumentParser(description=\
+        "Write files and a DataCite 4 standard json record\
+        to CaltechDATA repository")
+parser.add_argument('-ids', nargs='*', help='CaltechDATA IDs')
+parser.add_argument('-fnames', nargs='*', help='New Files')
+parser.add_argument('-delete', nargs='*', help='Files To Delete')
+args = parser.parse_args()
+
+#Get access token from TIND sed as environment variable with source token.bash
+token = os.environ['TINDTOK']
+
+production = False
+
+response = caltechdata_edit(token, args.ids, {}, args.fnames, args.delete, production)
+print(response)