aboutsummaryrefslogtreecommitdiffstats
path: root/nfvbenchvm/dib/elements/nfvbenchvm/post-install.d/51-cloudcfg-edit
blob: dc5103045eb6be67ba647d33c567e42aa8c571c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env python

import yaml
cloudcfg = "/etc/cloud/cloud.cfg"
user = "cloud-user"

with open(cloudcfg) as f:
  cfg = yaml.safe_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)