dont show create vm page if the user doesnt have enough resources

This commit is contained in:
Jordan Rodgers 2017-12-03 23:55:46 -05:00
parent 49fcb0b663
commit 4d0fb7ece0
4 changed files with 34 additions and 3 deletions

11
app.py
View file

@ -58,7 +58,13 @@ def create():
if request.method == 'GET':
usage = get_user_usage(proxmox, 'proxstar')
limits = get_user_usage_limits(user)
return render_template('create.html', username='com6056', usage=usage, limits=limits)
full_limits = check_user_limit(proxmox, user, usage, limits)
return render_template(
'create.html',
username='com6056',
usage=usage,
limits=limits,
full_limits=full_limits)
elif request.method == 'POST':
name = request.form['name']
cores = request.form['cores']
@ -68,7 +74,8 @@ def create():
if usage_check:
return usage_check
else:
vmid, mac = create_vm(proxmox, starrs, user, name, cores, memory, disk)
vmid, mac = create_vm(proxmox, starrs, user, name, cores, memory,
disk)
register_starrs(starrs, name, user, mac,
get_next_ip(starrs, '49net Public Fixed')[0][0])
return redirect("/proxstar/vm/{}".format(vmid))

View file

@ -123,6 +123,21 @@ def get_user_usage(proxmox, user):
return usage
def check_user_limit(proxmox, user, usage=None, limits=None):
full_limits = []
if not usage:
usage = get_user_usage(proxmox, user)
if not limits:
limits = get_user_usage_limits(user)
if usage['cpu'] >= limits['cpu']:
full_limits.append('CPU')
if usage['mem'] >= limits['mem']:
full_limits.append('Memory')
if usage['disk'] >= limits['disk']:
full_limits.append('Disk')
return full_limits
def check_user_usage(proxmox, user, vm_cpu, vm_mem, vm_disk):
limits = get_user_usage_limits(user)
cur_usage = get_user_usage(proxmox, user)

View file

@ -24,7 +24,6 @@ table, th, td {
margin: 5px 0 0 0;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
}
.panel p a {

View file

@ -9,6 +9,15 @@
<h3 class="panel-title">Create VM</h3>
</div>
<div class="panel-body">
{% if full_limits %}
<p>You have reached your limit for the following resources:</p>
<ul>
{% for limit in full_limits %}
<li>{{ limit }}</li>
{% endfor %}
</ul>
<p>Before you can create any more VMs, you must first either power off (CPU/Memory) or delete (Disk) existing VMs until you have enough resources available.</p>
{% else %}
<form action="create" method="post">
<div class="form-group">
<label for="name">VM Name</label>
@ -36,6 +45,7 @@
</div>
<button class="btn btn-success" type="submit" value="Create">Create</button>
</form>
{% endif %}
</div>
</div>
</div>