frankpolo

frankpolo / tenant_aware_managers.py

Created May 8, 2021 13:23 — forked from dlederle/tenant_aware_managers.py

	class TenantAwareManager(Manager):
	def get_queryset(self):
	tenant_id = current_tenant_id()

	# If the manager was built from a queryset using
	# SomeQuerySet.as_manager() or SomeManager.from_queryset(),
	# we want to use that queryset instead of TenantAwareQuerySet.
	if self._queryset_class != QuerySet:
	return super().get_queryset().filter(tenant__id=tenant_id)

frankpolo / tenant_context_manager.py

Created May 8, 2021 13:23 — forked from dlederle/tenant_context_manager.py

	@contextmanager
	def tenant(value):
	"""
	Context manager for tenants. Used to set and cleanup tennant.
	param, value, can be Tenant object or id.

	Using the context context manager
	```python
	with tenant(1):
	Profile.objects.get(pk=1)