setup_dns_zone 682 B

12345678910111213141516171819202122232425262728
  1. #!/usr/bin/python
  2. import json
  3. import urllib2
  4. import sys
  5. from jinja2 import Template
  6. URL = 'http://169.254.169.254/latest/user-data'
  7. try:
  8. data = json.load(urllib2.urlopen('http://169.254.169.254/latest/user-data'))
  9. ip = urllib2.urlopen('http://169.254.169.254/latest/user-data').rstrip()
  10. except:
  11. print >> sys.stderr, 'Was not able to connect to the Amazon API'
  12. sys.exit(2)
  13. zone_template="""
  14. zone "{{user_data['domain']}}" IN {
  15. type forward;
  16. forward only;
  17. forwarders { {{ipaddr}} port 8600; };
  18. };
  19. """
  20. zt = Template(zone_template)
  21. main_config = open("/etc/bind/zones.consul", "w")
  22. main_config.write(zt.render(user_data=data, ipaddr=ip))
  23. main_config.close()