Add caltechdata_unembargo function

Author: tmorrell

caltechdata_api/__init__.py

@@ -2,6 +2,7 @@
 from .caltechdata_write import send_s3
 from .caltechdata_edit import caltechdata_add
 from .caltechdata_edit import caltechdata_edit
+from .caltechdata_edit import caltechdata_unembargo
 from .customize_schema import customize_schema
 from .decustomize_schema import decustomize_schema
 from .get_metadata import get_metadata

caltechdata_api/caltechdata_edit.py

@@ -4,6 +4,49 @@
 from caltechdata_api import send_s3
 
 
+def caltechdata_unembargo(token, ids, production=False):
+    """Unembargo files from a record. Needed bacause the delete API call
+    doesn't remove the embargo data"""
+
+    # Ensure ids is an arrary
+    if isinstance(ids, int):
+        ids = [str(ids)]
+    if isinstance(ids, str):
+        ids = [ids]
+
+    headers = {"Authorization": "Bearer %s" % token, "Content-type": "application/json"}
+
+    if production == True:
+        url = "https://data.caltech.edu/submit/api/edit/"
+        api_url = "https://data.caltech.edu/api/record/"
+    else:
+        url = "https://cd-sandbox.tind.io/submit/api/edit/"
+        api_url = "https://cd-sandbox.tind.io/api/record/"
+
+    for idv in ids:
+        update = []
+        c = session()
+        existing = c.get(api_url + idv)
+        file_info = existing.json()["metadata"]
+        if "electronic_location_and_access" in file_info:
+            for ex in file_info["electronic_location_and_access"]:
+                name = ex["electronic_name"][0]
+                fu = ex["uniform_resource_identifier"].split("/")[-2]
+                update.append({"id": fu, "embargo_status": "open"})
+
+        metadata = {
+            "id": idv,
+            "embargo_date": "2000-01-01",
+            "files": {"update": update},
+        }
+
+        dat = json.dumps({"record": metadata})
+
+        c = session()
+        response = c.post(url, headers=headers, data=dat)
+        return response.text
+
+
 def caltechdata_edit(token, ids, metadata={}, files={}, delete={}, production=False):
     """Including files will only replaces files if they have the same name
     The delete option will delete any existing files with a given file extension

caltechdata_api/caltechdata_write.py

@@ -52,14 +52,14 @@ def send_s3(filepath, token, production=False):
 
     c = session()
     response = c.post(url, files=form, headers=s3headers)
-    print(response)
+    # print(response)
     if response.text:
         raise Exception(response.text)
 
     # print(chkurl + "/" + bucket + "/" + key + "/")
     # print(headers)
     response = c.get(chkurl + "/" + bucket + "/" + key + "/", headers=headers)
-    print(response)
+    # print(response)
     md5 = response.json()["md5"]
     filename = filepath.split("/")[-1]
 

codemeta.json

@@ -6,7 +6,7 @@
   "codeRepository": "https://github.com/caltechlibrary/caltechdata_api",
   "issueTracker": "https://github.com/caltechlibrary/caltechdata_api/issues",
   "license": "https://data.caltech.edu/license",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "author": [
     {
       "@type": "Person",