Changeset 2213 for trunk/packages/invirt-database/python

Timestamp:

Feb 27, 2009, 2:18:03 PM (16 years ago)

Author:

quentin

Message:

Fix get*Quotas

Location:

trunk/packages/invirt-database/python/database

Files:

• models.py (modified) (2 diffs)
• owner.py (modified) (2 diffs)

trunk/packages/invirt-database/python/database/models.py

@@ -7 +7 @@
 
 from invirt.database import record
-from invirt.database.owner import Owner
 
 __all__ = ['meta',
@@ -127 +126 @@
     _identity_field = 'autoinstall_id'
 
+from invirt.database.owner import Owner
+
 session.mapper(Machine, machine_table,
               properties={'nics': relation(NIC, backref="machine"),

trunk/packages/invirt-database/python/database/owner.py

@@ -1 +1 @@
 from invirt.database import record
+from invirt.database.models import session
 
 class Owner(record.Record):
@@ -18 +19 @@
     @classmethod
     def getMemoryQuotas(cls, owner):
-        owner_info = cls.query().filter_by(owner_id=owner).first()
+        owner_info = cls.query().get(owner)
         if owner_info == None:
             owner_info = cls(owner_id=owner)
+            session.expunge(owner_info)
         return (owner_info.get('ram_quota_total'), owner_info.get('ram_quota_single'))
 
     @classmethod
     def getDiskQuotas(cls, owner):
-        owner_info = cls.query().filter_by(owner_id=owner).first()
+        owner_info = cls.query().get(owner)
         if owner_info == None:
             owner_info = cls(owner_id=owner)
+            session.expunge(owner_info)
         return (owner_info.get('disk_quota_total'), owner_info.get('disk_quota_single'))
 
     @classmethod
     def getVMQuotas(cls, owner):
-        owner_info = cls.query().filter_by(owner_id=owner).first()
+        owner_info = cls.query().get(owner)
         if owner_info == None:
             owner_info = cls(owner_id=owner)
+            session.expunge(owner_info)
         return (owner_info.get('vms_quota_total'), owner_info.get('vms_quota_active'))