diff --git a/tenants/access.py b/tenants/access.py index d3ba646..a45bc14 100644 --- a/tenants/access.py +++ b/tenants/access.py @@ -1,20 +1,16 @@ from django.core.exceptions import FieldDoesNotExist - def can_pick_tenant(user): return user.is_superuser - def filter_for_user(model, user): tenant_accessor = get_tenant_accessor(model) query = {tenant_accessor + '__users': user} return model.objects.filter(**query) - def get_tenant_accessor(model): return _get_tenant_accessor(model, set()) - def _get_tenant_accessor(model, visited): visited.add(model) try: @@ -31,4 +27,4 @@ def _get_tenant_accessor(model, visited): return field.name + '__' + subquery raise ValueError('No tenant field found') else: - return tenant_field.name + return tenant_field.name \ No newline at end of file