Sven Luthi
c32161b8a2
modified: netbox_dns_bridge/dns_server.py modified: netbox_dns_bridge/management/commands/dns-export-zone.py modified: netbox_dns_bridge/management/commands/dns-export-zones.py modified: netbox_dns_bridge/management/commands/dns-settings.py modified: netbox_dns_bridge/management/commands/dns-transfer-endpoint.py modified: netbox_dns_bridge/migrations/0001_initial.py modified: netbox_dns_bridge/models.py modified: netbox_dns_bridge/request_handler.py modified: netbox_dns_bridge/signals.py modified: netbox_dns_bridge/utils.py Black code formating corrected
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
import os
|
|
import dns.zone
|
|
import dns.rdatatype
|
|
import dns.rdataclass
|
|
import dns.exception
|
|
import netbox_dns.models
|
|
from netbox_dns_bridge.utils import export_bind_zone_file
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--view",
|
|
type=str,
|
|
help="The name of the view the Zone to be exported is in",
|
|
)
|
|
parser.add_argument(
|
|
"--zone", type=str, help="The FQDN of the Zone to be exported"
|
|
)
|
|
parser.add_argument(
|
|
"--file", type=str, help="Path of the zone file to be written"
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
# if len(options) < 2:
|
|
# print("export-zone <zone name> <file path>")
|
|
# sys.exit(1)
|
|
|
|
# Load parameters
|
|
view_name = options["view"]
|
|
zone_name = options["zone"]
|
|
file_path = options["file"]
|
|
|
|
if not view_name:
|
|
raise CommandError("No --view parameter given")
|
|
elif not zone_name:
|
|
raise CommandError("No --zone parameter given")
|
|
elif not file_path:
|
|
raise CommandError("No --file parameter given")
|
|
|
|
try:
|
|
# Load the zone from NetBox DNS
|
|
nb_zone = netbox_dns.models.Zone.objects.get(
|
|
view__name=view_name, name=zone_name
|
|
)
|
|
|
|
if nb_zone:
|
|
export_bind_zone_file(nb_zone, file_path=file_path)
|
|
else:
|
|
print("Zone not found in Netbox. Aborting")
|
|
sys.exit(1)
|
|
|
|
except Exception as e:
|
|
raise CommandError(f"Failed to export zone: {e}")
|
|
|
|
self.stdout.write(
|
|
self.style.SUCCESS(f"Zone '{zone_name}' exported to '{file_path}'")
|
|
)
|