X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/a754e7beadaaa4d401bfeaac5b8b67df29fb7893..d782699e42f87e77f13dc63f9c7d81429c0f7ff9:/scripts/invirt-quota

diff --git a/scripts/invirt-quota b/scripts/invirt-quota
index f0fb16f..1efdae6 100755
--- a/scripts/invirt-quota
+++ b/scripts/invirt-quota
@@ -47,7 +47,7 @@ def main(argv):
     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)
 
@@ -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)
+            database.session.add(owner)
 
     database.session.commit()
     print owner