move delete_record to a more import friendly structure. clean up code

This commit is contained in:
Jeffrey Forman 2012-11-23 22:28:33 -05:00
parent 37d6c91a95
commit c42009f714
1 changed files with 9 additions and 9 deletions

View File

@ -107,6 +107,7 @@ def add_cname_record(dns_server, zone_name, originating_record, cname, ttl, key_
"""Add a Cname record.""" """Add a Cname record."""
if key_name == "None": if key_name == "None":
# TODO: Does this need to be changed to "key_name is None"
keyring = None keyring = None
else: else:
this_key = models.Key.objects.get(name=key_name) this_key = models.Key.objects.get(name=key_name)
@ -120,25 +121,24 @@ def add_cname_record(dns_server, zone_name, originating_record, cname, ttl, key_
"output" : response}] "output" : response}]
def delete_record(form_data, rr_items): def delete_record(dns_server, rr_list, key_name):
"""Delete a list of DNS records passed as strings in rr_items.""" """Delete a list of DNS records passed as strings in rr_items."""
if form_data["key_name"]: if key_name is None:
this_key = models.Key.objects.get(name=form_data["key_name"])
keyring = keyutils.create_keyring(this_key.name, this_key.data)
else:
keyring = None keyring = None
else:
this_key = models.Key.objects.get(name=key_name)
keyring = keyutils.create_keyring(this_key.name, this_key.data)
dns_server = form_data["dns_server"]
delete_response = [] delete_response = []
for current_rr_item in rr_items: for current_rr in rr_list:
re_record = re.search(r"(\w+)\.(.*)$", current_rr_item) re_record = re.search(r"(\w+)\.(.*)$", current_rr)
record = re_record.group(1) record = re_record.group(1)
domain = re_record.group(2) domain = re_record.group(2)
dns_update = dns.update.Update(domain, keyring = keyring) dns_update = dns.update.Update(domain, keyring = keyring)
dns_update.delete(record) dns_update.delete(record)
output = dns.query.tcp(dns_update, dns_server) output = dns.query.tcp(dns_update, dns_server)
delete_response.append({ "description" : "Delete record %s" % current_rr_item, delete_response.append({ "description" : "Delete record %s" % current_rr,
"output" : output }) "output" : output })
return delete_response return delete_response