Remove unused code.

confarchive/view.py

@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+"""Flask views."""
 
 import os
 
@@ -17,6 +17,7 @@ database.init_app(app)
 
 @app.route("/person/<int:person_id>", methods=["GET", "POST"])
 def person(person_id: int) -> str | Response:
+    """Person page."""
     item = model.Person.query.get(person_id)
     if flask.request.method == "POST" and check_admin_mode():
         qid = flask.request.form["wikidata_qid"] or None
@@ -388,14 +389,6 @@ def person_image_filename(person_id: int) -> str:
     person = model.Person.query.get(person_id)
     return os.path.join("wikidata_photo", "thumb", person.wikidata_photo[0])
 
-    for filename in person.wikidata_photo:
-        face_crop = "face_1_" + filename
-        full = os.path.join("static", "wikidata_photo", "face_cropped", face_crop)
-        if os.path.exists(full):
-            return os.path.join("wikidata_photo", "face_cropped", face_crop)
-
-    return os.path.join("wikidata_photo", "thumb", person.wikidata_photo[0])
-
 
 @app.route("/login", methods=["GET", "POST"])
 def login() -> str | Response:

confarchive/wikidata.py

@@ -8,12 +8,14 @@ import time
 import typing
 
 import requests
+import flask
 
 commons_url = "https://www.wikidata.org/w/api.php"
 wikidata_api = "https://www.wikidata.org/w/api.php"
 user_agent = "conference-archive/0.1 (contact: edward@4angle.com)"
 
 CallParams = dict[str, str | int]
+SearchHit = dict[str, typing.Any]
 
 s = requests.Session()
 s.headers.update({"User-Agent": user_agent})
@@ -24,29 +26,39 @@ def md5sum(s: str) -> str:
     return hashlib.md5(s.encode("utf-8")).hexdigest()
 
 
-def search(q: str) -> list[dict[str, typing.Any]]:
-    """Search Wikidata with caching."""
+def get_cache_filename(q: str) -> str:
+    """Cache filename for query."""
     q_md5 = md5sum(q)
+    data_dir = flask.current_app.config["DATA_DIR"]
+    return os.path.join(data_dir, "cache", q_md5 + ".json")
 
-    cache_filename = os.path.join("cache", q_md5 + ".json")
+
+def get_item_filename(qid: str) -> str:
+    data_dir = flask.current_app.config["DATA_DIR"]
+    return os.path.join(data_dir, "item", qid + ".json")
+
+
+def search(q: str) -> list[SearchHit]:
+    """Search Wikidata with caching."""
+    cache_filename = get_cache_filename(q)
 
     if os.path.exists(cache_filename):
         data = json.load(open(cache_filename))
-    else:
-        params: dict[str, str | int] = {
-            "action": "query",
-            "list": "search",
-            "format": "json",
-            "formatversion": 2,
-            "srsearch": q,
-            "srlimit": "10",
-        }
-        r = requests.get(wikidata_api, params=params)
-        open(cache_filename, "w").write(r.text)
-        data = r.json()
-        time.sleep(1)
+        return typing.cast(list[SearchHit], data["query"]["search"])
 
-    return typing.cast(list[dict[str, typing.Any]], data["query"]["search"])
+    params: dict[str, str | int] = {
+        "action": "query",
+        "list": "search",
+        "format": "json",
+        "formatversion": 2,
+        "srsearch": q,
+        "srlimit": "10",
+    }
+    r = requests.get(wikidata_api, params=params)
+    open(cache_filename, "w").write(r.text)
+    data = r.json()
+
+    return typing.cast(list[SearchHit], data["query"]["search"])
 
 
 def api_image_detail_call(filename: str) -> requests.Response:
@@ -65,9 +77,9 @@ def api_image_detail_call(filename: str) -> requests.Response:
 
 def get_item(qid: str) -> typing.Any:
     """Get an item from Wikidata."""
-    cache_filename = os.path.join("items", qid + ".json")
-    if os.path.exists(cache_filename):
-        item = json.load(open(cache_filename))
+    item_filename = get_item_filename(qid)
+    if os.path.exists(item_filename):
+        item = json.load(open(item_filename))
     else:
         params: dict[str, str | int] = {
             "action": "wbgetentities",
@@ -77,9 +89,8 @@ def get_item(qid: str) -> typing.Any:
         }
         r = s.get(wikidata_api, params=params)
         item = r.json()["entities"][qid]
-        with open(cache_filename, "w") as f:
+        with open(item_filename, "w") as f:
             json.dump(item, f, indent=2)
-        time.sleep(0.1)
     return item