Adding model to NIC table

[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()
     
-    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