forked from zaclys/searxng
		
	using Query class for parsing of search query
This commit is contained in:
		
							parent
							
								
									510aba5e66
								
							
						
					
					
						commit
						62d1a70c84
					
				
					 2 changed files with 13 additions and 56 deletions
				
			
		| 
						 | 
				
			
			@ -39,6 +39,7 @@ class Query(object):
 | 
			
		|||
        self.engines = []
 | 
			
		||||
        self.languages = []
 | 
			
		||||
    
 | 
			
		||||
    # parse query, if tags are set, which change the serch engine or search-language
 | 
			
		||||
    def parse_query(self):
 | 
			
		||||
        self.query_parts = []
 | 
			
		||||
        
 | 
			
		||||
| 
						 | 
				
			
			@ -55,7 +56,8 @@ class Query(object):
 | 
			
		|||
            parse_next = False
 | 
			
		||||
           
 | 
			
		||||
            # part does only contain spaces, skip
 | 
			
		||||
            if query_part.isspace():
 | 
			
		||||
            if query_part.isspace()\
 | 
			
		||||
               or query_part == '':
 | 
			
		||||
                parse_next = True
 | 
			
		||||
                self.query_parts.append(query_part)
 | 
			
		||||
                continue
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -25,6 +25,7 @@ from searx.engines import (
 | 
			
		|||
)
 | 
			
		||||
from searx.languages import language_codes
 | 
			
		||||
from searx.utils import gen_useragent
 | 
			
		||||
from searx.query import Query
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
number_of_searches = 0
 | 
			
		||||
| 
						 | 
				
			
			@ -235,7 +236,15 @@ class Search(object):
 | 
			
		|||
        self.pageno = int(pageno_param)
 | 
			
		||||
 | 
			
		||||
        # 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 = []
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -276,60 +285,6 @@ class Search(object):
 | 
			
		|||
                                    for x in categories[categ]
 | 
			
		||||
                                    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
 | 
			
		||||
    def search(self, request):
 | 
			
		||||
        global number_of_searches
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		
		Reference in a new issue