nfvbenchvm/dib/elements/nfvbenchvm/post-install.d/51-cloudcfg-edit

   1 #!/usr/bin/env python
   2
   3 import yaml
   4 cloudcfg = "/etc/cloud/cloud.cfg"
   5 user = "cloud-user"
   6
   7 with open(cloudcfg) as f:
   8   cfg = yaml.safe_load(f)
   9
  10 try:
  11   if cfg['system_info']['default_user']['name']:
  12     synver = "2"
  13 except KeyError:
  14   synver = "1"
  15
  16 if synver == "1":
  17   if cfg['user'] == user:
  18     print("No change needed")
  19     exit()
  20   else:
  21     cfg['user'] = user
  22 elif synver == "2":
  23   if cfg['system_info']['default_user']['name'] == user:
  24    print("No change needed")
  25    exit()
  26   else:
  27    # Change the user to cloud-user
  28    cfg['system_info']['default_user']['name'] = user
  29    cfg['system_info']['default_user']['gecos'] = "Cloud User"
  30    print cfg['system_info']['default_user']['name']
  31
  32 with open(cloudcfg, "w") as f:
  33   yaml.dump(cfg, f, default_flow_style=False)