X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-remote.git/blobdiff_plain/4801dff9f386623c83782f01dd8f3eed89dfff47..faa65854a2d76093c2460b18804996c072b8a9e7:/server/usr/sbin/invirt-remote-create?ds=sidebyside

diff --git a/server/usr/sbin/invirt-remote-create b/server/usr/sbin/invirt-remote-create
index 9f5817a..47939e9 100755
--- a/server/usr/sbin/invirt-remote-create
+++ b/server/usr/sbin/invirt-remote-create
@@ -18,7 +18,7 @@ def maxMemory(owner, xmlist):
     """
     Return the memory available for a new machine.
     """
-    machines = invirt.database.Machine.query().filter_by(owner=owner)
+    machines = invirt.database.Machine.query.filter_by(owner=owner)
     (quota_total, quota_single) = invirt.database.Owner.getMemoryQuotas(owner)
 
     active_machines = [m for m in machines if m.name in xmlist]
@@ -42,7 +42,7 @@ def main(argv):
 
     if operation == 'install':
         options = dict(arg.split('=', 1) for arg in args)
-        valid_keys = set(('mirror', 'dist', 'arch', 'imagesize', 'noinstall', 'preseed'))
+        valid_keys = set(('mirror', 'dist', 'arch', 'imagesize', 'noinstall', 'ks', 'preseed'))
         if not set(options.keys()).issubset(valid_keys):
             print >> sys.stderr, "Invalid argument. Use the help command to see valid arguments to install"
             return 1
@@ -64,7 +64,7 @@ def main(argv):
         return 1
 
     if operation == "create":
-        machine = invirt.database.Machine.query().filter_by(name=machine_name).first()
+        machine = invirt.database.Machine.query.filter_by(name=machine_name).first()
 
         owner = machine.owner
         vm_memory = machine.memory