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

app.py

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

proxmox.py

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

static/css/styles.css

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

templates/create.html

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