@@ -109,8 +109,20 @@ def render(template_name, **kwargs):
109def parse_query(query):
109def 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+ if query_parts[0].startswith('-'):
114+ engine_name = query_parts[0][1:].replace('_', ' ')
115+ if engine_name in engines:
113− query_engines.append({'category': 'TODO', 'name': query_parts[0][1:]})
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])
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
@@ -130,7 +142,10 @@ def index():
131 query, selected_engines = parse_query(request_data['q'].encode('utf-8'))
143 query, selected_engines = parse_query(request_data['q'].encode('utf-8'))
133− if not len(selected_engines):
145+ if len(selected_engines):
146+ selected_categories = list(set(engine['category']
147+ for engine in selected_engines))
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:]