diff --git a/admin/views.py b/admin/views.py index fd4ead4..8b71baf 100644 --- a/admin/views.py +++ b/admin/views.py @@ -117,7 +117,7 @@ def user_create(request): @superuser_only def user_update(request, pk): user = get_object_or_404(User, pk=pk) - attributes = UserAttributes.objects.get(user=user) + attributes, attributes_created = UserAttributes.objects.get_or_create(user=user) user_form = forms.UserForm(request.POST or None, instance=user) attributes_form = forms.UserAttributesForm( request.POST or None, instance=attributes diff --git a/instances/utils.py b/instances/utils.py index 53752b6..f9f5dcf 100644 --- a/instances/utils.py +++ b/instances/utils.py @@ -2,7 +2,7 @@ import os import random import string -from accounts.models import UserInstance +from accounts.models import UserInstance, UserAttributes from appsettings.settings import app_settings from django.conf import settings from django.utils.translation import gettext_lazy as _ @@ -26,7 +26,7 @@ def get_clone_free_names(size=10): def check_user_quota(user, instance, cpu, memory, disk_size): - ua = user.userattributes + ua, attributes_created = UserAttributes.objects.get_or_create(user=user) msg = "" if user.is_superuser: