def clean_dns(cls, instances, **kwargs):
"""
Clean DNS entries for each instance if DNSaaS integration is enabled.
"""
if not settings.ENABLE_DNSAAS_INTEGRATION:
raise DNSaaSIntegrationNotEnabledError()
dnsaas = DNSaaS()
# TODO: transaction?
for instance in instances:
ips = list(
instance.ipaddresses.exclude(is_management=True).values_list(
'address', flat=True))
if not ips:
logger.info('No IPs for %s - skipping cleaning DNS entries',
instance)
continue
records = dnsaas.get_dns_records(ips)
if len(records) > settings.DEPLOYMENT_MAX_DNS_ENTRIES_TO_CLEAN:
raise Exception(
'Cannot clean {} entries for {} - clean it manually'.format(
len(records), instance))
for record in records:
logger.warning('Deleting %s (%s / %s / %s) DNS record',
record['pk'], record['type'], record['name'],
record['content'])
if dnsaas.delete_dns_record(record['pk']):
raise Exception() # TODO
class DNSView(RalphDetailView):
icon = 'chain-broken'
name = 'dns_edit'
label = 'DNS'
url_name = 'dns_edit'
template_name = 'dns/dns_edit.html'
def __init__(self, *args, **kwargs):
if not settings.ENABLE_DNSAAS_INTEGRATION:
raise DNSaaSIntegrationNotEnabledError()
self.dnsaas = DNSaaS()
return super().__init__(*args, **kwargs)
def get_forms(self):
forms = []
ipaddresses = self.object.ipaddress_set.all().values_list('address',
flat=True)
if not ipaddresses:
# If ipaddresses is empty return empty form list because we can not
# identify the records do not have any IP address
return forms
initial = self.dnsaas.get_dns_records(ipaddresses)
for item in initial:
forms.append(DNSRecordForm(item))
if initial and initial[0]['type'] == RecordType.a.id:
# from API "A" record is always first
empty_form = DNSRecordForm(initial={'name': initial[0]['name']})
else:
empty_form = DNSRecordForm()
forms.append(empty_form)
return forms
def get(self, request, *args, **kwargs):
if 'forms' not in kwargs:
kwargs['forms'] = self.get_forms()
return super().get(request, *kwargs, **kwargs)
def post(self, request, *args, **kwargs):
forms = self.get_forms()
posted_form = DNSRecordForm(request.POST)
# Find form which request's data belongs to
for i, form in enumerate(forms):
if (str(form.data.get('pk',
'')) == str(posted_form.data.get('pk', ''))):
forms[i] = posted_form
break
if posted_form.is_valid():
if posted_form.data.get('delete'):
errors = self.dnsaas.delete_dns_record(form.data['pk'])
elif posted_form.cleaned_data.get('pk'):
errors = self.dnsaas.update_dns_record(
posted_form.cleaned_data)
else:
errors = self.dnsaas.create_dns_record(
posted_form.cleaned_data)
if errors:
for field_name, field_errors in errors.items():
for field_error in field_errors:
if field_name == 'non_field_errors':
field_name = None
posted_form.add_error(field_name, field_error)
else:
return HttpResponseRedirect('.')
kwargs['forms'] = forms
return self.get(request, *args, **kwargs)