Rename invirt-setquotas to invirt-quota.

[invirt/packages/invirt-database.git] / python / database / owner.py

diff --git a/python/database/owner.py b/python/database/owner.py
index f9e0128..196bd85 100755 (executable)
--- a/python/database/owner.py
+++ b/python/database/owner.py
@@ -7,15 +7,27 @@ MAX_VMS_ACTIVE = 4
 
 class Owner(object):
     def __repr__(self):
 
 class Owner(object):
     def __repr__(self):

+        a = self.ram_quota_total
+        b = self.ram_quota_single
+        c = self.disk_quota_total
+        d = self.disk_quota_single
+        e = self.vms_quota_total
+        f = self.vms_quota_active
+        if not a:
+            a = MAX_MEMORY_TOTAL
+        if not b:
+            b = MAX_MEMORY_SINGLE
+        if not c:
+            c = MAX_DISK_TOTAL
+        if not d:
+            d = MAX_DISK_SINGLE
+        if not e:
+            e = MAX_VMS_TOTAL
+        if not f:
+            f = MAX_VMS_ACTIVE

         return """<Owner %s: ram_quota_total=%s MB ram_quota_single=%s MB
         return """<Owner %s: ram_quota_total=%s MB ram_quota_single=%s MB

-disk_quota_total=%s MB disk_quota_single=%s MB
-vms_quota_total=%s vms_quota_active=%s >""" % (self.owner_id, 
-               self.ram_quota_total if self.ram_quota_total else MAX_MEMORY_TOTAL,
-               self.ram_quota_single if self.ram_quota_single else MAX_MEMORY_SINGLE, 
-               self.disk_quota_total if self.disk_quota_total else MAX_DISK_TOTAL,
-               self.disk_quota_single if self.disk_quota_single else MAX_DISK_SINGLE, 
-               self.vms_quota_total if self.vms_quota_total else MAX_VMS_TOTAL,
-               self.vms_quota_active if self.vms_quota_active else MAX_VMS_ACTIVE)
+disk_quota_total=%s GB disk_quota_single=%s GB
+vms_quota_total=%s vms_quota_active=%s >""" % (self.owner_id, a,b,c,d,e,f)

     def getMemoryQuotas(owner):
         owner_info = Owner.query().filter_by(owner_id=owner).first()
         if owner_info != None:
     def getMemoryQuotas(owner):
         owner_info = Owner.query().filter_by(owner_id=owner).first()
         if owner_info != None:

@@ -55,6 +67,6 @@ vms_quota_total=%s vms_quota_active=%s >""" % (self.owner_id,
                 quota_active = MAX_VMS_ACTIVE
         else:
             quota_total = MAX_VMS_TOTAL
                 quota_active = MAX_VMS_ACTIVE
         else:
             quota_total = MAX_VMS_TOTAL

-            quota_single = MAX_VMS_ACTIVE
+            quota_active = MAX_VMS_ACTIVE

         return (quota_total, quota_active)
     getVMQuotas = staticmethod(getVMQuotas)
         return (quota_total, quota_active)
     getVMQuotas = staticmethod(getVMQuotas)