Don't assume all object have a proxy access accessor.

This commit is contained in:
Roberto Rosario
2016-03-18 20:57:39 -04:00
parent 042cef3d0e
commit 66a49c8335

View File

@@ -26,7 +26,11 @@ class ModelPermission(object):
permissions = []
permissions.extend(cls._registry.get(type(instance)))
permissions.extend(cls._registry.get(cls._proxies.get(type(instance))))
proxy = cls._proxies.get(type(instance))
if proxy:
permissions.extend(cls._registry.get(proxy))
pks = [permission.stored_permission.pk for permission in set(permissions)]
return StoredPermission.objects.filter(pk__in=pks)