on master help sign in
[mod] engine load refactor
commit17b0c9f74a17d76add28ccf515c2b8092c455d8a authorAdam Tauber <asciimoo@gmail.com> dateFri 19 Feb 2016, 15:13 (11 years ago) parentdb72fc6
1 file changed+18−31
searx/engines/__init__.py+18−31
searx/engines/__init__.py+18−31
@@ -34,6 +34,13 @@ engines = {}
3434categories = {'general': []}
3535
3636engine_shortcuts = {}
37+engine_default_args = {'paging': False,
38+ 'categories': ['general'],
39+ 'language_support': True,
40+ 'safesearch': False,
41+ 'timeout': settings['outgoing']['request_timeout'],
42+ 'shortcut': '-',
43+ 'disabled': False}
3744
3845
3946def load_module(filename):
@@ -62,26 +69,9 @@ def load_engine(engine_data):
6269 continue
6370 setattr(engine, param_name, engine_data[param_name])
6471
65− if not hasattr(engine, 'paging'):
66− engine.paging = False
67−
68− if not hasattr(engine, 'categories'):
69− engine.categories = ['general']
70−
71− if not hasattr(engine, 'language_support'):
72− engine.language_support = True
73−
74− if not hasattr(engine, 'safesearch'):
75− engine.safesearch = False
76−
77− if not hasattr(engine, 'timeout'):
78− engine.timeout = settings['outgoing']['request_timeout']
79−
80− if not hasattr(engine, 'shortcut'):
81− engine.shortcut = ''
82−
83− if not hasattr(engine, 'disabled'):
84− engine.disabled = False
72+ for arg_name, arg_value in engine_default_args.iteritems():
73+ if not hasattr(engine, arg_name):
74+ setattr(engine, arg_name, arg_value)
8575
8676 # checking required variables
8777 for engine_attr in dir(engine):
@@ -100,18 +90,15 @@ def load_engine(engine_data):
10090 'errors': 0
10191 }
10292
103− if hasattr(engine, 'categories'):
104− for category_name in engine.categories:
105− categories.setdefault(category_name, []).append(engine)
106− else:
107− categories['general'].append(engine)
93+ for category_name in engine.categories:
94+ categories.setdefault(category_name, []).append(engine)
95+
96+ if engine.shortcut in engine_shortcuts:
97+ logger.error('Engine config error: ambigious shortcut: {0}'.format(engine.shortcut))
98+ sys.exit(1)
99+
100+ engine_shortcuts[engine.shortcut] = engine.name
108101
109− if engine.shortcut:
110− if engine.shortcut in engine_shortcuts:
111− logger.error('Engine config error: ambigious shortcut: {0}'
112− .format(engine.shortcut))
113− sys.exit(1)
114− engine_shortcuts[engine.shortcut] = engine.name
115102 return engine
116103
117104