From e13fe033ede605e2d41443e1e6732b93be681558 Mon Sep 17 00:00:00 2001
From: asciimoo <asciimoo@gmail.com>
Date: Sun, 27 Oct 2013 20:45:21 +0100
Subject: [PATCH] [enh] engine exception handling ++ stats

---
 searx/engines/__init__.py | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/searx/engines/__init__.py b/searx/engines/__init__.py
index 265f82976..27d1103b2 100644
--- a/searx/engines/__init__.py
+++ b/searx/engines/__init__.py
@@ -69,7 +69,7 @@ for section in engines_config.sections():
             print '[E] Engine config error: Missing attribute "{0}.{1}"'.format(engine.name, engine_attr)
             sys.exit(1)
     engines[engine.name] = engine
-    engine.stats = {'result_count': 0, 'search_count': 0, 'page_load_time': 0, 'score_count': 0}
+    engine.stats = {'result_count': 0, 'search_count': 0, 'page_load_time': 0, 'score_count': 0, 'errors': 0}
     if hasattr(engine, 'categories'):
         for category_name in engine.categories:
             categories.setdefault(category_name, []).append(engine)
@@ -84,7 +84,14 @@ def make_callback(engine_name, results, callback, params):
         cb_res = []
         response.search_params = params
         engines[engine_name].stats['page_load_time'] += (datetime.now() - params['started']).total_seconds()
-        for result in callback(response):
+        try:
+            search_results = callback(response)
+        except Exception, e:
+            engines[engine_name].stats['errors'] += 1
+            results[engine_name] = cb_res
+            print '[E] Error with engine "{0}":\n\t{1}'.format(engine_name, str(e))
+            return
+        for result in search_results:
             result['engine'] = engine_name
             cb_res.append(result)
         results[engine_name] = cb_res
@@ -169,8 +176,9 @@ def get_engines_stats():
     pageloads = []
     results = []
     scores = []
+    errors = []
 
-    max_pageload = max_results = max_score = 0
+    max_pageload = max_results = max_score = max_errors = 0
     for engine in engines.values():
         if engine.stats['search_count'] == 0:
             continue
@@ -183,9 +191,11 @@ def get_engines_stats():
         max_results = max(results_num, max_results)
         max_pageload = max(load_times, max_pageload)
         max_score = max(score, max_score)
+        max_errors = max(max_errors, engine.stats['errors'])
         pageloads.append({'avg': load_times, 'name': engine.name})
         results.append({'avg': results_num, 'name': engine.name})
         scores.append({'avg': score, 'name': engine.name})
+        errors.append({'avg': engine.stats['errors'], 'name': engine.name})
 
     for engine in pageloads:
         engine['percentage'] = int(engine['avg']/max_pageload*100)
@@ -196,8 +206,15 @@ def get_engines_stats():
     for engine in scores:
         engine['percentage'] = int(engine['avg']/max_score*100)
 
+    for engine in errors:
+        if max_errors:
+            engine['percentage'] = int(engine['avg']/max_errors*100)
+        else:
+            engine['percentage'] = 0
+
 
     return [('Page loads (sec)', sorted(pageloads, key=itemgetter('avg')))
            ,('Number of results', sorted(results, key=itemgetter('avg'), reverse=True))
            ,('Scores', sorted(scores, key=itemgetter('avg'), reverse=True))
+           ,('Errors', sorted(errors, key=itemgetter('avg'), reverse=True))
            ]