mirror of
https://github.com/retspen/webvirtcloud
synced 2024-11-01 03:54:15 +00:00
f8e681dbf4
added quota warning in instance page
189 lines
8 KiB
Python
189 lines
8 KiB
Python
from django.shortcuts import render
|
|
from django.http import HttpResponseRedirect
|
|
from django.core.urlresolvers import reverse
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth.decorators import login_required
|
|
from accounts.models import *
|
|
from instances.models import Instance
|
|
from accounts.forms import UserAddForm
|
|
from django.conf import settings
|
|
|
|
|
|
@login_required
|
|
def profile(request):
|
|
"""
|
|
:param request:
|
|
:return:
|
|
"""
|
|
|
|
error_messages = []
|
|
user = User.objects.get(id=request.user.id)
|
|
publickeys = UserSSHKey.objects.filter(user_id=request.user.id)
|
|
|
|
if request.method == 'POST':
|
|
if 'username' in request.POST:
|
|
username = request.POST.get('username', '')
|
|
email = request.POST.get('email', '')
|
|
user.first_name = username
|
|
user.email = email
|
|
user.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'oldpasswd' in request.POST:
|
|
oldpasswd = request.POST.get('oldpasswd', '')
|
|
password1 = request.POST.get('passwd1', '')
|
|
password2 = request.POST.get('passwd2', '')
|
|
if not password1 or not password2:
|
|
error_messages.append("Passwords didn't enter")
|
|
if password1 and password2 and password1 != password2:
|
|
error_messages.append("Passwords don't match")
|
|
if not user.check_password(oldpasswd):
|
|
error_messages.append("Old password is wrong!")
|
|
if not error_messages:
|
|
user.set_password(password1)
|
|
user.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'keyname' in request.POST:
|
|
keyname = request.POST.get('keyname', '')
|
|
keypublic = request.POST.get('keypublic', '')
|
|
for key in publickeys:
|
|
if keyname == key.keyname:
|
|
msg = _("Key name already exist")
|
|
error_messages.append(msg)
|
|
if keypublic == key.keypublic:
|
|
msg = _("Public key already exist")
|
|
error_messages.append(msg)
|
|
if not error_messages:
|
|
addkeypublic = UserSSHKey(user_id=request.user.id, keyname=keyname, keypublic=keypublic)
|
|
addkeypublic.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'keydelete' in request.POST:
|
|
keyid = request.POST.get('keyid', '')
|
|
delkeypublic = UserSSHKey.objects.get(id=keyid)
|
|
delkeypublic.delete()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
return render(request, 'profile.html', locals())
|
|
|
|
@login_required
|
|
def accounts(request):
|
|
"""
|
|
:param request:
|
|
:return:
|
|
"""
|
|
|
|
if not request.user.is_superuser:
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
error_messages = []
|
|
users = User.objects.all()
|
|
|
|
if request.method == 'POST':
|
|
if 'create' in request.POST:
|
|
form = UserAddForm(request.POST)
|
|
if form.is_valid():
|
|
data = form.cleaned_data
|
|
else:
|
|
for msg_err in form.errors.values():
|
|
error_messages.append(msg_err.as_text())
|
|
if not error_messages:
|
|
new_user = User.objects.create_user(data['name'], None, data['password'])
|
|
new_user.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'edit' in request.POST:
|
|
user_id = request.POST.get('user_id', '')
|
|
user_pass = request.POST.get('user_pass', '')
|
|
user_edit = User.objects.get(id=user_id)
|
|
user_edit.set_password(user_pass)
|
|
user_edit.is_staff = request.POST.get('user_is_staff', False)
|
|
user_edit.is_superuser = request.POST.get('user_is_superuser', False)
|
|
user_edit.save()
|
|
try:
|
|
userattributes = user_edit.userattributes
|
|
except UserAttributes.DoesNotExist:
|
|
userattributes = UserAttributes(user=user_edit)
|
|
userattributes.can_clone_instances = request.POST.get('userattributes_can_clone_instances', False)
|
|
userattributes_max_instances = request.POST.get('userattributes_max_instances', 0)
|
|
userattributes_max_cpus = request.POST.get('userattributes_max_cpus', 0)
|
|
userattributes_max_memory = request.POST.get('userattributes_max_memory', 0)
|
|
userattributes_max_disk_size = request.POST.get('userattributes_max_disk_size', 0)
|
|
userattributes.max_instances = userattributes_max_instances if userattributes_max_instances else 0
|
|
userattributes.max_cpus = userattributes_max_cpus if userattributes_max_cpus else 0
|
|
userattributes.max_memory = userattributes_max_memory if userattributes_max_memory else 0
|
|
userattributes.max_disk_size = userattributes_max_disk_size if userattributes_max_disk_size else 0
|
|
userattributes.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'block' in request.POST:
|
|
user_id = request.POST.get('user_id', '')
|
|
user_block = User.objects.get(id=user_id)
|
|
user_block.is_active = False
|
|
user_block.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'unblock' in request.POST:
|
|
user_id = request.POST.get('user_id', '')
|
|
user_unblock = User.objects.get(id=user_id)
|
|
user_unblock.is_active = True
|
|
user_unblock.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'delete' in request.POST:
|
|
user_id = request.POST.get('user_id', '')
|
|
try:
|
|
del_user_inst = UserInstance.objects.filter(user_id=user_id)
|
|
del_user_inst.delete()
|
|
finally:
|
|
user_delete = User.objects.get(id=user_id)
|
|
user_delete.delete()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
|
|
return render(request, 'accounts.html', locals())
|
|
|
|
|
|
@login_required
|
|
def account(request, user_id):
|
|
"""
|
|
:param request:
|
|
:return:
|
|
"""
|
|
|
|
if not request.user.is_superuser:
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
error_messages = []
|
|
user = User.objects.get(id=user_id)
|
|
user_insts = UserInstance.objects.filter(user_id=user_id)
|
|
instances = Instance.objects.all().order_by('name')
|
|
|
|
if user.username == request.user.username:
|
|
return HttpResponseRedirect(reverse('profile'))
|
|
|
|
if request.method == 'POST':
|
|
if 'delete' in request.POST:
|
|
user_inst = request.POST.get('user_inst', '')
|
|
del_user_inst = UserInstance.objects.get(id=user_inst)
|
|
del_user_inst.delete()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'permission' in request.POST:
|
|
user_inst = request.POST.get('user_inst', '')
|
|
inst_change = request.POST.get('inst_change', '')
|
|
inst_delete = request.POST.get('inst_delete', '')
|
|
edit_user_inst = UserInstance.objects.get(id=user_inst)
|
|
edit_user_inst.is_change = bool(inst_change)
|
|
edit_user_inst.is_delete = bool(inst_delete)
|
|
edit_user_inst.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if 'add' in request.POST:
|
|
inst_id = request.POST.get('inst_id', '')
|
|
|
|
if settings.ALLOW_INSTANCE_MULTIPLE_OWNER:
|
|
check_inst = UserInstance.objects.filter(instance_id=int(inst_id), user_id=int(user_id))
|
|
else:
|
|
check_inst = UserInstance.objects.filter(instance_id=int(inst_id))
|
|
|
|
if check_inst:
|
|
msg = _("Instance already added")
|
|
error_messages.append(msg)
|
|
else:
|
|
add_user_inst = UserInstance(instance_id=int(inst_id), user_id=int(user_id))
|
|
add_user_inst.save()
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
|
|
return render(request, 'account.html', locals())
|