Merge pull request #161 from alehaa/feature/columns

Add device and module type columns

netbox_inventory/models.py

@@ -65,15 +65,15 @@ class Asset(NetBoxModel, ImageAttachmentsMixin):
     device_type = models.ForeignKey(
         to='dcim.DeviceType',
         on_delete=models.PROTECT,
-        related_name='+',
+        related_name='assets',
         blank=True,
         null=True,
         verbose_name='Device Type',
     )
     module_type = models.ForeignKey(
         to='dcim.ModuleType',
         on_delete=models.PROTECT,
-        related_name='+',
+        related_name='assets',
         blank=True,
         null=True,
         verbose_name='Module Type',

netbox_inventory/tables.py

@@ -1,11 +1,15 @@
 from django.db.models.functions import Coalesce
 import django_tables2 as tables
+from django.utils.translation import gettext_lazy as _
 
 from netbox.tables import columns, NetBoxTable
 from tenancy.tables import ContactsColumnMixin
 from .models import Asset, Delivery, InventoryItemType, InventoryItemGroup, Purchase, Supplier
 from .template_content import WARRANTY_PROGRESSBAR
 
+from dcim.tables import DeviceTypeTable, ModuleTypeTable
+from utilities.tables import register_table_column
+
 __all__ = (
     'AssetTable',
     'SupplierTable',
@@ -497,3 +501,27 @@ class Meta(NetBoxTable.Meta):
             'asset_count',
             'inventoryitem_type_count',
         )
+
+
+# ========================
+# DCIM model table columns
+# ========================
+
+asset_count = columns.LinkedCountColumn(
+    viewname='plugins:netbox_inventory:asset_list',
+    url_params={'device_type_id': 'pk'},
+    verbose_name=_('Assets'),
+    accessor="assets__count",
+)
+
+register_table_column(asset_count, 'assets', DeviceTypeTable)
+
+
+asset_count = columns.LinkedCountColumn(
+    viewname='plugins:netbox_inventory:asset_list',
+    url_params={'module_type_id': 'pk'},
+    verbose_name=_('Assets'),
+    accessor="assets__count",
+)
+
+register_table_column(asset_count, 'assets', ModuleTypeTable)