Add authors to get_metadata and edit example

Author: tmorrell

caltechdata_api/get_metadata.py

@@ -8,20 +8,38 @@
 
 
 def get_metadata(
-    idv, production=True, validate=True, emails=False, schema="43", token=False
+    idv,
+    production=True,
+    validate=True,
+    emails=False,
+    schema="43",
+    token=False,
+    authors=False,
 ):
     # Returns just DataCite metadata or DataCite metadata with emails
 
     if production == True:
-        url = "https://data.caltech.edu/api/records/"
+        if authors:
+            url = "https://authors.library.caltech.edu/api/records/"
+        else:
+            url = "https://data.caltech.edu/api/records/"
         verify = True
     else:
-        url = "https://data.caltechlibrary.dev/api/records/"
+        if authors:
+            url = "https://authors.caltechlibrary.dev/api/records/"
+        else:
+            url = "https://data.caltechlibrary.dev/api/records/"
         verify = True
 
-    headers = {
-        "accept": "application/vnd.datacite.datacite+json",
-    }
+    if authors:
+        headers = {
+            "accept": "application/json",
+        }
+        validate = False
+    else:
+        headers = {
+            "accept": "application/vnd.datacite.datacite+json",
+        }
 
     if token:
         headers["Authorization"] = "Bearer %s" % token
@@ -58,6 +76,7 @@ def get_metadata(
         help="The CaltechDATA ID for each record of interest",
     )
     parser.add_argument("-test", dest="production", action="store_false")
+    parser.add_argument("-authors", dest="authors", action="store_true")
     parser.add_argument("-xml", dest="save_xml", action="store_true")
     parser.add_argument(
         "-skip_validate",
@@ -71,14 +90,17 @@ def get_metadata(
 
     production = args.production
     schema = args.schema
+    authors = args.authors
     skip_validate = args.skip_validate
     if skip_validate:
         validate = False
     else:
         validate = True
 
     for idv in args.ids:
-        metadata = get_metadata(idv, production, validate, schema)
+        metadata = get_metadata(
+            idv, production, validate, schema=schema, authors=authors
+        )
         outfile = open(str(idv) + ".json", "w")
         outfile.write(json.dumps(metadata, indent=4))
         outfile.close()

edit.py

@@ -15,6 +15,7 @@
 parser.add_argument("-fnames", nargs="*", help="New Files")
 parser.add_argument("-flinks", nargs="*", help="New File Links")
 parser.add_argument("-schema", default="43", help="Metadata Schema")
+parser.add_argument("-authors", action="store_true", help="Edit CaltechAUTHORS")
 args = parser.parse_args()
 
 # Get access token set as environment variable with source token.bash
@@ -26,7 +27,7 @@
 else:
     metadata = {}
 
-production = False
+production = True
 publish = True
 
 response = caltechdata_edit(
@@ -38,5 +39,6 @@
     args.schema,
     publish,
     args.flinks,
+    authors=args.authors,
 )
 print(response)