Add accept function for community-submitted records

Author: tmorrell

caltechdata_api/__init__.py

@@ -4,7 +4,11 @@
     add_file_links,
     send_to_community,
 )
-from .caltechdata_edit import caltechdata_edit, caltechdata_unembargo
+from .caltechdata_edit import (
+    caltechdata_edit,
+    caltechdata_unembargo,
+    caltechdata_accept,
+)
 from .customize_schema import customize_schema
 from .get_metadata import get_metadata
 from .download_file import download_file, download_url

caltechdata_api/caltechdata_edit.py

@@ -15,6 +15,43 @@ def caltechdata_unembargo(token, ids, production=False):
     print("caltechdaua_unembargo is not yet re-implemented")
 
 
+def caltechdata_accept(ids, token=None, production=False):
+    # Accept a record into a community
+
+    # If no token is provided, get from RDMTOK environment variable
+    if not token:
+        token = os.environ["RDMTOK"]
+
+    if production == True:
+        url = "https://data.caltech.edu"
+        verify = True
+    else:
+        url = "https://data.caltechlibrary.dev"
+        verify = True
+
+    headers = {
+        "Authorization": "Bearer %s" % token,
+        "Content-type": "application/json",
+    }
+
+    for idv in ids:
+
+        result = requests.get(
+            url + "/api/records/" + idv + "/draft/review", headers=headers
+        )
+
+        accept_link = result.json()["links"]["actions"]["accept"]
+        data = comment = {
+            "payload": {
+                "content": "This record is accepted automatically with the CaltechDATA API",
+                "format": "html",
+            }
+        }
+        result = requests.post(accept_link, json=data, headers=headers)
+        if result.status_code != 200:
+            raise Exception(result.text)
+
+
 def caltechdata_edit(
     ids,
     metadata={},

caltechdata_api/customize_schema.py

@@ -13,7 +13,7 @@ def grid_to_ror(grid):
     # We manually handle some incorrect/redundant GRID Ids
     if grid == "grid.451078.f":
         ror = "https://ror.org/00hm6j694"
-    elif grid == 'grid.5805.8':
+    elif grid == "grid.5805.8":
         ror = "https://ror.org/02en5vm52"
     else:
         url = f"https://api.ror.org/organizations?query.advanced=external_ids.GRID.all:{grid}"
@@ -336,7 +336,7 @@ def customize_schema_rdm(json_record):
                     funder["id"] = ror
                     fund.pop("funderIdentifierType")
                 elif fund["funderIdentifierType"] == "GRID":
-                    #We need this temporarily to round-trip data
+                    # We need this temporarily to round-trip data
                     ror = grid_to_ror(fund.pop("funderIdentifier"))
                     funder["id"] = ror
                     fund.pop("funderIdentifierType")