OwlBurst/Ponysearch
1
0
Fork 0
forked from Ponysearch/Ponysearch
Code Pull requests Activity

[enh] engine score weight support @ config and engines

Browse source
This commit is contained in:
asciimoo 2013-11-10 21:16:25 +01:00
parent 36fd512278
commit 6d956a7d1d
1 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
searx/engines/__init__.py
View file

@ -51,20 +51,20 @@ if not engines_config.sections():
print '[E] Error no engines found. Edit your engines.cfg' print '[E] Error no engines found. Edit your engines.cfg'
exit(2) exit(2)
for section in engines_config.sections(): for engine_config_name in engines_config.sections():
engine_data = engines_config.options(section) engine_data = engines_config.options(engine_config_name)
engine = load_module(engines_config.get(section, 'engine')+'.py') engine = load_module(engines_config.get(engine_config_name, 'engine')+'.py')
engine.name = section engine.name = engine_config_name
for param_name in engine_data: for param_name in engine_data:
if param_name == 'engine': if param_name == 'engine':
continue continue
if param_name == 'categories': if param_name == 'categories':
if engines_config.get(section, param_name) == 'none': if engines_config.get(engine_config_name, param_name) == 'none':
engine.categories = [] engine.categories = []
else: else:
engine.categories = map(str.strip, engines_config.get(section, param_name).split(',')) engine.categories = map(str.strip, engines_config.get(engine_config_name, param_name).split(','))
continue continue
setattr(engine, param_name, engines_config.get(section, param_name)) setattr(engine, param_name, engines_config.get(engine_config_name, param_name))
for engine_attr in dir(engine): for engine_attr in dir(engine):
if engine_attr.startswith('_'): if engine_attr.startswith('_'):
continue continue
@ -161,7 +161,12 @@ def search(query, request, selected_engines):
for i,res in enumerate(flat_res): for i,res in enumerate(flat_res):
res['parsed_url'] = urlparse(res['url']) res['parsed_url'] = urlparse(res['url'])
res['engines'] = [res['engine']] res['engines'] = [res['engine']]
score = int((flat_len - i)/engines_len)*settings.weights.get(res['engine'], 1)+1 weight = 1.0
if hasattr(engines[res['engine']], 'weight'):
weight = float(engines[res['engine']].weight)
elif res['engine'] in settings.weights:
weight = float(settings.weights[res['engine']])
score = int((flat_len - i)/engines_len)*weight+1
duplicated = False duplicated = False
for new_res in results: for new_res in results:
p1 = res['parsed_url'].path[:-1] if res['parsed_url'].path.endswith('/') else res['parsed_url'].path p1 = res['parsed_url'].path[:-1] if res['parsed_url'].path.endswith('/') else res['parsed_url'].path