|
@@ -69,7 +69,7 @@ for section in engines_config.sections():
|
69
|
69
|
print '[E] Engine config error: Missing attribute "{0}.{1}"'.format(engine.name, engine_attr)
|
70
|
70
|
sys.exit(1)
|
71
|
71
|
engines[engine.name] = engine
|
72
|
|
- engine.stats = {'result_count': 0, 'search_count': 0, 'page_load_time': 0, 'score_count': 0}
|
|
72
|
+ engine.stats = {'result_count': 0, 'search_count': 0, 'page_load_time': 0, 'score_count': 0, 'errors': 0}
|
73
|
73
|
if hasattr(engine, 'categories'):
|
74
|
74
|
for category_name in engine.categories:
|
75
|
75
|
categories.setdefault(category_name, []).append(engine)
|
|
@@ -84,7 +84,14 @@ def make_callback(engine_name, results, callback, params):
|
84
|
84
|
cb_res = []
|
85
|
85
|
response.search_params = params
|
86
|
86
|
engines[engine_name].stats['page_load_time'] += (datetime.now() - params['started']).total_seconds()
|
87
|
|
- for result in callback(response):
|
|
87
|
+ try:
|
|
88
|
+ search_results = callback(response)
|
|
89
|
+ except Exception, e:
|
|
90
|
+ engines[engine_name].stats['errors'] += 1
|
|
91
|
+ results[engine_name] = cb_res
|
|
92
|
+ print '[E] Error with engine "{0}":\n\t{1}'.format(engine_name, str(e))
|
|
93
|
+ return
|
|
94
|
+ for result in search_results:
|
88
|
95
|
result['engine'] = engine_name
|
89
|
96
|
cb_res.append(result)
|
90
|
97
|
results[engine_name] = cb_res
|
|
@@ -169,8 +176,9 @@ def get_engines_stats():
|
169
|
176
|
pageloads = []
|
170
|
177
|
results = []
|
171
|
178
|
scores = []
|
|
179
|
+ errors = []
|
172
|
180
|
|
173
|
|
- max_pageload = max_results = max_score = 0
|
|
181
|
+ max_pageload = max_results = max_score = max_errors = 0
|
174
|
182
|
for engine in engines.values():
|
175
|
183
|
if engine.stats['search_count'] == 0:
|
176
|
184
|
continue
|
|
@@ -183,9 +191,11 @@ def get_engines_stats():
|
183
|
191
|
max_results = max(results_num, max_results)
|
184
|
192
|
max_pageload = max(load_times, max_pageload)
|
185
|
193
|
max_score = max(score, max_score)
|
|
194
|
+ max_errors = max(max_errors, engine.stats['errors'])
|
186
|
195
|
pageloads.append({'avg': load_times, 'name': engine.name})
|
187
|
196
|
results.append({'avg': results_num, 'name': engine.name})
|
188
|
197
|
scores.append({'avg': score, 'name': engine.name})
|
|
198
|
+ errors.append({'avg': engine.stats['errors'], 'name': engine.name})
|
189
|
199
|
|
190
|
200
|
for engine in pageloads:
|
191
|
201
|
engine['percentage'] = int(engine['avg']/max_pageload*100)
|
|
@@ -196,8 +206,15 @@ def get_engines_stats():
|
196
|
206
|
for engine in scores:
|
197
|
207
|
engine['percentage'] = int(engine['avg']/max_score*100)
|
198
|
208
|
|
|
209
|
+ for engine in errors:
|
|
210
|
+ if max_errors:
|
|
211
|
+ engine['percentage'] = int(engine['avg']/max_errors*100)
|
|
212
|
+ else:
|
|
213
|
+ engine['percentage'] = 0
|
|
214
|
+
|
199
|
215
|
|
200
|
216
|
return [('Page loads (sec)', sorted(pageloads, key=itemgetter('avg')))
|
201
|
217
|
,('Number of results', sorted(results, key=itemgetter('avg'), reverse=True))
|
202
|
218
|
,('Scores', sorted(scores, key=itemgetter('avg'), reverse=True))
|
|
219
|
+ ,('Errors', sorted(errors, key=itemgetter('avg'), reverse=True))
|
203
|
220
|
]
|