summaryrefslogtreecommitdiff
path: root/searx/external_bang.py
diff options
context:
space:
mode:
Diffstat (limited to 'searx/external_bang.py')
-rw-r--r--searx/external_bang.py19
1 files changed, 14 insertions, 5 deletions
diff --git a/searx/external_bang.py b/searx/external_bang.py
index fde2f8eb5..00d54ec8f 100644
--- a/searx/external_bang.py
+++ b/searx/external_bang.py
@@ -1,13 +1,20 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring
+__all__ = ["get_bang_url"]
+
+import typing as t
+
from urllib.parse import quote_plus, urlparse
from searx.data import EXTERNAL_BANGS
LEAF_KEY = chr(16)
+if t.TYPE_CHECKING:
+ from searx.search.models import SearchQuery
+
-def get_node(external_bangs_db, bang):
+def get_node(external_bangs_db: dict[str, t.Any], bang: str):
node = external_bangs_db['trie']
after = ''
before = ''
@@ -20,7 +27,7 @@ def get_node(external_bangs_db, bang):
return node, before, after
-def get_bang_definition_and_ac(external_bangs_db, bang):
+def get_bang_definition_and_ac(external_bangs_db: dict[str, t.Any], bang: str):
node, before, after = get_node(external_bangs_db, bang)
bang_definition = None
@@ -39,7 +46,7 @@ def get_bang_definition_and_ac(external_bangs_db, bang):
return bang_definition, bang_ac_list
-def resolve_bang_definition(bang_definition, query):
+def resolve_bang_definition(bang_definition: str, query: str) -> tuple[str, int]:
url, rank = bang_definition.split(chr(1))
if url.startswith('//'):
url = 'https:' + url
@@ -54,7 +61,9 @@ def resolve_bang_definition(bang_definition, query):
return (url, rank)
-def get_bang_definition_and_autocomplete(bang, external_bangs_db=None): # pylint: disable=invalid-name
+def get_bang_definition_and_autocomplete(
+ bang: str, external_bangs_db: dict[str, t.Any] | None = None
+): # pylint: disable=invalid-name
if external_bangs_db is None:
external_bangs_db = EXTERNAL_BANGS
@@ -81,7 +90,7 @@ def get_bang_definition_and_autocomplete(bang, external_bangs_db=None): # pylin
return bang_definition, new_autocomplete
-def get_bang_url(search_query, external_bangs_db=None):
+def get_bang_url(search_query: "SearchQuery", external_bangs_db: dict[str, t.Any] | None = None) -> str | None:
"""
Redirects if the user supplied a correct bang search.
:param search_query: This is a search_query object which contains preferences and the submitted queries.