mirror of https://github.com/searxng/searxng.git
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
from __future__ import annotations
|
|
from functools import reduce
|
|
from operator import mul
|
|
|
|
from flask_babel import gettext
|
|
from typing import Callable
|
|
from searx.answerers.models import AnswerDict, AnswerSelfInfoDict
|
|
from searx.search.models import BaseQuery
|
|
|
|
|
|
keywords = ('min', 'max', 'avg', 'sum', 'prod')
|
|
|
|
|
|
stastistics_map: dict[str, Callable[[list[float]], float]] = {
|
|
'min': lambda args: min(args),
|
|
'max': lambda args: max(args),
|
|
'avg': lambda args: sum(args) / len(args),
|
|
'sum': lambda args: sum(args),
|
|
'prod': lambda args: reduce(mul, args, 1),
|
|
}
|
|
|
|
|
|
# required answerer function
|
|
# can return a list of results (any result type) for a given query
|
|
def answer(query: BaseQuery) -> list[AnswerDict]:
|
|
parts = query.query.split()
|
|
|
|
if len(parts) < 2:
|
|
return []
|
|
|
|
try:
|
|
args: list[float] = list(map(float, parts[1:]))
|
|
except Exception:
|
|
return []
|
|
|
|
func = parts[0]
|
|
|
|
if func not in stastistics_map:
|
|
return []
|
|
|
|
return [{'answer': str(stastistics_map[func](args))}]
|
|
|
|
|
|
# required answerer function
|
|
# returns information about the answerer
|
|
def self_info() -> AnswerSelfInfoDict:
|
|
return {
|
|
'name': gettext('Statistics functions'),
|
|
'description': gettext('Compute {functions} of the arguments').format(functions='/'.join(keywords)),
|
|
'examples': ['avg 123 548 2.04 24.2'],
|
|
}
|