searx

commit: 17b0c9f74a17d76add28ccf515c2b8092c455d8a
parent: db72fc64498963adcfe412aa9d203a9de4d78d79
Author: Adam Tauber <asciimoo@gmail.com>
Date:   Fri, 19 Feb 2016 15:13:01 +0100

[mod] engine load refactor

Diffstat:
M
searx/engines/__init__.py
49
++++++++++++++++++-------------------------------

1 file changed, 18 insertions(+), 31 deletions(-)
diff --git a/searx/engines/__init__.py b/searx/engines/__init__.py
@@ -34,6 +34,13 @@ engines = {}
 categories = {'general': []}
 
 engine_shortcuts = {}
+engine_default_args = {'paging': False,
+                       'categories': ['general'],
+                       'language_support': True,
+                       'safesearch': False,
+                       'timeout': settings['outgoing']['request_timeout'],
+                       'shortcut': '-',
+                       'disabled': False}
 
 
 def load_module(filename):
@@ -62,26 +69,9 @@ def load_engine(engine_data):
             continue
         setattr(engine, param_name, engine_data[param_name])
 
-    if not hasattr(engine, 'paging'):
-        engine.paging = False
-
-    if not hasattr(engine, 'categories'):
-        engine.categories = ['general']
-
-    if not hasattr(engine, 'language_support'):
-        engine.language_support = True
-
-    if not hasattr(engine, 'safesearch'):
-        engine.safesearch = False
-
-    if not hasattr(engine, 'timeout'):
-        engine.timeout = settings['outgoing']['request_timeout']
-
-    if not hasattr(engine, 'shortcut'):
-        engine.shortcut = ''
-
-    if not hasattr(engine, 'disabled'):
-        engine.disabled = False
+    for arg_name, arg_value in engine_default_args.iteritems():
+        if not hasattr(engine, arg_name):
+            setattr(engine, arg_name, arg_value)
 
     # checking required variables
     for engine_attr in dir(engine):
@@ -100,18 +90,15 @@ def load_engine(engine_data):
         'errors': 0
     }
 
-    if hasattr(engine, 'categories'):
-        for category_name in engine.categories:
-            categories.setdefault(category_name, []).append(engine)
-    else:
-        categories['general'].append(engine)
+    for category_name in engine.categories:
+        categories.setdefault(category_name, []).append(engine)
+
+    if engine.shortcut in engine_shortcuts:
+        logger.error('Engine config error: ambigious shortcut: {0}'.format(engine.shortcut))
+        sys.exit(1)
+
+    engine_shortcuts[engine.shortcut] = engine.name
 
-    if engine.shortcut:
-        if engine.shortcut in engine_shortcuts:
-            logger.error('Engine config error: ambigious shortcut: {0}'
-                         .format(engine.shortcut))
-            sys.exit(1)
-        engine_shortcuts[engine.shortcut] = engine.name
     return engine