bcollet/netbox-exporter
1
0
Fork 0
Code Issues Pull requests Releases Packages Activity Actions

Add support for custom records

Browse source
This commit is contained in:
root 2022-11-11 23:31:43 +01:00
parent 00e8bc7ca3
commit 60a17916ac
1 changed files with 22 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
exporter.py
View file

@ -22,15 +22,20 @@ def ptr(nb, args):
device_cluster = {} device_cluster = {}
vm_primary = {} vm_primary = {}
records = {} records = {}
extra_records = []
serial = 0 serial = 0
af = ipaddress.ip_network(args.prefix).version af = ipaddress.ip_network(args.prefix).version
extras = nb.extras.config_contexts.get(name=args.prefix)
devices = nb.dcim.devices.all() devices = nb.dcim.devices.all()
vms = nb.virtualization.virtual_machines.filter(has_primary_ip=True) vms = nb.virtualization.virtual_machines.filter(has_primary_ip=True)
addresses = nb.ipam.ip_addresses.filter(parent=args.prefix) addresses = nb.ipam.ip_addresses.filter(parent=args.prefix)
if extras is not None:
extra_records = extras.data['records']
for device in devices: for device in devices:
last_updated = int(datetime.timestamp(datetime.strptime(device.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 last_updated > serial: serial = last_updated
if device.virtual_chassis: if device.virtual_chassis:
@ -43,7 +48,7 @@ def ptr(nb, args):
device_primary[device.id] = device.primary_ip6.id device_primary[device.id] = device.primary_ip6.id
for vm in vms: for vm in vms:
last_updated = int(datetime.timestamp(datetime.strptime(vm.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 last_updated > serial: serial = last_updated
if af == 4 and vm.primary_ip4: if af == 4 and vm.primary_ip4:
@ -53,7 +58,7 @@ def ptr(nb, args):
vm_primary[vm.id] = vm.primary_ip6.id vm_primary[vm.id] = vm.primary_ip6.id
for address in addresses: for address in addresses:
last_updated = int(datetime.timestamp(datetime.strptime(address.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 if last_updated > serial: serial = last_updated
ip = ipaddress.ip_interface(address.address).ip ip = ipaddress.ip_interface(address.address).ip
@ -70,20 +75,20 @@ def ptr(nb, args):
if address.assigned_object.device.id in device_primary and address.id == device_primary[address.assigned_object.device.id]: 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}] records[ptr] = [{"type":"PTR","rr":address.assigned_object.device.name}]
else: else:
iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name) 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))}] records[ptr] = [{"type":"PTR","rr":".".join((iname,address.assigned_object.device.name))}]
elif address.assigned_object_type == 'virtualization.vminterface': 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]: 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}] records[ptr] = [{"type":"PTR","rr":address.assigned_object.virtual_machine.name}]
else: else:
iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name) 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))}] records[ptr] = [{"type":"PTR","rr":".".join((iname,address.assigned_object.virtual_machine.name))}]
file_loader = FileSystemLoader(templates_dir) file_loader = FileSystemLoader(templates_dir)
env = Environment(loader=file_loader) env = Environment(loader=file_loader)
template = env.get_template(args.template) template = env.get_template(args.template)
output = template.render(serial=serial,records=records) output = template.render(serial=serial,records=records, extra_records=extra_records)
if not hasattr(args, 'output'): if not hasattr(args, 'output'):
print(output) print(output)
@ -101,18 +106,22 @@ def dns(nb, args):
records = {} records = {}
serial = 0 serial = 0
extras = nb.extras.config_contexts.get(name=args.domain)
devices = nb.dcim.devices.filter(name__iew=args.domain) devices = nb.dcim.devices.filter(name__iew=args.domain)
clusters = nb.dcim.virtual_chassis.filter(domain=args.domain) clusters = nb.dcim.virtual_chassis.filter(domain=args.domain)
cluster_devices = nb.dcim.devices.filter(virtual_chassis_member=True) cluster_devices = nb.dcim.devices.filter(virtual_chassis_member=True)
vms = nb.virtualization.virtual_machines.filter(name__iew=args.domain) vms = nb.virtualization.virtual_machines.filter(name__iew=args.domain)
addresses = nb.ipam.ip_addresses.all() addresses = nb.ipam.ip_addresses.all()
if extras is not None:
extra_records = extras.data['records']
for cluster in clusters: for cluster in clusters:
clusters_id.append(cluster.id) clusters_id.append(cluster.id)
for cluster_device in cluster_devices: for cluster_device in cluster_devices:
if cluster_device.virtual_chassis.id in clusters_id: 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.%fZ'))) 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 if last_updated > serial: serial = last_updated
devices_id.append(cluster_device.id) devices_id.append(cluster_device.id)
@ -127,7 +136,7 @@ def dns(nb, args):
for device in devices: for device in devices:
last_updated = int(datetime.timestamp(datetime.strptime(device.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 last_updated > serial: serial = last_updated
devices_id.append(device.id) devices_id.append(device.id)
@ -139,7 +148,7 @@ def dns(nb, args):
primary_ip[device.primary_ip6.id] = device.name primary_ip[device.primary_ip6.id] = device.name
for vm in vms: for vm in vms:
last_updated = int(datetime.timestamp(datetime.strptime(vm.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 last_updated > serial: serial = last_updated
vm_id.append(vm.id) vm_id.append(vm.id)
@ -151,7 +160,7 @@ def dns(nb, args):
primary_ip[vm.primary_ip6.id] = vm.name primary_ip[vm.primary_ip6.id] = vm.name
for address in addresses: for address in addresses:
last_updated = int(datetime.timestamp(datetime.strptime(address.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ'))) 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 if last_updated > serial: serial = last_updated
ip = ipaddress.ip_interface(address.address).ip ip = ipaddress.ip_interface(address.address).ip
@ -173,7 +182,7 @@ def dns(nb, args):
records[primary_ip[address.id]].append({"type":type,"rr":ip}) 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: elif address.assigned_object_type == 'dcim.interface' and address.assigned_object.device.id in devices_id:
iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name.lower()) iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower()
if address.assigned_object.device.id in vc_devices: if address.assigned_object.device.id in vc_devices:
fname = ".".join((iname,vc_devices[address.assigned_object.device.id])) fname = ".".join((iname,vc_devices[address.assigned_object.device.id]))
@ -185,7 +194,7 @@ def dns(nb, args):
records[fname].append({"type":type,"rr":ip}) records[fname].append({"type":type,"rr":ip})
elif address.assigned_object_type == 'virtualization.vminterface' and address.assigned_object.virtual_machine.id in vm_id: elif address.assigned_object_type == 'virtualization.vminterface' and address.assigned_object.virtual_machine.id in vm_id:
iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name.lower()) iname = re.sub(r'[^a-zA-Z0-9]', '-',address.assigned_object.name).lower()
fname = ".".join((iname,address.assigned_object.virtual_machine.name)) fname = ".".join((iname,address.assigned_object.virtual_machine.name))
if fname not in records: if fname not in records:
@ -195,7 +204,7 @@ def dns(nb, args):
file_loader = FileSystemLoader(templates_dir) file_loader = FileSystemLoader(templates_dir)
env = Environment(loader=file_loader) env = Environment(loader=file_loader)
template = env.get_template(args.template) template = env.get_template(args.template)
output = template.render(serial=serial,records=records) output = template.render(serial=serial,records=records,extra_records=extra_records)
if not hasattr(args, 'output'): if not hasattr(args, 'output'):
print(output) print(output)