fix(17542) add contact to tunnels

This commit is contained in:
Antoine Keranflec'h 2024-10-10 15:38:48 +02:00
parent e3c3ca191c
commit 5b2241aaaf
2 changed files with 9 additions and 3 deletions

View File

@ -6,7 +6,7 @@ from django.urls import reverse
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from netbox.models import ChangeLoggedModel, OrganizationalModel, PrimaryModel from netbox.models import ChangeLoggedModel, OrganizationalModel, PrimaryModel
from netbox.models.features import CustomFieldsMixin, CustomLinksMixin, TagsMixin from netbox.models.features import CustomFieldsMixin, CustomLinksMixin, TagsMixin, ContactsMixin
from vpn.choices import * from vpn.choices import *
__all__ = ( __all__ = (
@ -16,7 +16,7 @@ __all__ = (
) )
class TunnelGroup(OrganizationalModel): class TunnelGroup(ContactsMixin, OrganizationalModel):
""" """
An administrative grouping of Tunnels. This can be used to correlate peer-to-peer tunnels which form a mesh, An administrative grouping of Tunnels. This can be used to correlate peer-to-peer tunnels which form a mesh,
for example. for example.
@ -30,7 +30,7 @@ class TunnelGroup(OrganizationalModel):
return reverse('vpn:tunnelgroup', args=[self.pk]) return reverse('vpn:tunnelgroup', args=[self.pk])
class Tunnel(PrimaryModel): class Tunnel(ContactsMixin, PrimaryModel):
name = models.CharField( name = models.CharField(
verbose_name=_('name'), verbose_name=_('name'),
max_length=100, max_length=100,

View File

@ -62,6 +62,9 @@ class TunnelGroupBulkDeleteView(generic.BulkDeleteView):
filterset = filtersets.TunnelGroupFilterSet filterset = filtersets.TunnelGroupFilterSet
table = tables.TunnelGroupTable table = tables.TunnelGroupTable
@register_model_view(Tunnel, 'contacts')
class TunnelGroupContactsView(ObjectContactsView):
queryset = TunnelGroup.objects.all()
# #
# Tunnels # Tunnels
@ -121,6 +124,9 @@ class TunnelBulkDeleteView(generic.BulkDeleteView):
filterset = filtersets.TunnelFilterSet filterset = filtersets.TunnelFilterSet
table = tables.TunnelTable table = tables.TunnelTable
@register_model_view(Tunnel, 'contacts')
class TunnelContactsView(ObjectContactsView):
queryset = Tunnel.objects.all()
# #
# Tunnel terminations # Tunnel terminations