j3d1/webvirtcloud
1
0
Fork 0
mirror of https://github.com/retspen/webvirtcloud synced 2025-07-31 12:41:08 +00:00
Code Issues Projects Releases Wiki Activity

Get UserAttributes object using get_or_create

Browse source 
This is done to automatically create the UserAttributes object in case LDAP User Backend didn't create it.
This commit is contained in:
Amélie Krejčí 2023-09-23 09:52:48 +02:00
parent ca54ae0c65
commit 1b2b3a3bce
No known key found for this signature in database
GPG key ID: 65B7BB4B1A0D9402
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
admin/views.py
View file

@ -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

4
instances/utils.py
View file

@ -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: