django-db-multitenant redis mapper example

redis_tenant_mapper.py

"""Maps a request to a tenant using Redis.

Redis should store a map of hostname -> tenant name.  The tenant name
is then used to form the database and cache names.
"""

from db_multitenant import mapper
from django.http import Http404
import redis

REDIS_POOL = redis.ConnectionPool(host='localhost', port=6379, db=0)

class RedisTenantMapper(mapper.TenantMapper):
    def get_tenant_name(self, request):
       """Assumes Redis maps hostname -> tenant name."""
        hostname = request.get_host().split(':')[0].lower()
        r = redis.Redis(connection_pool=REDIS_POOL)
        name = r.get(hostname)
        if not name:
            raise Http404('Unknown tenant for hostname: "%s"' % hostname)
        return name

    def get_dbname(self, request):
        """Returns tenant-<tenant_name>, using tenant name from redis."""
        return 'tenant-%s' % self.get_tenant_name(request)

    def get_cache_prefix(self, request):
        """Returns tenant-<tenant_name>, using tenant name from redis."""
        return 'tenant-%s' % self.get_tenant_name(request)