diff options
Diffstat (limited to 'searx/enginelib')
| -rw-r--r-- | searx/enginelib/traits.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/searx/enginelib/traits.py b/searx/enginelib/traits.py index 6402fde22..e1a5df79a 100644 --- a/searx/enginelib/traits.py +++ b/searx/enginelib/traits.py @@ -14,8 +14,7 @@ from __future__ import annotations import json import dataclasses import types -from typing import Dict, Iterable, Union, Callable, Optional, TYPE_CHECKING -from typing_extensions import Literal, Self +from typing import Dict, Literal, Iterable, Union, Callable, Optional, TYPE_CHECKING from searx import locales from searx.data import data_dir, ENGINE_TRAITS @@ -136,7 +135,7 @@ class EngineTraits: return EngineTraits(**dataclasses.asdict(self)) @classmethod - def fetch_traits(cls, engine: Engine) -> Union[Self, None]: + def fetch_traits(cls, engine: Engine) -> Union['EngineTraits', None]: """Call a function ``fetch_traits(engine_traits)`` from engines namespace to fetch and set properties from the origin engine in the object ``engine_traits``. If function does not exists, ``None`` is returned. @@ -204,7 +203,7 @@ class EngineTraitsMap(Dict[str, EngineTraits]): json.dump(self, f, indent=2, sort_keys=True, cls=EngineTraitsEncoder) @classmethod - def from_data(cls) -> Self: + def from_data(cls) -> 'EngineTraitsMap': """Instantiate :class:`EngineTraitsMap` object from :py:obj:`ENGINE_TRAITS`""" obj = cls() for k, v in ENGINE_TRAITS.items(): @@ -212,7 +211,7 @@ class EngineTraitsMap(Dict[str, EngineTraits]): return obj @classmethod - def fetch_traits(cls, log: Callable) -> Self: + def fetch_traits(cls, log: Callable) -> 'EngineTraitsMap': from searx import engines # pylint: disable=cyclic-import, import-outside-toplevel names = list(engines.engines) |