more workaround for API breakage

[invirt/packages/invirt-database.git] / scripts / invirt-quota

diff --git a/scripts/invirt-quota b/scripts/invirt-quota
index f0fb16f..1efdae6 100755 (executable)
--- a/scripts/invirt-quota
+++ b/scripts/invirt-quota
@@ -47,7 +47,7 @@ def main(argv):
     database.connect()
     database.session.begin()
     
     database.connect()
     database.session.begin()
     

-    owner = database.Owner.query().filter_by(owner_id=owner_id).first()
+    owner = database.session.query(database.Owner).filter_by(owner_id=owner_id).first()

     if owner is None:
         owner = database.Owner(owner_id=owner_id)
 
     if owner is None:
         owner = database.Owner(owner_id=owner_id)
 

@@ -57,6 +57,7 @@ def main(argv):
         val = getattr(opts, resource+scope)
         if val is not None:
             setattr(owner, resource+'_quota_'+scope, val if val >= 0 else None)
         val = getattr(opts, resource+scope)
         if val is not None:
             setattr(owner, resource+'_quota_'+scope, val if val >= 0 else None)

+            database.session.add(owner)

 
     database.session.commit()
     print owner
 
     database.session.commit()
     print owner