1 files changed, 6 insertions, 4 deletions
diff --git a/tools/compileconfig.py b/tools/compileconfig.py
index 0ee3fab..b5e5053 100755
--- a/tools/compileconfig.py
+++ b/tools/compileconfig.py
@@ -5,7 +5,7 @@
 
 import argparse
 import sys
-import yaml
+import readconfig
 import re
 import base64
 
@@ -405,7 +405,6 @@ def gen_testmakefile(config, testmakefile, machines, shellvars=False):
 
     configfile.close()
 
-
 def main():
     parser = argparse.ArgumentParser(description="")
     parser.add_argument('--config', help="System configuration", required=True)
@@ -415,13 +414,16 @@ def main():
     parser.add_argument("--machines", type=int, metavar="n", help="Number of machines")
     args = parser.parse_args()
 
-    config = yaml.load(open(args.config))
     if args.testmakefile and args.machines:
+        config = readconfig.read_config(args.config)
         gen_testmakefile(config, args.testmakefile, args.machines)
     elif args.testshellvars and args.machines:
+        config = readconfig.read_config(args.config)
         gen_testmakefile(config, args.testshellvars, args.machines, shellvars=True)
     elif args.localconfig:
-        localconfig = yaml.load(open(args.localconfig))
+        localconfig = readconfig.read_config(args.localconfig)
+        config = readconfig.verify_and_read_config(args.config, localconfig["logadminkey"])
+
         localnodes = localconfig["localnodes"]
         for localnode in localnodes:
             gen_config(localnode, config, localconfig)