msg = None

if hasattr(response, 'status_code'):

if response.status_code == 401:

msg = "\n[ERROR] An authorization error has occurred, please check your auth_token is correct."

elif response.status_code == 404:

# This is a non-fatal error

print "--> Not found, continuing"

return

elif response.status_code == 429:

msg = "\n[ERROR] You have exceeded DigitalOcean's Rate Limit of 1200 requests per hour. " \

"Please wait an hour before trying again and type the last domain synchronised when prompted."

if msg is None:

msg = "\n[ERROR] An unknown error has occurred, please re-run the script."

print >>sys.stderr, msg

if hasattr(response, 'status_code'):

print >>sys.stderr, "Response status code:", response.status_code

try:

content_msg = json.loads(response.content)['message']

print >>sys.stderr, "Response message:", content_msg

except AttributeError:

debug_msg = json.dumps(response, indent=4, sort_keys=True)

print >>sys.stderr, "Response body:", debug_msg

dnsName = str(dnsName)

if domain not in dnsName and dnsName != '@':

return dnsName + '.' + domain + '.'

else:

# Catches the @ symbol case too.

print "\nChecking DigitalOcean DNS for", domain

response = requests.get(domain_records_url, headers=headers)

if response.status_code == 200:

print "--> Domain records found"

elif response.status_code == 404:

print "--> Domain records not found, creating zone"

data = {"name": domain, "ip_address": ip}

response = requests.post(base_url, data=json.dumps(data), headers=headers).json()

if 'domain' in response and 'name' in response['domain'] and response['domain']['name'] == domain:

print "--> Successfully created zone for", domain

# Wipe the default DigitalOcean records

wipe_zone(domain_records_url)

else:

handle_error(response)

else:

# Synchronise zone

print "\nSynchronising DNS zone for", domain, "..."

# First get all the existing records

existing_records = requests.get(domain_records_url+"?per_page=9999", headers=headers).json().get('domain_records', [])

# Create an array to hold all the updated records

updated_records = []

# Create an array to hold synchronised record IDs

synced_record_ids = []

# Get the BIND raw DNS dump

bindfile = bindfolder + domain + bindextension

with open(bindfile, "r") as dns_file:

dns_dump = dns_file.read()

dns_dump = "$ORIGIN {0}.\n{1}".format(domain, dns_dump)

zone = dns.zone.from_text(dns_dump)

for name, node in zone.nodes.items():

name = str(name)

print "\nRecord name:", name

print "Qualified name:", qualifyName(name, domain)

rdatasets = node.rdatasets

for rset in rdatasets:

print "--> TTL:", str(rset.ttl)

print "--> Type:", rdatatype.to_text(rset.rdtype)

for rdata in rset:

data = None

priority = None

port = None

weight = None

if rset.rdtype == MX:

priority = rdata.preference

print "--> Priority:", priority

if unicode(rdata.exchange) == "@":

data = "%s." % (domain)

else:

data = "%s.%s." % (rdata.exchange, domain)

elif rset.rdtype == CNAME:

if unicode(rdata) == "@":

data = "@"

else:

data = rdata.target

elif rset.rdtype == A:

data = rdata.address

elif rset.rdtype == AAAA:

data = rdata.address.lower()

elif rset.rdtype == NS:

data = rdata.target

elif rset.rdtype == SRV:

priority = rdata.priority

weight = rdata.weight

port = rdata.port

data = rdata.target

elif rset.rdtype == TXT:

data = " ".join('"{0}"'.format(string) for string in rdata.strings)

if data:

print "--> Data:", data

data = unicode(data)

type = rdatatype.to_text(rset.rdtype)

# Try and find an existing record

record_id = None

for record in existing_records:

if type in ["CNAME", "MX", "NS", "SRV"] and data[-1:] == ".":

check_data = data[:-1]

else:

check_data = data

if record['name'] == name and record['type'] == type and record['data'] == check_data:

record_id = record['id']

synced_record_ids.append(record_id)

break

if record_id:

print "--> Already exists, skipping"

else:

if type in ["CNAME", "MX", "NS", "SRV"] and data != "@" and data[-1:] != ".":

data = "{0}.{1}.".format(data, domain)

post_data = {

"type": type,

"name": name,

"data": data,

"priority": priority,

"port": port,

"weight": weight

}

# Collect records to be updated into the updated_records array

print "--> Queuing to update"

updated_records.append(post_data)

# Delete any records that exist with DigitalOcean that have been removed

print "\nRemoving deleted records"

for record in existing_records:

if record['id'] not in synced_record_ids and record['type'] != 'SOA':

response = requests.delete("{0}/{1}".format(domain_records_url, record["id"]), headers=headers)

if response.status_code == 204:

print "--> Deleted record", record["name"], "IN", record["type"], record["data"]

else:

handle_error(response)

print "--> Done"

# Finally, post the responses for the updated records

print "\nPosting updated records"

for record in updated_records:

response = requests.post(domain_records_url, data=json.dumps(record), headers=headers).json()

if 'domain_record' in response:

print "--> Updated record", record["name"], "IN", record["type"], record["data"]

else:

handle_error(response)

print "--> Done"

# Wipe all the existing records for a given domain

print "\nWiping default DigitalOcean records..."

response = requests.get(domain_records_url, headers=headers).json()

# print "Response body:", json.dumps(response, indent=4, sort_keys=True)

if 'domain_records' in response:

for record in response['domain_records']:

response = requests.delete("{0}/{1}".format(domain_records_url, record["id"]), headers=headers)

if response.status_code == 204:

print "--> Deleted record", record["type"], record["data"]

else:

handle_error(response)

print "--> Done"

else: