"""Redis Connector.

This is just a simple Redis connector that provides a Redis Client
or a FakeRedis client in the context of testing.
"""

import redis

from availability import config


if config.ENVIRONMENT == 'test':
    import fakeredis
    redis_client_cls = fakeredis.FakeStrictRedis
else:
    redis_client_cls = redis.StrictRedis

redis_client = redis_client_cls(**config.REDIS_CONFIG)