From 5c934e19bc672f4cd49c02fea8a33d1787500978 Mon Sep 17 00:00:00 2001 From: Edward Betts Date: Sun, 24 Sep 2023 21:41:05 +0100 Subject: [PATCH] Types and docstrings. --- confarchive/wikidata.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/confarchive/wikidata.py b/confarchive/wikidata.py index 0df3a6e..19fd78f 100644 --- a/confarchive/wikidata.py +++ b/confarchive/wikidata.py @@ -13,6 +13,8 @@ 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] + s = requests.Session() s.headers.update({"User-Agent": user_agent}) @@ -49,7 +51,7 @@ def search(q: str) -> list[dict[str, typing.Any]]: def api_image_detail_call(filename: str) -> requests.Response: """Call the Commons API.""" - call_params = { + call_params: CallParams = { "format": "json", "formatversion": 2, "action": "query", @@ -102,6 +104,7 @@ def download_photo(filename: str) -> None: def get_photo(filename: str) -> None: + """Download filename and resize.""" save_to = os.path.join("static", "wikidata_photo", filename) thumb = os.path.join("static", "wikidata_photo", "thumb", filename) if not os.path.exists(save_to):