From 4d5f2c64c0ad6051e0d4048cd972d62c74fb5f2b Mon Sep 17 00:00:00 2001
From: Carlin MacKenzie <carlin.mackenzie@gmail.com>
Date: Fri, 3 Jun 2022 09:50:12 +0100
Subject: [PATCH 1/2] Add redirect for doi-prefix as specified

---
 scholia/app/views.py | 20 +++++++++++++++++++-
 scholia/query.py     | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/scholia/app/views.py b/scholia/app/views.py
index 7d1d61094..f695c7509 100644
--- a/scholia/app/views.py
+++ b/scholia/app/views.py
@@ -14,7 +14,7 @@
 from ..arxiv import metadata_to_quickstatements, string_to_arxiv
 from ..arxiv import get_metadata as get_arxiv_metadata
 from ..query import (arxiv_to_qs, cas_to_qs, atomic_symbol_to_qs, doi_to_qs,
-                     github_to_qs, biorxiv_to_qs, chemrxiv_to_qs,
+                     doi_prefix_to_qs, github_to_qs, biorxiv_to_qs, chemrxiv_to_qs,
                      identifier_to_qs, inchikey_to_qs, issn_to_qs, orcid_to_qs,
                      viaf_to_qs, q_to_class, q_to_dois, random_author,
                      twitter_to_qs, cordis_to_qs, mesh_to_qs, pubmed_to_qs,
@@ -721,6 +721,24 @@ def redirect_doi(doi):
     return render_template('404-doi.html', doi=doi)
 
 
+@main.route('/doi-prefix/<path:doi>')
+def redirect_doi_prefix(doi):
+    """Detect and redirect for DOI.
+
+    Parameters
+    ----------
+    doi : str
+        DOI identifier.
+
+    """
+    normalize_doi = remove_special_characters_url(doi)
+    qs = doi_prefix_to_qs(normalize_doi)
+    if len(qs) > 0:
+        q = qs[0]
+        return redirect(url_for('app.show_publisher', q=q), code=302)
+    return render_template('404-doi.html', doi=doi)
+
+
 @main.route('/event/' + q_pattern)
 def show_event(q):
     """Return HTML rendering for specific event.
diff --git a/scholia/query.py b/scholia/query.py
index 0e48bf6d5..2059e4a5c 100644
--- a/scholia/query.py
+++ b/scholia/query.py
@@ -383,6 +383,44 @@ def doi_to_qs(doi):
             for item in data['results']['bindings']]
 
 
+def doi_prefix_to_qs(doi):
+    """Convert DOI prefix to Wikidata ID.
+
+    Wikidata Query Service is used to resolve the DOI.
+
+    The DOI string is converted to uppercase before any
+    query is made. Uppercase DOIs are default in Wikidata.
+
+    Parameters
+    ----------
+    doi : str
+        DOI prefix identifier
+
+    Returns
+    -------
+    qs : list of str
+        Strings of Wikidata ID.
+
+    Examples
+    --------
+    >>> doi_prefix_to_qs('10.1186') == ['Q463494']
+    True
+
+    >>> doi_prefix_to_qs('10.1016') == ['Q746413']
+    True
+    """
+    query = 'select ?work where {{ ?work wdt:P1662 "{doi}" }}'.format(
+        doi=escape_string(doi.upper()))
+
+    url = 'https://query.wikidata.org/sparql'
+    params = {'query': query, 'format': 'json'}
+    response = requests.get(url, params=params, headers=HEADERS)
+    data = response.json()
+
+    return [item['work']['value'][31:]
+            for item in data['results']['bindings']]
+
+
 def iso639_to_q(language):
     """Convert ISO639 to Q item.
 

From ea954f99a89a9f2933ad3ee7d25102c18389d26e Mon Sep 17 00:00:00 2001
From: Carlin MacKenzie <carlin.mackenzie@gmail.com>
Date: Fri, 3 Jun 2022 09:54:31 +0100
Subject: [PATCH 2/2] Change 404 page

---
 scholia/app/views.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scholia/app/views.py b/scholia/app/views.py
index f695c7509..4af247a48 100644
--- a/scholia/app/views.py
+++ b/scholia/app/views.py
@@ -736,7 +736,7 @@ def redirect_doi_prefix(doi):
     if len(qs) > 0:
         q = qs[0]
         return redirect(url_for('app.show_publisher', q=q), code=302)
-    return render_template('404-doi.html', doi=doi)
+    return render_template('404.html', doi=doi)
 
 
 @main.route('/event/' + q_pattern)