From 113dfd363026c6b6312ca7cd657c655fa3e854e8 Mon Sep 17 00:00:00 2001
From: Edward Betts <edward@4angle.com>
Date: Wed, 6 Dec 2023 20:58:01 +0000
Subject: [PATCH] Refactor: new get_hit_count function
---
web_view.py | 17 +++++++++--------
1 file changed, 9 insertions(+), 8 deletions(-)
diff --git a/web_view.py b/web_view.py
index f72aff1..0f8a1c6 100755
--- a/web_view.py
+++ b/web_view.py
@@ -84,24 +84,25 @@ def article_url(title: str) -> str:
return flask.url_for("article_page", url_title=title.replace(" ", "_"))
+def get_hit_count(q: str) -> int:
+ """Search Wikipedia and return hit count."""
+ return typing.cast(int, run_search(q, limit=0)["searchinfo"]["totalhits"])
+
+
def search_count(q: str) -> int:
"""How often does this article title appear in Wikipedia."""
- query = run_search(article_title_to_search_query(q), limit=0)
- return typing.cast(int, query["searchinfo"]["totalhits"]) - 1
+ return get_hit_count(article_title_to_search_query(q)) - 1
def search_count_with_link(q: str) -> int:
- """How often does this article title appear in Wikipedia."""
- query = run_search(article_title_to_search_query(q) + f' linksto:"{q}"', limit=0)
- return typing.cast(int, query["searchinfo"]["totalhits"])
+ """Articles in Wikipedia that include this search term and a link."""
+ return get_hit_count(article_title_to_search_query(q) + f' linksto:"{q}"')
def search_no_link(q: str) -> tuple[int, list[Hit]]:
"""Search for mentions of article title with no link included."""
query = run_search(article_title_to_search_query(q) + f' -linksto:"{q}"', "max")
- totalhits = query["searchinfo"]["totalhits"]
- results = query["search"]
- return (totalhits, results)
+ return (query["searchinfo"]["totalhits"], query["search"])
@app.before_request