summaryrefslogtreecommitdiff
path: root/searx/botdetection/valkeydb.py
blob: 3b8699786e57cbf3eec40c395395bf73ae11b2c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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