diff --git a/collabs/admin.py b/collabs/admin.py index c713184..ba055cd 100644 --- a/collabs/admin.py +++ b/collabs/admin.py @@ -69,17 +69,18 @@ class Collabs_hour_Admin(admin.ModelAdmin): if request.user.is_superuser: #kwargs['form'] = Collabs_hour_Form_admin # ModelForm print("PASS SUPERUSER") - if ( obj.userName == request.user.username or obj.user == request.user) and not obj.bNoticed: - permission = Permission.objects.get(codename='change_collabs_hour') - request.user.user_permissions.add(permission) - request.user = User.objects.get(pk=request.user.id) - print("CAN EDIT") - else: - permission = Permission.objects.get(codename='change_collabs_hour') - request.user.user_permissions.remove(permission) - request.user.user_permissions.clear() - request.user = User.objects.get(pk=request.user.id) - print("RESET CAN EDIT") + if obj != None: + if (obj.userName == request.user.username or obj.user == request.user) and not obj.bNoticed: + permission = Permission.objects.get(codename='change_collabs_hour') + request.user.user_permissions.add(permission) + request.user = User.objects.get(pk=request.user.id) + print("CAN EDIT") + else: + permission = Permission.objects.get(codename='change_collabs_hour') + request.user.user_permissions.remove(permission) + request.user.user_permissions.clear() + request.user = User.objects.get(pk=request.user.id) + print("RESET CAN EDIT") return super().get_form(request, obj, **kwargs)