|
@@ -67,6 +67,7 @@ for section in engines_config.sections():
|
67
|
67
|
print '[E] Engine config error: Missing attribute "{0}.{1}"'.format(engine.name, engine_attr)
|
68
|
68
|
sys.exit(1)
|
69
|
69
|
engines[engine.name] = engine
|
|
70
|
+ engine.stats = {'result_count': 0, 'search_count': 0}
|
70
|
71
|
if hasattr(engine, 'categories'):
|
71
|
72
|
for category_name in engine.categories:
|
72
|
73
|
categories.setdefault(category_name, []).append(engine)
|
|
@@ -120,6 +121,9 @@ def search(query, request, selected_categories):
|
120
|
121
|
)
|
121
|
122
|
requests.append(req)
|
122
|
123
|
grequests.map(requests)
|
|
124
|
+ for engine_name,engine_results in results.items():
|
|
125
|
+ engines[engine_name].stats['search_count'] += 1
|
|
126
|
+ engines[engine_name].stats['result_count'] += len(engine_results)
|
123
|
127
|
flat_res = filter(None, chain.from_iterable(izip_longest(*results.values())))
|
124
|
128
|
flat_len = len(flat_res)
|
125
|
129
|
results = []
|
|
@@ -150,3 +154,13 @@ def search(query, request, selected_categories):
|
150
|
154
|
results.append(res)
|
151
|
155
|
|
152
|
156
|
return sorted(results, key=itemgetter('score'), reverse=True)
|
|
157
|
+
|
|
158
|
+def get_engines_stats():
|
|
159
|
+ stats = {}
|
|
160
|
+
|
|
161
|
+ for engine in engines.values():
|
|
162
|
+ if engine.stats['search_count'] == 0:
|
|
163
|
+ continue
|
|
164
|
+ stats[engine.name] = {'Average number of results': engine.stats['result_count']/float(engine.stats['search_count'])}
|
|
165
|
+
|
|
166
|
+ return stats
|