From f8b92a6955dfdd3d5348af63951ef2728d8948b8 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Fri, 14 Dec 2018 22:06:40 +0800
Subject: multiple lookup extids (web view, no tests yet)

---
 python/fatcat_web/routes.py | 37 +++++++++++++++++++++++++------------
 1 file changed, 25 insertions(+), 12 deletions(-)

(limited to 'python/fatcat_web')

diff --git a/python/fatcat_web/routes.py b/python/fatcat_web/routes.py
index 0bb7a230..0150fca5 100644
--- a/python/fatcat_web/routes.py
+++ b/python/fatcat_web/routes.py
@@ -61,11 +61,15 @@ def container_create():
 
 @app.route('/container/lookup', methods=['GET'])
 def container_lookup():
-    issnl = request.args.get('issnl')
-    if issnl is None:
+    extid = None
+    for key in ('issnl', 'wikidata_qid'):
+        if request.args.get(key):
+            extid = key
+            break
+    if extid is None:
         abort(400)
     try:
-        resp = api.lookup_container(issnl)
+        resp = api.lookup_container(**{extid: request.args.get(extid)})
     except ApiException as ae:
         abort(ae.status)
     return redirect('/container/{}'.format(resp.ident))
@@ -101,11 +105,14 @@ def creator_edit_view(ident):
 
 @app.route('/creator/lookup', methods=['GET'])
 def creator_lookup():
-    orcid = request.args.get('orcid')
-    if orcid is None:
+    for key in ('orcid', 'wikidata_qid'):
+        if request.args.get(key):
+            extid = key
+            break
+    if extid is None:
         abort(400)
     try:
-        resp = api.lookup_creator(orcid)
+        resp = api.lookup_creator(**{extid: request.args.get(extid)})
     except ApiException as ae:
         abort(ae.status)
     return redirect('/creator/{}'.format(resp.ident))
@@ -142,11 +149,14 @@ def file_edit_view(ident):
 
 @app.route('/file/lookup', methods=['GET'])
 def file_lookup():
-    sha1 = request.args.get('sha1')
-    if sha1 is None:
+    for key in ('md5', 'sha1', 'sha256'):
+        if request.args.get(key):
+            extid = key
+            break
+    if extid is None:
         abort(400)
     try:
-        resp = api.lookup_file(sha1)
+        resp = api.lookup_file(**{extid: request.args.get(extid)})
     except ApiException as ae:
         abort(ae.status)
     return redirect('/file/{}'.format(resp.ident))
@@ -161,11 +171,14 @@ def file_view(ident):
 
 @app.route('/release/lookup', methods=['GET'])
 def release_lookup():
-    doi = request.args.get('doi')
-    if doi is None:
+    for key in ('doi', 'wikidata_qid', 'pmid', 'pmcid', 'isbn13', 'core_id'):
+        if request.args.get(key):
+            extid = key
+            break
+    if extid is None:
         abort(400)
     try:
-        resp = api.lookup_release(doi)
+        resp = api.lookup_release(**{extid: request.args.get(extid)})
     except ApiException as ae:
         abort(ae.status)
     return redirect('/release/{}'.format(resp.ident))
-- 
cgit v1.2.3