Quellcode durchsuchen

[enh] engine exception handling ++ stats

asciimoo vor 11 Jahren
Ursprung
Commit
e13fe033ed
1 geänderte Dateien mit 20 neuen und 3 gelöschten Zeilen
  1. 20
    3
      searx/engines/__init__.py

+ 20
- 3
searx/engines/__init__.py Datei anzeigen

@@ -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
            ]