mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Update signal handlers (WIP)

Browse Source
This commit is contained in:
jeremystretch
2022-05-02 08:26:02 -04:00
parent 5d37f9f975
commit c3742f63fd
2 changed files with 23 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/dcim/models/device_components.py
View File

@ -139,18 +139,6 @@ class LinkTermination(models.Model):
help_text="Treat as if a cable is connected" help_text="Treat as if a cable is connected"
) )
# Generic relations to Cable. These ensure that an attached Cable is deleted if the terminated object is deleted.
_cabled_as_a = GenericRelation(
to='dcim.Cable',
content_type_field='termination_a_type',
object_id_field='termination_a_id'
)
_cabled_as_b = GenericRelation(
to='dcim.Cable',
content_type_field='termination_b_type',
object_id_field='termination_b_id'
)
class Meta: class Meta:
abstract = True abstract = True

39
netbox/dcim/signals.py
View File

@ -5,7 +5,7 @@ from django.db.models.signals import post_save, post_delete, pre_delete
from django.dispatch import receiver from django.dispatch import receiver
from .choices import LinkStatusChoices from .choices import LinkStatusChoices
from .models import Cable, CablePath, Device, PathEndpoint, PowerPanel, Rack, Location, VirtualChassis from .models import Cable, CablePath, CableTermination, Device, PathEndpoint, PowerPanel, Rack, Location, VirtualChassis
from .utils import create_cablepath, rebuild_paths from .utils import create_cablepath, rebuild_paths
@ -80,12 +80,6 @@ def update_connected_endpoints(instance, created, raw=False, **kwargs):
return return
# TODO: Update link peer fields # TODO: Update link peer fields
# Cache the Cable on its termination points
for term in instance.terminations.all():
if term.termination.cable != instance:
logger.debug(f"Updating termination A for cable {instance}: {term}")
term.termination.cable = instance
term.save()
# # Create/update cable paths # # Create/update cable paths
# if created: # if created:
@ -104,6 +98,19 @@ def update_connected_endpoints(instance, created, raw=False, **kwargs):
# rebuild_paths(instance) # rebuild_paths(instance)
@receiver(post_save, sender=CableTermination)
def cache_cable_on_endpoints(instance, created, raw=False, **kwargs):
if not raw:
model = instance.termination_type.model_class()
model.objects.filter(pk=instance.termination_id).update(cable=instance.cable)
@receiver(post_delete, sender=CableTermination)
def clear_cable_on_endpoints(instance, **kwargs):
model = instance.termination_type.model_class()
model.objects.filter(pk=instance.termination_id).update(cable=None)
@receiver(post_delete, sender=Cable) @receiver(post_delete, sender=Cable)
def nullify_connected_endpoints(instance, **kwargs): def nullify_connected_endpoints(instance, **kwargs):
""" """
@ -111,15 +118,15 @@ def nullify_connected_endpoints(instance, **kwargs):
""" """
logger = logging.getLogger('netbox.dcim.cable') logger = logging.getLogger('netbox.dcim.cable')
# Disassociate the Cable from its termination points # # Disassociate the Cable from its termination points
if instance.termination_a: # if instance.termination_a:
logger.debug(f"Nullifying termination A for cable {instance}") # logger.debug(f"Nullifying termination A for cable {instance}")
model = instance.termination_a_type.model_class() # model = instance.termination_a_type.model_class()
model.objects.filter(pk__in=instance.termination_a_ids).update(_link_peer_type=None, _link_peer_id=None) # model.objects.filter(pk__in=instance.termination_a_ids).update(_link_peer_type=None, _link_peer_id=None)
if instance.termination_b: # if instance.termination_b:
logger.debug(f"Nullifying termination B for cable {instance}") # logger.debug(f"Nullifying termination B for cable {instance}")
model = instance.termination_b_type.model_class() # model = instance.termination_b_type.model_class()
model.objects.filter(pk__in=instance.termination_b_ids).update(_link_peer_type=None, _link_peer_id=None) # model.objects.filter(pk__in=instance.termination_b_ids).update(_link_peer_type=None, _link_peer_id=None)
# Delete and retrace any dependent cable paths # Delete and retrace any dependent cable paths
for cablepath in CablePath.objects.filter(path__contains=instance): for cablepath in CablePath.objects.filter(path__contains=instance):