first commit

netbox_librenms_plugin/tables/ipaddresses.py

@@ -0,0 +1,122 @@
+import django_tables2 as tables
+from django.utils.html import format_html, mark_safe
+from netbox.tables.columns import ToggleColumn
+from utilities.paginator import EnhancedPaginator
+
+from netbox_librenms_plugin.utils import get_table_paginate_count
+
+
+class IPAddressTable(tables.Table):
+    """
+    Table for displaying LibreNMS IP address data.
+    """
+
+    def __init__(self, *args, **kwargs):
+        """Initialize IP address table."""
+        super().__init__(*args, **kwargs)
+
+    class Meta:
+        """Meta options for IPAddressTable."""
+
+        sequence = [
+            "selection",
+            "address",
+            "prefix_length",
+            "device",
+            "interface_name",
+            "vrf",
+        ]
+        attrs = {
+            "class": "table table-hover object-list",
+            "id": "librenms-ipaddress-table",
+        }
+        row_attrs = {
+            "data-interface": lambda record: record["ip_address"],
+            "data-name": lambda record: record["ip_address"],
+        }
+
+    selection = ToggleColumn(
+        orderable=False,
+        visible=True,
+        attrs={"td": {"data-col": "selection"}, "input": {"name": "select"}},
+        accessor="ip_address",
+    )
+
+    address = tables.Column(
+        accessor="ip_address",
+        verbose_name="IP Address",
+        linkify=lambda record: record.get("ip_url"),
+        attrs={"td": {"data-col": "address"}},
+    )
+    prefix_length = tables.Column(
+        accessor="prefix_length",
+        verbose_name="Prefix Length",
+        attrs={"td": {"data-col": "prefix"}},
+    )
+    device = tables.Column(
+        linkify=lambda record: record.get("device_url"),
+        attrs={"td": {"data-col": "device"}},
+    )
+    interface_name = tables.Column(
+        accessor="interface_name",
+        verbose_name="Interface",
+        linkify=lambda record: record.get("interface_url"),
+        attrs={"td": {"data-col": "interface"}},
+    )
+    vrf = tables.TemplateColumn(
+        template_code="""
+        <select id="vrf_select_{{ record.ip_address|slugify }}" class="form-select vrf-select" data-ip="{{ record.ip_address }}" data-prefix="{{ record.prefix_length }}" data-row-id="{{ record.ip_address }}" name="vrf_{{ record.ip_address }}">
+            <option value="">Global</option>
+            {% for vrf in record.vrfs %}
+                <option value="{{ vrf.pk }}" {% if record.vrf_id == vrf.pk %}selected{% endif %}>
+                    {{ vrf.name }}
+                </option>
+            {% endfor %}
+        </select>
+        """,
+        attrs={"td": {"data-col": "vrf"}},
+        verbose_name="VRF",
+    )
+    status = tables.Column(
+        verbose_name="Status",
+        attrs={"td": {"data-col": "status"}},
+    )
+
+    def render_status(self, value, record):
+        """Render the status column with appropriate buttons or text styling"""
+        if value == "update":
+            return format_html(
+                '<button type="submit" class="btn btn-sm btn-warning" onclick="document.getElementById(\'selected_ip\').value=\'{}\'">'
+                '<i class="mdi mdi-pencil" aria-hidden="true"></i> Update</button>',
+                record["ip_address"],
+            )
+        elif value == "matched":
+            return mark_safe('<span class="text-success"><i class="mdi mdi-check-circle"></i> Synced</span>')
+        elif record.get("interface_url"):
+            return format_html(
+                '<button type="submit" class="btn btn-sm btn-primary" onclick="document.getElementById(\'selected_ip\').value=\'{}\'">'
+                '<i class="mdi mdi-plus-thick" aria-hidden="true"></i> Create</button>',
+                record["ip_address"],
+            )
+        return mark_safe('<span class="text-muted">Missing NetBox Object</span>')
+
+    def render_device(self, value, record):
+        """Render the device column with a link if available"""
+        if url := record.get("device_url"):
+            return format_html('<a href="{}">{}</a>', url, value)
+        return value
+
+    def render_interface_name(self, value, record):
+        """Render the interface column with a link if available"""
+        if url := record.get("interface_url"):
+            return format_html('<a href="{}">{}</a>', url, value)
+        return value
+
+    def configure(self, request):
+        """Configure the table"""
+        paginate = {
+            "paginator_class": EnhancedPaginator,
+            "per_page": get_table_paginate_count(request, self.prefix),
+        }
+
+        tables.RequestConfig(request, paginate).configure(self)