import logging
from functools import wraps
from os import environ
from pynyaata.cache.simple import SimpleCache
from pynyaata.types import Cache
from redis import RedisError


CACHE_TIMEOUT = int(environ.get("CACHE_TIMEOUT", 60 * 60))
REDIS_URL = environ.get("REDIS_URL", "")

client: Cache = SimpleCache()
if REDIS_URL:
    try:
        from pynyaata.cache.redis import RedisCache

        client = RedisCache()
    except RedisError as e:
        logging.error(e)


def cache_data(f):
    @wraps(f)
    def wrapper(*args, **kwds):
        bridge = args[0]
        key = f"pynyaata.{bridge.__class__.__name__}.{f.__name__}.{bridge.query}.{bridge.page}"
        ret = client.get(key)

        if ret:
            return ret

        ret = f(*args, **kwds)
        client.set(key, ret)

        return ret

    return wrapper