create-vm fix

1 files changed, 10 insertions, 1 deletions

diff --git a/create-vm b/create-vm
index 454db06..e085598 100755
--- a/create-vm
+++ b/create-vm
@@ -7,6 +7,7 @@ import subprocess
 import sys
 import tempfile
 import time
+import yaml
 
 
 def cloud_init_iso(iso, hostname, pubkey):
@@ -83,7 +84,15 @@ def wait_for_ssh(hostname):
 
 
 def main():
-    base, vm, memory, cpus, img, size, pubkey = sys.argv[1:]
+    config = yaml.safe_load(open(sys.argv[1]))
+
+    base = config["base_image"]
+    vm = config["name"]
+    img = config["image_file"]
+    size = config["image_size"]
+    pubkey = config["public_key"]
+    memory = config.get("memory", 1024)
+    cpus = config.get("cpus", 1)
 
     memory = int(memory)
     cpus = int(cpus)