Add publisher function

caltechlibrary · Mar 5, 2024 · 69fd83d · 69fd83d
1 parent 1e89efc
commit 69fd83d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/ames/harvesters/__init__.py b/ames/harvesters/__init__.py
@@ -16,3 +16,4 @@
 from .caltechauthors import get_records_from_date
 from .caltechauthors import get_request_comments
 from .caltechauthors import get_request_id_title
+from .caltechauthors import get_publisher
diff --git a/ames/harvesters/caltechauthors.py b/ames/harvesters/caltechauthors.py
@@ -62,6 +62,22 @@ def get_request_comments(token, request):
     return cleaned
 
 
+def get_publisher(token, record, test=False, draft=True):
+    if test:
+        url = "https://authors.caltechlibrary.dev/api/records"
+    else:
+        url = "https://authors.library.caltech.edu/api/records"
+    url = url + "/" + record
+    headers = {
+        "Authorization": "Bearer %s" % token,
+        "Content-type": "application/json",
+    }
+    if draft:
+        url = url + "/draft"
+    response = requests.get(url, headers=headers)
+    return response.json()["metadata"]["publisher"]
+
+
 def get_author_records(token, author_identifier, year=None, test=False):
     if test:
         url = "https://authors.caltechlibrary.dev/api/records"