roles/ntp-utils/library/tune_master_server_params.py

   1 #!/bin/env python
   2 # pylint: skip-file
   3 from ansible.module_utils.basic import *
   4 import os, json
   5 import re, sys
   6 import syslog
   7 import re
   8
   9 if __name__ == '__main__':
  10     fields = {
  11             "server_extra_params": {"required":True, "type": str},
  12     }
  13
  14     module = AnsibleModule(argument_spec=fields)
  15
  16     server_extra_params = module.params['server_extra_params']
  17
  18     ntp_file = "/etc/ntp.conf"
  19     tmp_file = "/etc/ntp.conf.mod"
  20
  21     try:
  22         filter = re.compile('^server.*')
  23         lines = []
  24         with open(ntp_file, 'r') as f:
  25             lines = f.readlines()
  26
  27         updated = []
  28         if lines:
  29             for line in lines:
  30                 if filter.match(line) and line.find(server_extra_params) == -1:
  31                     tmp = line.split('\n')[0]
  32                     tmp = tmp + " " + server_extra_params + '\n'
  33                     updated.append(tmp)
  34                 else:
  35                     updated.append(line)
  36
  37             with open(tmp_file, 'w') as f:
  38                 for line in updated:
  39                     f.write(line)
  40
  41             os.rename(tmp_file, ntp_file)
  42
  43             module.exit_json(changed=True)
  44
  45     except Exception as exp:
  46         module.fail_json(msg=str(ex))