X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/41910267b9c749ae6d6f49fd2569600689dd9914..24bbf8e59bf6b9b0066a5db02cfbd2bb48f83a1f:/code/controls.py

diff --git a/code/controls.py b/code/controls.py
index 013e04c..86c23c2 100644
--- a/code/controls.py
+++ b/code/controls.py
@@ -226,7 +226,7 @@ def commandResult(username, state, fields):
     machine = validation.Validate(username, state, machine_id=fields.getfirst('machine_id')).machine
     action = fields.getfirst('action')
     cdrom = fields.getfirst('cdrom')
-    if cdrom is not None and not CDROM.get(cdrom):
+    if cdrom is not None and not CDROM.query().filter_by(cdrom_id=cdrom).one():
         raise CodeError("Invalid cdrom type '%s'" % cdrom)    
     if action not in ('Reboot', 'Power on', 'Power off', 'Shutdown', 
                       'Delete VM'):