diff options
Diffstat (limited to 'searx/favicons/config.py')
| -rw-r--r-- | searx/favicons/config.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/searx/favicons/config.py b/searx/favicons/config.py new file mode 100644 index 000000000..1c18b1631 --- /dev/null +++ b/searx/favicons/config.py @@ -0,0 +1,62 @@ +# SPDX-License-Identifier: AGPL-3.0-or-later +# pylint: disable=missing-module-docstring + +from __future__ import annotations + +import pathlib +from pydantic import BaseModel + +from searx.compat import tomllib +from .cache import FaviconCacheConfig +from .proxy import FaviconProxyConfig + +CONFIG_SCHEMA: int = 1 +"""Version of the configuration schema.""" + +TOML_CACHE: dict[str, "FaviconConfig"] = {} +"""Cache config objects by TOML's filename.""" + +DEFAULT_CFG_TOML = pathlib.Path(__file__).parent / "favicons.toml" + + +class FaviconConfig(BaseModel): + """The class aggregates configurations of the favicon tools""" + + cfg_schema: int + """Config's schema version. The specification of the version of the schema + is mandatory, currently only version :py:obj:`CONFIG_SCHEMA` is supported. + By specifying a version, it is possible to ensure downward compatibility in + the event of future changes to the configuration schema""" + + cache: FaviconCacheConfig = FaviconCacheConfig() + """Setup of the :py:obj:`.cache.FaviconCacheConfig`.""" + + proxy: FaviconProxyConfig = FaviconProxyConfig() + """Setup of the :py:obj:`.proxy.FaviconProxyConfig`.""" + + @classmethod + def from_toml_file(cls, cfg_file: pathlib.Path, use_cache: bool) -> "FaviconConfig": + """Create a config object from a TOML file, the ``use_cache`` argument + specifies whether a cache should be used. + """ + + cached = TOML_CACHE.get(str(cfg_file)) + if use_cache and cached: + return cached + + with cfg_file.open("rb") as f: + + cfg = tomllib.load(f) + cfg = cfg.get("favicons", cfg) + + schema = cfg.get("cfg_schema") + if schema != CONFIG_SCHEMA: + raise ValueError( + f"config schema version {CONFIG_SCHEMA} is needed, version {schema} is given in {cfg_file}" + ) + + cfg = cls(**cfg) + if use_cache and cached: + TOML_CACHE[str(cfg_file.resolve())] = cfg + + return cfg |