| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- from django.core.management import BaseCommand, CommandError
- from django.db import transaction
- from desecapi import pdns
- from desecapi.models import Domain, RRset, RR
- class Command(BaseCommand):
- help = 'Import authoritative data from pdns, making the local database consistent with pdns.'
- def add_arguments(self, parser):
- parser.add_argument('domain-name', nargs='*',
- help='Domain name to import. If omitted, will import all domains that are known locally.')
- def handle(self, *args, **options):
- domains = Domain.objects.all()
- if options['domain-name']:
- domains = domains.filter(name__in=options['domain-name'])
- domain_names = domains.values_list('name', flat=True)
- for domain_name in options['domain-name']:
- if domain_name not in domain_names:
- raise CommandError('{} is not a known domain'.format(domain_name))
- for domain in domains:
- self.stdout.write('%s ...' % domain.name, ending='')
- try:
- self._sync_domain(domain)
- self.stdout.write(' synced')
- except Exception as e:
- if str(e).startswith('Could not find domain ') \
- and domain.owner.locked:
- self.stdout.write(' skipped')
- else:
- self.stdout.write(' failed')
- msg = 'Error while processing {}: {}'.format(domain.name, e)
- raise CommandError(msg)
- @staticmethod
- @transaction.atomic
- def _sync_domain(domain):
- domain.rrset_set.all().delete()
- rrsets = []
- rrs = []
- for rrset_data in pdns.get_rrset_datas(domain):
- if rrset_data['type'] in RRset.RESTRICTED_TYPES:
- continue
- records = rrset_data.pop('records')
- rrset = RRset(**rrset_data)
- rrsets.append(rrset)
- rrs.extend([RR(rrset=rrset, content=record) for record in records])
- RRset.objects.bulk_create(rrsets)
- RR.objects.bulk_create(rrs)
|
