diff --git a/netbox_branching/jobs.py b/netbox_branching/jobs.py index 3140716..c657067 100644 --- a/netbox_branching/jobs.py +++ b/netbox_branching/jobs.py @@ -5,6 +5,7 @@ from core.signals import handle_changed_object, handle_deleted_object from netbox.jobs import JobRunner from utilities.exceptions import AbortTransaction +from .signal_receivers import validate_object_deletion_in_branch from .utilities import ListHandler __all__ = ( @@ -58,6 +59,7 @@ def _disconnect_signal_receivers(self): post_save.disconnect(handle_changed_object) m2m_changed.disconnect(handle_changed_object) pre_delete.disconnect(handle_deleted_object) + pre_delete.disconnect(validate_object_deletion_in_branch) def _reconnect_signal_receivers(self): """ @@ -66,6 +68,7 @@ def _reconnect_signal_receivers(self): post_save.connect(handle_changed_object) m2m_changed.connect(handle_changed_object) pre_delete.connect(handle_deleted_object) + pre_delete.connect(validate_object_deletion_in_branch) def run(self, commit=True, *args, **kwargs): # Initialize logging