gistfile1.txt

from django.conf import settings
from django.core.cache import parse_backend_uri

try:
    import redis as redislib
except:
    redislib = None

connections = {}

if not connections:  # don't set this repeatedly
    for alias, backend in settings.REDIS_BACKENDS.items():
        _, server, params = parse_backend_uri(backend)

        try:
            socket_timeout = float(params.pop('socket_timeout'))
        except (KeyError, ValueError):
            socket_timeout = None
        password = params.pop('password', None)
        if ':' in server:
            host, port = server.split(':')
            try:
                port = int(port)
            except (ValueError, TypeError):
                port = 6379
        else:
            host = 'localhost'
            port = 6379

        connections[alias] = redislib.Redis(host=host, port=port, db=0,
                                            password=password,
                                            socket_timeout=socket_timeout)


def mock_redis():
    connections['master'] = MockRedis()


class MockRedis(object):
    """A fake redis we can use for testing."""
    kv = {}

    def pipeline(self, **kw):
        return self

    def execute(self):
        pass

    def get(self, key):
        return self.kv.get(key)

    def set(self, key, val):
        self.kv[key] = val

    def sadd(self, key, val):
        v = self.kv.get(key)
        if not v:
            self.kv[key] = set()
            v = self.kv[key]
        if isinstance(v, set):
            v.add(val)
            return True
        return False

    def smembers(self, key):
        v = self.kv.get(key, set())
        if isinstance(v, set):
            return v

    def delete(self, key):
        if key in self.kv:
            del self.kv[key]
            return True
        return False

56: dict.setdefault