X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/730dd3b604099ac2cfe8ed41a476a845b342ecb6..966553f52ce277810245551803109ba11014d72e:/code/controls.py

diff --git a/code/controls.py b/code/controls.py
index 841d21b..79aa311 100644
--- a/code/controls.py
+++ b/code/controls.py
@@ -11,7 +11,7 @@ import sys
 import time
 import re
 import cache_acls
-import cPickle
+import yaml
 
 # ... and stolen from xend/uuid.py
 def randomUUID():
@@ -139,8 +139,8 @@ def createVm(owner, contact, name, memory, disk_size, machine_type, cdrom, clone
 
 def getList(machines):
     """Return a dictionary mapping machine  to dicts."""
-    value_string = remctl('web', 'listvms', '--pickle')
-    value_dict = cPickle.loads(value_string)
+    value_string = remctl('web', 'listvms')
+    value_dict = yaml.safe_load(value_string)
 
     d = dict((m, value_dict[m.name]) for m in machines if m.name in value_dict)
     return d