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 # allow SSH password auth
  11 cfg['ssh_pwauth'] = "1"
  12
  13 try:
  14   if cfg['system_info']['default_user']['name']:
  15     synver = "2"
  16 except KeyError:
  17   synver = "1"
  18
  19 if synver == "1":
  20   if cfg['user'] == user:
  21     print("No change needed")
  22     exit()
  23   else:
  24     cfg['user'] = user
  25 elif synver == "2":
  26   if cfg['system_info']['default_user']['name'] == user:
  27    print("No change needed")
  28    exit()
  29   else:
  30    # Change the user to cloud-user
  31    cfg['system_info']['default_user']['name'] = user
  32    cfg['system_info']['default_user']['gecos'] = "Cloud User"
  33    print cfg['system_info']['default_user']['name']
  34
  35 with open(cloudcfg, "w") as f:
  36   yaml.dump(cfg, f, default_flow_style=False)