Browse Source

[enh][mod] better query parsing ++ selecting categories with query strings

asciimoo 11 years ago
parent
commit
db825d4226
1 changed files with 18 additions and 3 deletions
  1. 18
    3
      searx/webapp.py

+ 18
- 3
searx/webapp.py View File

109
 def parse_query(query):
109
 def parse_query(query):
110
     query_engines = []
110
     query_engines = []
111
     query_parts = query.split()
111
     query_parts = query.split()
112
-    if query_parts[0].startswith('-') and query_parts[0][1:] in engines:
113
-        query_engines.append({'category': 'TODO', 'name': query_parts[0][1:]})
112
+
113
+    if query_parts[0].startswith('-'):
114
+        engine_name = query_parts[0][1:].replace('_', ' ')
115
+        if engine_name in engines:
116
+            query_engines.append({'category': 'none',
117
+                                  'name': query_parts[0][1:]})
118
+    elif query_parts[0].startswith('!'):
119
+        category_name = query_parts[0][1:].replace('_', ' ')
120
+        if category_name in categories:
121
+            query_engines.extend({'category': category_name,
122
+                                  'name': engine.name}
123
+                                 for engine in categories[category_name])
124
+
125
+    if len(query_engines):
114
         query = query.replace(query_parts[0], '', 1).strip()
126
         query = query.replace(query_parts[0], '', 1).strip()
115
     return query, query_engines
127
     return query, query_engines
116
 
128
 
130
 
142
 
131
     query, selected_engines = parse_query(request_data['q'].encode('utf-8'))
143
     query, selected_engines = parse_query(request_data['q'].encode('utf-8'))
132
 
144
 
133
-    if not len(selected_engines):
145
+    if len(selected_engines):
146
+        selected_categories = list(set(engine['category']
147
+                                       for engine in selected_engines))
148
+    else:
134
         for pd_name, pd in request_data.items():
149
         for pd_name, pd in request_data.items():
135
             if pd_name.startswith('category_'):
150
             if pd_name.startswith('category_'):
136
                 category = pd_name[9:]
151
                 category = pd_name[9:]