forked from Ponysearch/Ponysearch
63 lines
2 KiB
Python
63 lines
2 KiB
Python
|
# 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
|