mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
import logging
|
|
|
|
from django.db.models.signals import post_save, post_delete
|
|
from django.dispatch import receiver
|
|
|
|
from dcim.models import CablePath, Interface
|
|
from dcim.utils import create_cablepath
|
|
from .models import WirelessLink
|
|
|
|
|
|
#
|
|
# Wireless links
|
|
#
|
|
|
|
@receiver(post_save, sender=WirelessLink)
|
|
def update_connected_interfaces(instance, created, raw=False, **kwargs):
|
|
"""
|
|
When a WirelessLink is saved, save a reference to it on each connected interface.
|
|
"""
|
|
logger = logging.getLogger('netbox.wireless.wirelesslink')
|
|
if raw:
|
|
logger.debug(f"Skipping endpoint updates for imported wireless link {instance}")
|
|
return
|
|
|
|
if instance.interface_a.wireless_link != instance:
|
|
logger.debug(f"Updating interface A for wireless link {instance}")
|
|
instance.interface_a.wireless_link = instance
|
|
instance.interface_a._link_peer = instance.interface_b
|
|
instance.interface_a.save()
|
|
if instance.interface_b.cable != instance:
|
|
logger.debug(f"Updating interface B for wireless link {instance}")
|
|
instance.interface_b.wireless_link = instance
|
|
instance.interface_b._link_peer = instance.interface_a
|
|
instance.interface_b.save()
|
|
|
|
# Create/update cable paths
|
|
if created:
|
|
for interface in (instance.interface_a, instance.interface_b):
|
|
create_cablepath(interface)
|
|
|
|
|
|
@receiver(post_delete, sender=WirelessLink)
|
|
def nullify_connected_interfaces(instance, **kwargs):
|
|
"""
|
|
When a WirelessLink is deleted, update its two connected Interfaces
|
|
"""
|
|
logger = logging.getLogger('netbox.wireless.wirelesslink')
|
|
|
|
if instance.interface_a is not None:
|
|
logger.debug(f"Nullifying interface A for wireless link {instance}")
|
|
Interface.objects.filter(pk=instance.interface_a.pk).update(
|
|
wireless_link=None,
|
|
_link_peer_type=None,
|
|
_link_peer_id=None
|
|
)
|
|
if instance.interface_b is not None:
|
|
logger.debug(f"Nullifying interface B for wireless link {instance}")
|
|
Interface.objects.filter(pk=instance.interface_b.pk).update(
|
|
wireless_link=None,
|
|
_link_peer_type=None,
|
|
_link_peer_id=None
|
|
)
|
|
|
|
# Delete and retrace any dependent cable paths
|
|
for cablepath in CablePath.objects.filter(path__contains=instance):
|
|
cablepath.delete()
|