X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/4ce0edc046482af00c58171cb63064da34561e79..17757a4f472d40dcbe3f9a4d071f3283cc17d745:/code/main.py diff --git a/code/main.py b/code/main.py index 4627c66..03d7eb1 100755 --- a/code/main.py +++ b/code/main.py @@ -257,8 +257,12 @@ def vnc(username, state, path, fields): token = {'data': pickled_data, 'digest': m.digest()} token = cPickle.dumps(token) token = base64.urlsafe_b64encode(token) - port = 10003 + [config_host["hostname"] for config_host in invirt_config["hosts"] - ].index(controls.listHost(machine)) + host = controls.listHost(machine) + if host: + port = 10003 + [config_host["hostname"] for config_host in invirt_config["hosts"] + ].index(host) + else: + port = 5900 # dummy status = controls.statusInfo(machine) has_vnc = hasVnc(status)