class TenantRequestFactory(RequestFactory):
tm = TenantMainMiddleware(lambda r: r)
def __init__(self, tenant, **defaults):
super().__init__(**defaults)
self.tenant = tenant
def generic(self, *args, **kwargs):
if "HTTP_HOST" not in kwargs:
kwargs["HTTP_HOST"] = self.tenant.get_primary_domain().domain
return super().generic(*args, **kwargs)
class TenantClient(Client):
tm = TenantMainMiddleware()
def __init__(self, tenant, enforce_csrf_checks=False, **defaults):
super().__init__(enforce_csrf_checks, **defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super().get(path, data, **extra)
def post(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super().post(path, data, **extra)
def patch(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super().patch(path, data, **extra)
def put(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super().put(path, data, **extra)
def delete(self, path, data='', **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super().delete(path, data, **extra)
def login(self, **credentials):
# Create a dummy HttpRequest object and add HTTP_HOST
request = HttpRequest()
request.META['HTTP_HOST'] = self.tenant.get_primary_domain().domain
request.tenant = self.tenant
# Authenticate using django contrib's authenticate which passes the request on
# to custom backends
user = authenticate(request, **credentials)
if user:
super(TenantClient, self)._login(user)
return True
else:
return False
class TenantClient(Client):
tm = TenantMainMiddleware()
def __init__(self, tenant, enforce_csrf_checks=False, **defaults):
super(TenantClient, self).__init__(enforce_csrf_checks, **defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantClient, self).get(path, data, **extra)
def post(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantClient, self).post(path, data, **extra)
def patch(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantClient, self).patch(path, data, **extra)
def put(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantClient, self).put(path, data, **extra)
def delete(self,
path,
data='',
content_type='application/octet-stream',
**extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantClient, self).delete(path, data, **extra)
class TenantRequestFactory(RequestFactory):
tm = TenantMainMiddleware()
def __init__(self, tenant, **defaults):
super(TenantRequestFactory, self).__init__(**defaults)
self.tenant = tenant
def get(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantRequestFactory, self).get(path, data, **extra)
def post(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantRequestFactory, self).post(path, data, **extra)
def patch(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantRequestFactory, self).patch(path, data, **extra)
def put(self, path, data={}, **extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantRequestFactory, self).put(path, data, **extra)
def delete(self,
path,
data='',
content_type='application/octet-stream',
**extra):
if 'HTTP_HOST' not in extra:
extra['HTTP_HOST'] = self.tenant.get_primary_domain().domain
return super(TenantRequestFactory, self).delete(path, data, **extra)