Fix jis's bug

[invirt/packages/invirt-web.git] / code / validation.py

diff --git a/code/validation.py b/code/validation.py
index 874dbb0..c0e3aeb 100644 (file)
--- a/code/validation.py
+++ b/code/validation.py
@@ -59,7 +59,7 @@ def maxDisk(user, machine=None):
     else:
         machine_id = None
     disk_usage = Disk.query().filter_by(Disk.c.machine_id != machine_id,
-                                        owner=user).sum(Disk.c.size)
+                                        owner=user).sum(Disk.c.size) or 0
     return min(MAX_DISK_SINGLE, MAX_DISK_TOTAL-disk_usage/1024.)
 
 def cantAddVm(user):