netbox2aquilon: Add support for multiple VM disks

As added in NetBox 3.7.5.

Author: jrha

aquilon/netbox2aquilon.py

@@ -153,14 +153,33 @@ def _netbox_copy_vm(self, virtual_machine):
             '--memory', f'{virtual_machine.memory}',
         ])
 
-        cmds.append([
-            'add_disk',
-            '--machine', f'{virtual_machine.aq_machine_name}',
-            '--disk', 'sda',
-            '--controller', 'sata',
-            '--size', f'{virtual_machine.disk}',
-            '--boot',
-        ])
+        # Check if VM has any new-style virtual disks defined,
+        # If so, use them and set the first as bootable,
+        # If not, fall back to the classic single bootable disk method.
+        virtual_disks = self.get_disks_from_device(virtual_machine)
+        if virtual_disks:
+            boot = True
+            for disk in virtual_disks:
+                cmd = [
+                    'add_disk',
+                    '--machine', f'{virtual_machine.aq_machine_name}',
+                    '--disk', f'{disk.name}',
+                    '--controller', 'sata',
+                    '--size', f'{disk.size}',
+                ]
+                if boot:
+                    cmd.append('--boot')
+                    boot = False
+                cmds.append(cmd)
+        else:
+            cmds.append([
+                'add_disk',
+                '--machine', f'{virtual_machine.aq_machine_name}',
+                '--disk', 'sda',
+                '--controller', 'sata',
+                '--size', f'{virtual_machine.disk}',
+                '--boot',
+            ])
 
         return cmds
 

aquilon/scd_netbox.py

@@ -183,3 +183,15 @@ def get_addresses_from_interface(self, interface):
                 )
 
         return ipv4_addresses
+
+    def get_disks_from_device(self, device):
+        """
+        Get all virtual disks associated with a virtual machine
+        """
+        if isinstance(device, pynetbox.models.virtualization.VirtualMachines):
+            filtered_disks = self.netbox.virtualization.virtual_disks.filter(virtual_machine_id=device.id)
+        else:
+            logging.error('Unsupported device type for disks "%s"', type(device))
+            sys.exit(1)
+
+        return filtered_disks