Reformat code

This commit is contained in:
Michael Herrmann
2023-10-16 14:50:09 +02:00
parent d802c48f6c
commit 8953acc2f7
+1 -5
View File
@@ -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