from cachelib import BaseCache
from cachelib.simple import SimpleCache
from cachelib.uwsgi import UWSGICache

from payee.config import Config


def get_uwsgi_cache_object() -> BaseCache:
    """Get uwsgi cache object.

    Returns:
        object: uwsgi cache object
    """
    if Config.ENVIRONMENT == Config.DEV_ENVIRONMENT:
        return SimpleCache()

    return UWSGICache(cache='uwsgi_cache')