Add the scipts to build NFVbench VM

Change-Id: I9632333777260468d6f07cf0149c789626f20145
Signed-off-by: Yichen Wang <yicwang@cisco.com>

1 files changed, 33 insertions, 0 deletions

diff --git a/nfvbenchvm/dib/elements/nfvbenchvm/post-install.d/51-cloudcfg-edit b/nfvbenchvm/dib/elements/nfvbenchvm/post-install.d/51-cloudcfg-edit
new file mode 100755
index 0000000..3e4647a
--- /dev/null
+++ b/nfvbenchvm/dib/elements/nfvbenchvm/post-install.d/51-cloudcfg-edit
@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+
+import yaml
+cloudcfg = "/etc/cloud/cloud.cfg"
+user = "cloud-user"
+
+with open(cloudcfg) as f:
+  cfg = yaml.load(f)
+
+try:
+  if cfg['system_info']['default_user']['name']:
+    synver = "2"
+except KeyError:
+  synver = "1"
+
+if synver == "1":
+  if cfg['user'] == user:
+    print("No change needed")
+    exit()
+  else:
+    cfg['user'] = user
+elif synver == "2":
+  if cfg['system_info']['default_user']['name'] == user:
+   print("No change needed")
+   exit()
+  else:
+   # Change the user to cloud-user
+   cfg['system_info']['default_user']['name'] = user
+   cfg['system_info']['default_user']['gecos'] = "Cloud User"
+   print cfg['system_info']['default_user']['name']
+
+with open(cloudcfg, "w") as f:
+  yaml.dump(cfg, f, default_flow_style=False)