wownero-funding-system/wowfunding/cache.py

import json

import redis
from flask_session import RedisSessionInterface

import settings
from wowfunding.bin.utils import json_encoder


def redis_args():
    args = {
        "host": settings.REDIS_HOST,
        "port": settings.REDIS_PORT,
        'socket_connect_timeout': 2,
        'socket_timeout': 2,
        'retry_on_timeout': True
    }
    if settings.REDIS_PASSWD:
        args["password"] = settings.REDIS_PASSWD
    return args


class JsonRedisSerializer:
    @staticmethod
    def loads(val):
        try:
            return json.loads(val).get("wow", {})
        except ValueError:
            return

    @staticmethod
    def dumps(val):
        try:
            return json.dumps({"wow": val})
        except ValueError:
            return


class JsonRedis(RedisSessionInterface):
    serializer = JsonRedisSerializer

    def __init__(self, key_prefix, use_signer=False, decode_responses=True):
        super(JsonRedis, self).__init__(
            redis=redis.Redis(decode_responses=decode_responses, **redis_args()),
            key_prefix=key_prefix,
            use_signer=use_signer)


class WowCache:
    def __init__(self):
        self._cache = redis.StrictRedis(**redis_args())

    def get(self, key):
        try:
            return json.loads(self._cache.get(key))
        except:
            return {}

    def set(self, key: str, data: dict, expiry = 300):
        self._cache.set(key, json.dumps(data, default=json_encoder), ex=expiry)