edward/add-links
1
0
Fork 0
Code Issues 9 Pull requests Packages Projects Releases Wiki Activity

Refactor: new get_hit_count function

Browse source
This commit is contained in:
Edward Betts 2023-12-06 20:58:01 +00:00
parent 98ea589c58
commit 113dfd3630
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
web_view.py
View file

@ -84,24 +84,25 @@ def article_url(title: str) -> str:
return flask.url_for("article_page", url_title=title.replace(" ", "_")) 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: def search_count(q: str) -> int:
"""How often does this article title appear in Wikipedia.""" """How often does this article title appear in Wikipedia."""
query = run_search(article_title_to_search_query(q), limit=0) return get_hit_count(article_title_to_search_query(q)) - 1
return typing.cast(int, query["searchinfo"]["totalhits"]) - 1
def search_count_with_link(q: str) -> int: def search_count_with_link(q: str) -> int:
"""How often does this article title appear in Wikipedia.""" """Articles in Wikipedia that include this search term and a link."""
query = run_search(article_title_to_search_query(q) + f' linksto:"{q}"', limit=0) return get_hit_count(article_title_to_search_query(q) + f' linksto:"{q}"')
return typing.cast(int, query["searchinfo"]["totalhits"])
def search_no_link(q: str) -> tuple[int, list[Hit]]: def search_no_link(q: str) -> tuple[int, list[Hit]]:
"""Search for mentions of article title with no link included.""" """Search for mentions of article title with no link included."""
query = run_search(article_title_to_search_query(q) + f' -linksto:"{q}"', "max") query = run_search(article_title_to_search_query(q) + f' -linksto:"{q}"', "max")
totalhits = query["searchinfo"]["totalhits"] return (query["searchinfo"]["totalhits"], query["search"])
results = query["search"]
return (totalhits, results)
@app.before_request @app.before_request