From 7d9f05445a2338795aca6b6807c4fa001a2a9586 Mon Sep 17 00:00:00 2001
From: Benjamin Collet <benjamin@collet.eu>
Date: Fri, 9 Apr 2021 17:34:06 +0200
Subject: [PATCH] Bugfixes + cluster support

---
 exporter.py | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/exporter.py b/exporter.py
index 18f7e22..33e3e16 100755
--- a/exporter.py
+++ b/exporter.py
@@ -19,12 +19,13 @@ templates_dir = os.path.dirname(os.path.realpath(__file__))
 
 def ptr(nb, args):
     device_primary = {}
+    device_cluster = {}
     vm_primary = {}
     records = {}
     serial = 0
     af = ipaddress.ip_network(args.prefix).version
 
-    devices = nb.dcim.devices.filter(has_primary_ip=True)
+    devices = nb.dcim.devices.all()
     vms = nb.virtualization.virtual_machines.filter(has_primary_ip=True)
     addresses = nb.ipam.ip_addresses.filter(parent=args.prefix)
 
@@ -32,6 +33,9 @@ def ptr(nb, args):
         last_updated = int(datetime.timestamp(datetime.strptime(device.last_updated, '%Y-%m-%dT%H:%M:%S.%fZ')))
         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
 
@@ -59,14 +63,17 @@ def ptr(nb, args):
             records[ptr] = [{"type":"PTR","rr":address.dns_name}]
 
         elif address.assigned_object_type == 'dcim.interface':
-            if address.id == device_primary[address.assigned_object.device.id]:
+            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.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-z0-9]', '-',address.assigned_object.name)
                 records[ptr] = [{"type":"PTR","rr":".".join((iname,address.assigned_object.device.name))}]
 
         elif address.assigned_object_type == 'virtualization.vminterface':
-            if 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}]
             else:
                 iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name)
@@ -124,7 +131,10 @@ def dns(nb, args):
             type = "AAAA"
 
         if address.dns_name and address.dns_name.endswith(args.domain):
-            records[address.dns_name] = [{"type":type,"rr":ip}]
+            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:
@@ -133,7 +143,7 @@ def dns(nb, args):
             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-z0-9]', '-',address.assigned_object.name)
+            iname = re.sub(r'[^a-z0-9]', '-',address.assigned_object.name.lower())
             fname = ".".join((iname,address.assigned_object.device.name))
 
             if fname not in records:
@@ -141,16 +151,13 @@ def dns(nb, args):
             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-z0-9]', '-',address.assigned_object.name)
+            iname = re.sub(r'[^a-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})
 
-        else:
-            sys.exit(1)
-
     file_loader = FileSystemLoader(templates_dir)
     env = Environment(loader=file_loader)
     template = env.get_template("zonefile.jj2")