forked from Ponysearch/Ponysearch
[enh] add statistics answerer
This commit is contained in:
parent
971ed0abd1
commit
7eed8a5dd9
1 changed files with 51 additions and 0 deletions
51
searx/answerers/statistics/answerer.py
Normal file
51
searx/answerers/statistics/answerer.py
Normal file
|
@ -0,0 +1,51 @@
|
||||||
|
from functools import reduce
|
||||||
|
from operator import mul
|
||||||
|
|
||||||
|
from flask_babel import gettext
|
||||||
|
|
||||||
|
keywords = ('min',
|
||||||
|
'max',
|
||||||
|
'avg',
|
||||||
|
'sum',
|
||||||
|
'prod')
|
||||||
|
|
||||||
|
|
||||||
|
# required answerer function
|
||||||
|
# can return a list of results (any result type) for a given query
|
||||||
|
def answer(query):
|
||||||
|
parts = query.query.split()
|
||||||
|
|
||||||
|
if len(parts) < 2:
|
||||||
|
return []
|
||||||
|
|
||||||
|
try:
|
||||||
|
args = map(float, parts[1:])
|
||||||
|
except:
|
||||||
|
return []
|
||||||
|
|
||||||
|
func = parts[0]
|
||||||
|
answer = None
|
||||||
|
|
||||||
|
if func == 'min':
|
||||||
|
answer = min(args)
|
||||||
|
elif func == 'max':
|
||||||
|
answer = max(args)
|
||||||
|
elif func == 'avg':
|
||||||
|
answer = sum(args)/len(args)
|
||||||
|
elif func == 'sum':
|
||||||
|
answer = sum(args)
|
||||||
|
elif func == 'prod':
|
||||||
|
answer = reduce(mul, args, 1)
|
||||||
|
|
||||||
|
if answer is None:
|
||||||
|
return []
|
||||||
|
|
||||||
|
return [{'answer': unicode(answer)}]
|
||||||
|
|
||||||
|
|
||||||
|
# required answerer function
|
||||||
|
# returns information about the answerer
|
||||||
|
def self_info():
|
||||||
|
return {'name': gettext('Statistics functions'),
|
||||||
|
'description': gettext('Compute {functions} of the arguments').format(functions='/'.join(keywords)),
|
||||||
|
'examples': ['avg 123 548 2.04 24.2']}
|
Loading…
Reference in a new issue