small fixes to Record

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

diff --git a/python/database/owner.py b/python/database/owner.py
index adb176c..43e7ca4 100755 (executable)
--- a/python/database/owner.py
+++ b/python/database/owner.py
@@ -15,26 +15,23 @@ class Owner(NullableRecord):
     _format = dict([(_k,_unitFormatter(_v[1])) for _k,_v in _f.items()])
     _identity_field = 'owner_id'
 
     _format = dict([(_k,_unitFormatter(_v[1])) for _k,_v in _f.items()])
     _identity_field = 'owner_id'
 

-    def getMemoryQuotas(owner):
-        owner_info = Owner.query().filter_by(owner_id=owner).first()
+    @classmethod
+    def getMemoryQuotas(cls, owner):
+        owner_info = cls.query().filter_by(owner_id=owner).first()

         if owner_info == None:
         if owner_info == None:

-            owner_info = Owner(owner_id=owner)
+            owner_info = cls(owner_id=owner)

         return (owner_info.get('ram_quota_total'), owner_info.get('ram_quota_single'))
         return (owner_info.get('ram_quota_total'), owner_info.get('ram_quota_single'))

-    getMemoryQuotas = staticmethod(getMemoryQuotas)

 
 

-    def getDiskQuotas(owner):
-        owner_info = Owner.query().filter_by(owner_id=owner).first()
+    @classmethod
+    def getDiskQuotas(cls, owner):
+        owner_info = cls.query().filter_by(owner_id=owner).first()

         if owner_info == None:
         if owner_info == None:

-            owner_info = Owner(owner_id=owner)
+            owner_info = cls(owner_id=owner)

         return (owner_info.get('disk_quota_total'), owner_info.get('disk_quota_single'))
         return (owner_info.get('disk_quota_total'), owner_info.get('disk_quota_single'))

-    getDiskQuotas = staticmethod(getDiskQuotas)

 
 

-    def getVMQuotas(owner):
-        owner_info = Owner.query().filter_by(owner_id=owner).first()
+    @classmethod
+    def getVMQuotas(cls, owner):
+        owner_info = cls.query().filter_by(owner_id=owner).first()

         if owner_info == None:
         if owner_info == None:

-            owner_info = Owner(owner_id=owner)
+            owner_info = cls(owner_id=owner)

         return (owner_info.get('vms_quota_total'), owner_info.get('vms_quota_active'))
         return (owner_info.get('vms_quota_total'), owner_info.get('vms_quota_active'))

-    getVMQuotas = staticmethod(getVMQuotas)
-
-    def _ignore(self):
-        return super(Owner, self)._ignore() + ['getMemoryQuotas', 'getDiskQuotas', 'getVMQuotas']