summaryrefslogtreecommitdiff
path: root/searx/metrics/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'searx/metrics/__init__.py')
-rw-r--r--searx/metrics/__init__.py14
1 files changed, 6 insertions, 8 deletions
diff --git a/searx/metrics/__init__.py b/searx/metrics/__init__.py
index 00a811635..da833a8c9 100644
--- a/searx/metrics/__init__.py
+++ b/searx/metrics/__init__.py
@@ -1,11 +1,9 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring
-import typing
import math
import contextlib
from timeit import default_timer
-from operator import itemgetter
from searx.engines import engines
from searx.openmetrics import OpenMetricsFamily
@@ -30,8 +28,8 @@ __all__ = [
ENDPOINTS = {'search'}
-histogram_storage: typing.Optional[HistogramStorage] = None
-counter_storage: typing.Optional[CounterStorage] = None
+histogram_storage: HistogramStorage = None # type: ignore
+counter_storage: CounterStorage = None # type: ignore
@contextlib.contextmanager
@@ -57,11 +55,11 @@ def histogram(*args, raise_on_not_found=True):
return h
-def counter_inc(*args):
+def counter_inc(*args: str):
counter_storage.add(1, *args)
-def counter_add(value, *args):
+def counter_add(value: int, *args: str):
counter_storage.add(value, *args)
@@ -69,7 +67,7 @@ def counter(*args):
return counter_storage.get(*args)
-def initialize(engine_names=None, enabled=True):
+def initialize(engine_names: list[str] | None = None, enabled: bool = True) -> None:
"""
Initialize metrics
"""
@@ -174,7 +172,7 @@ def get_reliabilities(engline_name_list, checker_results):
return reliabilities
-def get_engines_stats(engine_name_list):
+def get_engines_stats(engine_name_list: list[str]):
assert counter_storage is not None
assert histogram_storage is not None