open.source/whoogle-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix handling of bangs

Browse Source
This commit is contained in:
Joao Ramos 2022-09-23 13:09:40 +02:00
parent 74503d542e
commit 15f6e432ff
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
app/utils/bangs.py
View File

@ -53,24 +53,29 @@ def resolve_bang(query: str, bangs_dict: dict) -> str:
wasn't a match or didn't contain a bang operator
"""
# if ! no in request simply return (speed up processing)
if '!' not in query:
return ''
split_query = query.strip().split(' ')
# Ensure bang search is case insensitive
query = query.lower()
split_query = query.split(' ')
for operator in bangs_dict.keys():
if operator not in split_query \
and operator[1:] + operator[0] not in split_query:
continue
# Assumes query starts with bang
operator = split_query[0].lower()
bang_query = query.replace(
operator if operator in split_query else operator[1:] +
operator[0], ''
).strip()
# rebuild the query string
bang_query = ' '.join(split_query[1:]).strip()
print("split_query", bang_query)
bang_url = bangs_dict[operator]['url']
# Check if operator is a key in bangs and get bang if exists
bang = bangs_dict.get(operator, None)
if bang:
bang_url = bang['url']
if bang_query:
return bang_url.replace('{}', bang_query, 1)
else:
parsed_url = urlparse.urlparse(bang_url)
print("no query", f'{parsed_url.scheme}://{parsed_url.netloc}')
return f'{parsed_url.scheme}://{parsed_url.netloc}'
return ''