#!/usr/bin/env python3 import argparse import ipaddress import os import pynetbox import sys import yaml import re from datetime import datetime from jinja2 import Environment, FileSystemLoader config = os.path.join(os.path.dirname(os.path.realpath(__file__)),'config.yml') with open(config, 'r') as ymlfile: cfg = yaml.load(ymlfile, Loader=yaml.FullLoader) templates_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)),'templates') def ptr(nb, args): device_primary = {} device_cluster = {} vm_primary = {} records = {} extra_records = [] serial = 0 af = ipaddress.ip_network(args.prefix).version extras = nb.extras.config_contexts.get(name=args.prefix) devices = nb.dcim.devices.all() vms = nb.virtualization.virtual_machines.filter(has_primary_ip=True) addresses = nb.ipam.ip_addresses.filter(parent=args.prefix) if extras is not None: extra_records = extras.data['records'] for device in devices: last_updated = int(datetime.timestamp(datetime.strptime(device.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated if device.virtual_chassis: device_cluster[device.id] = device.virtual_chassis.name if af == 4 and device.primary_ip4: device_primary[device.id] = device.primary_ip4.id elif af == 6 and device.primary_ip6: device_primary[device.id] = device.primary_ip6.id for vm in vms: last_updated = int(datetime.timestamp(datetime.strptime(vm.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated if af == 4 and vm.primary_ip4: vm_primary[vm.id] = vm.primary_ip4.id elif af == 6 and vm.primary_ip6: vm_primary[vm.id] = vm.primary_ip6.id for address in addresses: last_updated = int(datetime.timestamp(datetime.strptime(address.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated ip = ipaddress.ip_interface(address.address).ip ptr = ipaddress.ip_address(ip).reverse_pointer if address.dns_name: records[ptr] = [{"type":"PTR","rr":address.dns_name}] elif address.assigned_object_type == 'dcim.interface': if address.assigned_object.device.id in device_cluster: address.assigned_object.device.name = device_cluster[address.assigned_object.device.id] if address.assigned_object.device.name is not None: if address.assigned_object.device.id in device_primary and address.id == device_primary[address.assigned_object.device.id]: records[ptr] = [{"type":"PTR","rr":address.assigned_object.device.name}] else: iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower() records[ptr] = [{"type":"PTR","rr":".".join((iname,address.assigned_object.device.name))}] elif address.assigned_object_type == 'virtualization.vminterface': if address.assigned_object.virtual_machine.id in vm_primary and address.id == vm_primary[address.assigned_object.virtual_machine.id]: records[ptr] = [{"type":"PTR","rr":address.assigned_object.virtual_machine.name}] else: iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower() records[ptr] = [{"type":"PTR","rr":".".join((iname,address.assigned_object.virtual_machine.name))}] file_loader = FileSystemLoader(templates_dir) env = Environment(loader=file_loader) template = env.get_template(args.template) output = template.render(serial=serial,records=records, extra_records=extra_records) if not hasattr(args, 'output'): print(output) else: f = open(args.output,'w') f.write(output) f.close() def dns(nb, args): devices_id = [] clusters_id = [] vm_id = [] primary_ip = {} vc_devices = {} records = {} extra_records = [] serial = 0 extras = nb.extras.config_contexts.get(name=args.domain) devices = nb.dcim.devices.filter(name__iew=args.domain) clusters = nb.dcim.virtual_chassis.filter(domain=args.domain) cluster_devices = nb.dcim.devices.filter(virtual_chassis_member=True) vms = nb.virtualization.virtual_machines.filter(name__iew=args.domain) addresses = nb.ipam.ip_addresses.all() if extras is not None: extra_records = extras.data['records'] for cluster in clusters: clusters_id.append(cluster.id) for device in devices: last_updated = int(datetime.timestamp(datetime.strptime(device.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated devices_id.append(device.id) if device.primary_ip4: primary_ip[device.primary_ip4.id] = device.name if device.primary_ip6: primary_ip[device.primary_ip6.id] = device.name for cluster_device in cluster_devices: if cluster_device.virtual_chassis.id in clusters_id: last_updated = int(datetime.timestamp(datetime.strptime(cluster_device.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated devices_id.append(cluster_device.id) vc_devices[cluster_device.id] = cluster_device.virtual_chassis.name if cluster_device.primary_ip4: primary_ip[cluster_device.primary_ip4.id] = cluster_device.virtual_chassis.name if cluster_device.primary_ip6: primary_ip[cluster_device.primary_ip6.id] = cluster_device.virtual_chassis.name for vm in vms: last_updated = int(datetime.timestamp(datetime.strptime(vm.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated vm_id.append(vm.id) if vm.primary_ip4: primary_ip[vm.primary_ip4.id] = vm.name if vm.primary_ip6: primary_ip[vm.primary_ip6.id] = vm.name for address in addresses: last_updated = int(datetime.timestamp(datetime.strptime(address.last_updated, '%Y-%m-%dT%H:%M:%S.%f%z'))) if last_updated > serial: serial = last_updated ip = ipaddress.ip_interface(address.address).ip if ipaddress.ip_address(ip).version == 4: type = "A" else: type = "AAAA" if address.dns_name and address.dns_name.endswith(args.domain): if address.dns_name not in records: records[address.dns_name] = [] records[address.dns_name].append({"type":type,"rr":ip}) elif address.id in primary_ip: if primary_ip[address.id] not in records: records[primary_ip[address.id]] = [] records[primary_ip[address.id]].append({"type":type,"rr":ip}) elif address.assigned_object_type == 'dcim.interface' and address.assigned_object.device.id in devices_id: iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower() if address.assigned_object.device.id in vc_devices: fname = ".".join((iname,vc_devices[address.assigned_object.device.id])) else: fname = ".".join((iname,address.assigned_object.device.name)) if fname not in records: records[fname] = [] records[fname].append({"type":type,"rr":ip}) elif address.assigned_object_type == 'virtualization.vminterface' and address.assigned_object.virtual_machine.id in vm_id: iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower() fname = ".".join((iname,address.assigned_object.virtual_machine.name)) if fname not in records: records[fname] = [] records[fname].append({"type":type,"rr":ip}) file_loader = FileSystemLoader(templates_dir) env = Environment(loader=file_loader) template = env.get_template(args.template) output = template.render(serial=serial,records=records,extra_records=extra_records) if not hasattr(args, 'output'): print(output) else: f = open(args.output,'w') f.write(output) f.close() def autodns(nb, args): if not 'zones' in cfg: return for zone in cfg['zones']: if zone['type'] == 'reverse': ptr(nb, argparse.Namespace(prefix=zone['name'],template=zone['template'],output=zone['file'])) elif zone['type'] == 'forward': dns(nb, argparse.Namespace(domain=zone['name'],template=zone['template'],output=zone['file'])) if __name__ == '__main__': nb = pynetbox.api( cfg['netbox']['url'], token=cfg['netbox']['token'] ) parser = argparse.ArgumentParser(description='Netbox API exporter') subparsers = parser.add_subparsers(help='Action to perform',dest='action',required=True) subparser = subparsers.add_parser('autodns', help='Generate all DNS zone files using configuration') subparser = subparsers.add_parser('ptr', help='Generate reverse DNS zone file for prefix') subparser.add_argument('prefix', type=str, help='Prefix') subparser.add_argument('template', type=str, help='template') subparser = subparsers.add_parser('dns', help='Generate DNS zone file for domain') subparser.add_argument('domain', type=str, help='Domain') subparser.add_argument('template', type=str, help='template') args = parser.parse_args() globals()[args.action](nb, args)