diff options
Diffstat (limited to 'searx/botdetection/valkeydb.py')
| -rw-r--r-- | searx/botdetection/valkeydb.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/searx/botdetection/valkeydb.py b/searx/botdetection/valkeydb.py new file mode 100644 index 000000000..3b8699786 --- /dev/null +++ b/searx/botdetection/valkeydb.py @@ -0,0 +1,22 @@ +# SPDX-License-Identifier: AGPL-3.0-or-later +"""Providing a Valkey database for the botdetection methods.""" + +from __future__ import annotations + +import valkey + +__all__ = ["set_valkey_client", "get_valkey_client"] + +CLIENT: valkey.Valkey | None = None +"""Global Valkey DB connection (Valkey client object).""" + + +def set_valkey_client(valkey_client: valkey.Valkey): + global CLIENT # pylint: disable=global-statement + CLIENT = valkey_client + + +def get_valkey_client() -> valkey.Valkey: + if CLIENT is None: + raise ValueError("No connection to the Valkey database has been established.") + return CLIENT |