zoneupdate/zoneupdate.py

73 lines
2.9 KiB
Python
Raw Normal View History

2020-06-02 00:26:21 +00:00
#!/bin/python3
import requests
import json
import toml
import re
config=toml.load('zoneupdate.toml')
def get_current_public_ip(ip_services, debug=False):
'''
resolve public ip using simple service, from github.com/fyah/gandi-api-v5/gandyn-livedns.py
'''
for service in ip_services:
try:
r = requests.get(service)
res = r.text
if re.match('\d+\.\d+\.\d+\.\d+', res):
return res
except Exception as e:
#there was a problem resolving our public with the service...
if debug:
print ('Failed to resolve public ip using %s' % service)
print (e)
#try with next
continue
def get_sub_records(domain, subdomain, api_url, api_key, debug=False):
url = api_url + '/domains/' + domain + '/records/' + subdomain
headers = {'Authorization': "Apikey " + api_key}
resp = requests.get(url, headers=headers)
json_object = json.loads(resp._content)
return json_object[0]
def update_sub_records(domain, subdomain, api_url, api_key, ip, ttl, debug=False):
url = api_url + '/domains/' + domain + '/records/' + subdomain
payload = {"rrset_type": "A", "rrset_ttl": ttl, "rrset_values": [ip]}
items = {"items": [payload]}
headers = {'Content-Type': 'application/json', 'Authorization': "Apikey " + api_key}
resp = requests.put(url, data=json.dumps(items), headers=headers)
json_object = json.loads(resp._content)
return json_object
for domain in config['domains']:
for subdomain in config[domain]['subdomains']:
url = subdomain + '.' + domain
api_url=config['api_url']
api_key=config[domain]['api_key']
debug=config['debug']
ttl=config['ttl']
print ("Checking IP of %s" % url)
try:
sub_records = get_sub_records(domain,subdomain,api_url=config['api_url'],api_key=config[domain]['api_key'],debug=config['debug'])
old_ip=(sub_records['rrset_values'])
old_ip=old_ip[0]
new_ip=get_current_public_ip(ip_services=config['ip_services'], debug=config['debug'])
except Exception as e:
print ('Failed to find sub-record for domain %s, attempting to create' % url)
new_ip=get_current_public_ip(ip_services=config['ip_services'], debug=config['debug'])
resp = update_sub_records(domain,subdomain,api_url=config['api_url'],api_key=config[domain]['api_key'],ip=new_ip,ttl=config['ttl'],debug=config['debug'])
print (resp)
continue
if new_ip != old_ip:
print ('IP check failed, updating')
resp = update_sub_records(domain,subdomain,api_url=config['api_url'],api_key=config[domain]['api_key'],ip=new_ip,ttl=config['ttl'],debug=config['debug'])
print (resp)
else:
print ('IP check complete')