mirror of https://github.com/searxng/searxng.git
using Query class for parsing of search query
This commit is contained in:
parent
510aba5e66
commit
62d1a70c84
|
@ -39,6 +39,7 @@ class Query(object):
|
||||||
self.engines = []
|
self.engines = []
|
||||||
self.languages = []
|
self.languages = []
|
||||||
|
|
||||||
|
# parse query, if tags are set, which change the serch engine or search-language
|
||||||
def parse_query(self):
|
def parse_query(self):
|
||||||
self.query_parts = []
|
self.query_parts = []
|
||||||
|
|
||||||
|
@ -55,7 +56,8 @@ class Query(object):
|
||||||
parse_next = False
|
parse_next = False
|
||||||
|
|
||||||
# part does only contain spaces, skip
|
# part does only contain spaces, skip
|
||||||
if query_part.isspace():
|
if query_part.isspace()\
|
||||||
|
or query_part == '':
|
||||||
parse_next = True
|
parse_next = True
|
||||||
self.query_parts.append(query_part)
|
self.query_parts.append(query_part)
|
||||||
continue
|
continue
|
||||||
|
|
|
@ -25,6 +25,7 @@ from searx.engines import (
|
||||||
)
|
)
|
||||||
from searx.languages import language_codes
|
from searx.languages import language_codes
|
||||||
from searx.utils import gen_useragent
|
from searx.utils import gen_useragent
|
||||||
|
from searx.query import Query
|
||||||
|
|
||||||
|
|
||||||
number_of_searches = 0
|
number_of_searches = 0
|
||||||
|
@ -235,7 +236,15 @@ class Search(object):
|
||||||
self.pageno = int(pageno_param)
|
self.pageno = int(pageno_param)
|
||||||
|
|
||||||
# parse query, if tags are set, which change the serch engine or search-language
|
# parse query, if tags are set, which change the serch engine or search-language
|
||||||
self.parse_query()
|
query_obj = Query(self.query, self.blocked_engines)
|
||||||
|
query_obj.parse_query()
|
||||||
|
|
||||||
|
# get last selected language in query, if possible
|
||||||
|
# TODO support search with multible languages
|
||||||
|
if len(query_obj.languages):
|
||||||
|
self.lang = query_obj.languages[-1]
|
||||||
|
|
||||||
|
self.engines = query_obj.engines
|
||||||
|
|
||||||
self.categories = []
|
self.categories = []
|
||||||
|
|
||||||
|
@ -276,60 +285,6 @@ class Search(object):
|
||||||
for x in categories[categ]
|
for x in categories[categ]
|
||||||
if not x.name in self.blocked_engines)
|
if not x.name in self.blocked_engines)
|
||||||
|
|
||||||
# parse query, if tags are set, which change the serch engine or search-language
|
|
||||||
def parse_query(self):
|
|
||||||
query_parts = self.query.split()
|
|
||||||
modified = False
|
|
||||||
|
|
||||||
# check if language-prefix is set
|
|
||||||
if query_parts[0].startswith(':'):
|
|
||||||
lang = query_parts[0][1:].lower()
|
|
||||||
|
|
||||||
# check if any language-code is equal with declared language-codes
|
|
||||||
for lc in language_codes:
|
|
||||||
lang_id, lang_name, country = map(str.lower, lc)
|
|
||||||
|
|
||||||
# if correct language-code is found, set it as new search-language
|
|
||||||
if lang == lang_id\
|
|
||||||
or lang_id.startswith(lang)\
|
|
||||||
or lang == lang_name\
|
|
||||||
or lang == country:
|
|
||||||
self.lang = lang
|
|
||||||
modified = True
|
|
||||||
break
|
|
||||||
|
|
||||||
# check if category/engine prefix is set
|
|
||||||
elif query_parts[0].startswith('!'):
|
|
||||||
prefix = query_parts[0][1:].replace('_', ' ')
|
|
||||||
|
|
||||||
# check if prefix is equal with engine shortcut
|
|
||||||
if prefix in engine_shortcuts\
|
|
||||||
and not engine_shortcuts[prefix] in self.blocked_engines:
|
|
||||||
modified = True
|
|
||||||
self.engines.append({'category': 'none',
|
|
||||||
'name': engine_shortcuts[prefix]})
|
|
||||||
|
|
||||||
# check if prefix is equal with engine name
|
|
||||||
elif prefix in engines\
|
|
||||||
and not prefix in self.blocked_engines:
|
|
||||||
modified = True
|
|
||||||
self.engines.append({'category': 'none',
|
|
||||||
'name': prefix})
|
|
||||||
|
|
||||||
# check if prefix is equal with categorie name
|
|
||||||
elif prefix in categories:
|
|
||||||
modified = True
|
|
||||||
# using all engines for that search, which are declared under that categorie name
|
|
||||||
self.engines.extend({'category': prefix,
|
|
||||||
'name': engine.name}
|
|
||||||
for engine in categories[prefix]
|
|
||||||
if not engine in self.blocked_engines)
|
|
||||||
|
|
||||||
# if language, category or engine were specificed in this query, search for more tags which does the same
|
|
||||||
if modified:
|
|
||||||
self.query = self.query.replace(query_parts[0], '', 1).strip()
|
|
||||||
self.parse_query()
|
|
||||||
|
|
||||||
# do search-request
|
# do search-request
|
||||||
def search(self, request):
|
def search(self, request):
|
||||||
global number_of_searches
|
global number_of_searches
|
||||||
|
|
Loading…
Reference in New Issue