Add single record affiliation enhancement script

Author: tmorrell

ames/matchers/caltechauthors.py

@@ -563,7 +563,9 @@ def process_link_updates(input_csv):
     return results
 
 
-def add_authors_affiliations(record, token, dimensions_key, allowed_identifiers=None):
+def add_authors_affiliations(
+    record, token, dimensions_key, allowed_identifiers=None, ignore_mismatch=False
+):
     # Add dimensions affiliations to a record
 
     record_id = record["id"]
@@ -592,7 +594,7 @@ def add_authors_affiliations(record, token, dimensions_key, allowed_identifiers=
             publication = publication[0]
             dimensions_authors = publication.get("authors", [])
             existing_authors = record["metadata"]["creators"]
-            if len(dimensions_authors) == len(existing_authors):
+            if len(dimensions_authors) == len(existing_authors) or ignore_mismatch:
                 for position in range(len(dimensions_authors)):
                     author = existing_authors[position]
                     dimensions_author = dimensions_authors[position]

run_authors_affiliation_enhancement_singlerecord.py

@@ -0,0 +1,21 @@
+import os, sys, argparse, json
+from ames.matchers import add_authors_affiliations
+from caltechdata_api import get_metadata
+
+token = os.environ["CTATOK"]
+dimensions_key = os.environ["DIMKEY"]
+
+parser = argparse.ArgumentParser(
+    prog="authors_affitiliation_enhancement_singlerecord",
+    description="Adds affiliations from dimensions to single authors record, ignoring any mismatch in author numbers",
+)
+
+parser.add_argument(
+    "record_identifier", type=str, help="The rdm record identifier to enhance"
+)
+
+args = parser.parse_args()
+record_identifier = args.record_identifier
+record = get_metadata(record_identifier, authors=True)
+
+add_authors_affiliations(record, token, dimensions_key, ignore_mismatch=True)