python script to fix fatcat ISSN-Ls

1 files changed, 75 insertions, 0 deletions

diff --git a/fix_invalid_issnl.py b/fix_invalid_issnl.py
new file mode 100755
index 0000000..521f334
--- /dev/null
+++ b/fix_invalid_issnl.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+
+"""
+This is a one-off script for pushing ISSN-L fixes to fatcat via the API.
+
+It excpects to have fatcat python libraries available. Run it like:
+
+    export FATCAT_API_AUTH_TOKEN="..."
+    ./fix_invalid_issnl.py ~/code/chocula/invalid_fatcat_issnl.tsv
+
+It creates a new editgroup, which you'll need to merge/accept manually.
+
+Defaults to QA API endpoint; edit the file to switch to prod.
+"""
+
+import os, sys
+import csv
+
+from fatcat_tools import authenticated_api
+from fatcat_client import Editgroup, ContainerEntity
+from fatcat_client.rest import ApiException
+
+API_ENDPOINT = "https://api.qa.fatcat.wiki/v0"
+
+
+def run(api, row_iter):
+
+    eg = api.create_editgroup(Editgroup(description=
+        "Update or merge containers with invalid (by checksum) ISSN-L. Using the fix_invalid_issnl.py script from chocula repo."))
+    print("Editgroup ident: {}".format(eg.editgroup_id))
+    for row in row_iter:
+        #print(row)
+        fixed_issnl = row['fixed_issnl'].strip()
+        if not fixed_issnl:
+            print("SKIP")
+            continue
+        assert row['issnl'].strip() != fixed_issnl
+        invalid = api.get_container(row['fatcat_ident'])
+        assert invalid.state == "active"
+        try:
+            fixed = api.lookup_container(issnl=row['fixed_issnl'])
+        except ApiException as ae:
+            if ae.status != 404:
+                raise ae
+            fixed = None
+
+        if fixed:
+            # merge/redirect
+            assert fixed.state == "active"
+            print("MERGE: {} -> {}".format(invalid.ident, fixed.ident))
+            invalid.redirect = fixed.ident
+            api.update_container(eg.editgroup_id, invalid.ident,
+                ContainerEntity(redirect=fixed.ident))
+        else:
+            # update in-place with fixed ISSN-L
+            print("FIX: {}: {}".format(invalid.ident, fixed_issnl))
+            invalid.issnl = fixed_issnl
+            api.update_container(eg.editgroup_id, invalid.ident, invalid)
+
+    # intentionally not merging editgroup
+    print("Editgroup ident: {}".format(eg.editgroup_id))
+    print("DONE")
+
+def main():
+    api = authenticated_api(
+        API_ENDPOINT,
+        # token is an optional kwarg (can be empty string, None, etc)
+        token=os.environ.get("FATCAT_API_AUTH_TOKEN"))
+
+    path = sys.argv[1]
+    reader = csv.DictReader(open(path), delimiter='\t')
+    run(api, reader)
+
+if __name__ == '__main__':
+    main()