Creating a Table for a Plugin with Custom Columns from Other NetBox Models

[Brescou]

I am creating an impact management module that is linked with IPAddress, and I want to create a column that displays what the IP address is associated with, either an interface for a device, a VM interface for a VM, or a name stored in custom_field_data.

Here is the code I have so far:

class ImpactListView(generic.ObjectListView):
    queryset = IPAddress.objects.annotate(
        assigned_to=Case(
            When(
                assigned_object_type=ContentType.objects.get_for_model(Interface),
                then=Subquery(
                    Interface.objects.filter(pk=OuterRef('assigned_object_id')).values('device__name')[:1]
                )
            ),
            When(
                assigned_object_type=ContentType.objects.get_for_model(VMInterface),
                then=Subquery(
                    VMInterface.objects.filter(pk=OuterRef('assigned_object_id')).values('virtual_machine__name')[:1]
                )
            ),
            default=Value('cf_nom_long'),
            output_field=CharField(),
        )
    )
    table = ImpactTable

class Impact(NetBoxModel):
    impact = models.TextField(help_text='Impact')
    redundancy = models.BooleanField(default=False)
    ip_address = models.ForeignKey('ipam.IPAddress', on_delete=models.CASCADE, null=True, blank=True)
    vrf = models.ForeignKey('ipam.VRF', on_delete=models.CASCADE, null=True, blank=True)

class ImpactTable(NetBoxTable):
    ip_address = tables.Column(accessor=Accessor("ip_address.address"), verbose_name='IP Address', linkify=True)
    vrf = tables.Column(accessor=Accessor('vrf.name'), verbose_name='VRF', linkify=True)
    role = tables.Column(accessor='ip_address.role', verbose_name='Role', linkify=True)
    assigned_to = tables.Column(verbose_name='Assigned To')
    impact = tables.Column(accessor='impact', verbose_name='Impact')
    redundancy = tables.Column(verbose_name='Redundancy')

    class Meta(NetBoxTable.Meta):
        model = Impact
        fields = ('ip_address', 'vrf', 'role', 'assigned_to', 'impact',  'redundancy')
        default_columns = ('ip_address', 'vrf', 'role', 'assigned_to', 'impact',  'redundancy')

I would appreciate any feedback or suggestions on how to properly create the assigned_to column that accurately displays the association of the IP address.

For now, the table seems empty but in the TableQuerySet I saw all my IP address records.

Thank you!