Extend contract support to Device

Changes to be committed:
	modified:   pyproject.toml
	modified:   src/netbox_contract/__init__.py
	modified:   src/netbox_contract/tables.py
	modified:   src/netbox_contract/template_content.py
	modified:   src/netbox_contract/views.py

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "netbox-contract"
-version = "2.0.0"
+version = "2.0.1"
 authors = [
   { name="Marc Lebreuil", email="[email protected]" },
 ]

src/netbox_contract/__init__.py

@@ -4,7 +4,7 @@ class ContractsConfig(PluginConfig):
     name = 'netbox_contract'
     verbose_name = 'Netbox contract'
     description = 'Contract management plugin for Netbox'
-    version = '2.0.0'
+    version = '2.0.1'
     author = 'Marc Lebreuil'
     author_email = '[email protected]'
     base_url = 'contracts'

src/netbox_contract/tables.py

@@ -21,8 +21,38 @@ class ContractAssignementListTable(NetBoxTable):
 
     class Meta(NetBoxTable.Meta):
         model = ContractAssignement
-        fields = ('pk', 'content_type', 'content_object', 'contract', 'actions')
-        default_columns = ('pk', 'content_type', 'content_object', 'contract')
+        fields = ('pk', 'content_type', 'content_object', 'contract','contract__external_partie', 'actions')
+        default_columns = ('pk', 'content_type', 'content_object', 'contract','contract__external_partie')
+
+class ContractAssignementObjectTable(NetBoxTable):
+    contract = tables.Column(
+        linkify=True
+    )
+    actions = columns.ActionsColumn(
+        actions=('edit', 'delete')
+    )
+
+    class Meta(NetBoxTable.Meta):
+        model = ContractAssignement
+        fields = ('pk','contract','contract__external_partie','contract__status', 'actions')
+        default_columns = ('pk', 'contract','contract__external_partie','contract__status')
+
+class ContractAssignementContractTable(NetBoxTable):
+    content_type = columns.ContentTypeColumn(
+        verbose_name='Object Type'
+    )
+    content_object = tables.Column(
+        linkify=True,
+        orderable=False
+    )
+    actions = columns.ActionsColumn(
+        actions=('edit', 'delete')
+    )
+
+    class Meta(NetBoxTable.Meta):
+        model = ContractAssignement
+        fields = ('pk', 'content_type', 'content_object','actions')
+        default_columns = ('pk', 'content_type', 'content_object')
 
 class ContractListTable(NetBoxTable):
 

src/netbox_contract/template_content.py

@@ -1,6 +1,7 @@
 from django.contrib.contenttypes.models import ContentType
 from extras.plugins import PluginTemplateExtension
 from circuits.models import Circuit
+from dcim.models import Device
 from .models import ContractAssignement
 from . import tables
 
@@ -23,11 +24,25 @@ def full_width_page(self):
         circuit = self.context['object']
         circuit_type = ContentType.objects.get_for_model(Circuit)
         contract_assignements = ContractAssignement.objects.filter(content_type__pk=circuit_type.id, object_id=circuit.id)
-        assignements_table = tables.ContractAssignementListTable(contract_assignements)
+        assignements_table = tables.ContractAssignementObjectTable(contract_assignements)
         assignements_table.configure(self.context['request'])
 
         return self.render('contract_assignements_bottom.html', extra_context={
             'assignements_table': assignements_table,
         })
 
-template_extensions = [ CircuitContracts, CircuitContractAssignements]
+class DeviceContractAssignements(PluginTemplateExtension):
+    model = 'dcim.device'
+
+    def full_width_page(self):
+        device = self.context['object']
+        device_type = ContentType.objects.get_for_model(Device)
+        contract_assignements = ContractAssignement.objects.filter(content_type__pk=device_type.id, object_id=device.id)
+        assignements_table = tables.ContractAssignementObjectTable(contract_assignements)
+        assignements_table.configure(self.context['request'])
+
+        return self.render('contract_assignements_bottom.html', extra_context={
+            'assignements_table': assignements_table,
+        })
+
+template_extensions = [ CircuitContracts, CircuitContractAssignements, DeviceContractAssignements]

src/netbox_contract/views.py

@@ -85,7 +85,7 @@ def get_extra_context(self, request, instance):
         invoices_table.configure(request)
         circuit_table = tables.ContractCircuitListTable(instance.circuit.all())
         circuit_table.configure(request)
-        assignements_table = tables.ContractAssignementListTable(instance.assignments.all())
+        assignements_table = tables.ContractAssignementContractTable(instance.assignments.all())
         assignements_table.configure(request)
 
         return {