#284 add tags in custom objects table

Author: arthanson

netbox_custom_objects/tables.py

@@ -9,6 +9,7 @@
 from netbox.tables import NetBoxTable, columns
 from utilities.permissions import get_permission_for_model
 
+from netbox_custom_objects.constants import APP_LABEL
 from netbox_custom_objects.models import CustomObject, CustomObjectType
 from netbox_custom_objects.utilities import get_viewname
 
@@ -56,7 +57,7 @@ class CustomObjectTypeTable(NetBoxTable):
         verbose_name=_('Comments'),
     )
     tags = columns.TagColumn(
-        url_name='circuits:provider_list'
+        url_name='plugins:netbox_custom_objects:customobjecttype_list'
     )
     name = tables.Column(
         verbose_name=_('Name'),
@@ -87,6 +88,31 @@ class Meta(NetBoxTable.Meta):
         )
 
 
+class CustomObjectTagColumn(columns.TagColumn):
+    """
+    Custom TagColumn that generates tag filter URLs with the custom_object_type slug.
+    """
+    template_code = """
+    {% load helpers %}
+    {% for tag in value.all %}
+        <a href="{% url 'plugins:netbox_custom_objects:customobject_list' custom_object_type=record.custom_object_type.slug %}?tag={{ tag.slug }}">
+            <span {% if tag.description %}title="{{ tag.description }}"{% endif %} class="badge" style="color: {{ tag.color|fgcolor }}; background-color: #{{ tag.color }}">{{ tag }}</span>
+        </a>
+    {% empty %}
+        <span class="text-muted">&mdash;</span>
+    {% endfor %}
+    """
+
+    def __init__(self):
+        # Override parent __init__ to use our custom template
+        tables.TemplateColumn.__init__(
+            self,
+            orderable=False,
+            template_code=self.template_code,
+            verbose_name=_('Tags'),
+        )
+
+
 class CustomObjectActionsColumn(columns.ActionsColumn):
 
     def render(self, record, table, **kwargs):
@@ -178,6 +204,7 @@ class CustomObjectTable(NetBoxTable):
     actions = CustomObjectActionsColumn(
         actions=('edit', 'delete'),
     )
+    tags = CustomObjectTagColumn()
 
     class Meta(NetBoxTable.Meta):
         model = CustomObject
@@ -188,6 +215,7 @@ class Meta(NetBoxTable.Meta):
             "custom_object_type",
             "created",
             "last_updated",
+            "tags",
         )
         default_columns = (
             "pk",