diff --git a/apps/utils/permission.py b/apps/utils/permission.py index ade75e09..bb2ce0aa 100755 --- a/apps/utils/permission.py +++ b/apps/utils/permission.py @@ -15,9 +15,10 @@ def get_user_perms_map(user): """ user_perms_map = {} if user.is_superuser: - perms = Permission.objects.all().values_list('code', flat=True) - for i in perms: - user_perms_map[i] = {} + for perm in Permission.objects.all(): + if perm.codes: + for code in perm.codes: + user_perms_map[code] = {} else: objs = UserPost.objects.filter(user=user).exclude(post=None) for i in objs: