summaryrefslogtreecommitdiff
path: root/searx/shared/redisdb.py
blob: d0071f72c8c569faa8bfb2c354db545d7b81d8ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# SPDX-License-Identifier: AGPL-3.0-or-later
# lint: pylint
"""Implementation of the redis client (redis-py_).

.. _redis-py: https://github.com/redis/redis-py

This implementation uses the :ref:`settings redis` setup from ``settings.yml``.
A redis DB connect can be tested by::

  >>> from searx.shared import redisdb
  >>> redisdb.init()
  True
  >>> db = redisdb.client()
  >>> db.set("foo", "bar")
  True
  >>> db.get("foo")
  b'bar'
  >>>

"""

import os
import pwd
import logging
import redis
from searx import get_setting


logger = logging.getLogger('searx.shared.redisdb')
_client = None


def client() -> redis.Redis:
    return _client


def initialize():
    global _client  # pylint: disable=global-statement
    try:
        _client = redis.Redis.from_url(get_setting('redis.url'))
        logger.info("connected redis: %s", get_setting('redis.url'))
    except redis.exceptions.ConnectionError as exc:
        _pw = pwd.getpwuid(os.getuid())
        logger.error("[%s (%s)] can't connect redis DB ...", _pw.pw_name, _pw.pw_uid)
        logger.error("  %s", exc)