Configuration helpers for flask-cache

utils.py

from six.moves import urllib

def config(uri):
    parsed = urllib.parse.urlparse(uri)
    options = dict(urllib.parse.parse_qsl(parsed.query))
    cache_type = parsed.scheme.lower()

    if cache_type not in ['null', 'simple', 'file', 'memcached', 'saslmemcached', 'redis']:
        raise Exception('Misconfiguration')

    def _config():
        yield 'CACHE_TYPE', cache_type
        if 'timeout' in options:
            yield 'CACHE_DEFAULT_TIMEOUT', int(options['timeout'])
        if 'prefix' in options:
            yield 'CACHE_KEY_PREFIX', options['prefix']

        if cache_type in ['simple', 'file']:
            if 'threshold' in options:
                yield 'CACHE_THRESHOLD', int(options['threshold'])
        if cache_type in ['file']:
            yield 'CACHE_DIR', parsed.path[1:]
        if cache_type in ['memcached', 'saslmemcached']:
            yield 'CACHE_MEMCACHED_SERVERS', parsed.host
        if cache_type in ['saslmemcached']:
            yield 'CACHE_MEMCACHED_USERNAME', parsed.username
            yield 'CACHE_MEMCACHED_PASSWORD', parsed.password
        if cache_type in ['redis']:
            yield 'CACHE_REDIS_HOST', parsed.hostname
            if parsed.path:
                yield 'CACHE_REDIS_DB', int(parsed.path[1:])
            if parsed.port:
                yield 'CACHE_REDIS_PORT', parsed.port
            if parsed.password:
                yield 'CACHE_REDIS_PASSWORD', parsed.password
    return dict(_config())