tools: config-create: Add support for using a data directory

[jailhouse.git] / tools / jailhouse-config-create

diff --git a/tools/jailhouse-config-create b/tools/jailhouse-config-create
index dce9f6d44a302e791b753befe610d47fca6fee05..ab2ea6d134b039ab5f0f6e42c031ff5d07d47422 100755 (executable)
--- a/tools/jailhouse-config-create
+++ b/tools/jailhouse-config-create
@@ -20,7 +20,12 @@ import argparse
 import struct
 from mako.template import Template
 
-abspath = os.path.abspath(os.path.dirname(sys.argv[0]))
+datadir = None
+
+if datadir:
+    template_default_dir = datadir + "/jailhouse"
+else:
+    template_default_dir = os.path.abspath(os.path.dirname(sys.argv[0]))
 
 # pretend to be part of the jailhouse tool
 sys.argv[0] = sys.argv[0].replace('-', ' ')
@@ -38,8 +43,8 @@ parser.add_argument('-r', '--root',
                     type=str)
 parser.add_argument('-t', '--template-dir',
                     help='the directory where the templates are located,'
-                         'the default is "' + abspath + '"',
-                    default=abspath,
+                         'the default is "' + template_default_dir + '"',
+                    default=template_default_dir,
                     action='store',
                     type=str)