Merge pull request #184 from netboxlabs/183-delete-cot

183 delete cot

Author: bctiemann

netbox_custom_objects/models.py

@@ -553,6 +553,11 @@ def delete(self, *args, **kwargs):
         self.clear_model_cache(self.id)
 
         model = self.get_model()
+
+        # Delete all CustomObjectTypeFields that reference this CustomObjectType
+        for field in CustomObjectTypeField.objects.filter(related_object_type=self.content_type):
+            field.delete()
+
         object_type = ObjectType.objects.get_for_model(model)
         ObjectChange.objects.filter(changed_object_type=object_type).delete()
         super().delete(*args, **kwargs)

netbox_custom_objects/views.py

@@ -189,6 +189,16 @@ def _get_dependent_objects(self, obj):
         dependent_objects = super()._get_dependent_objects(obj)
         model = obj.get_model()
         dependent_objects[model] = list(model.objects.all())
+
+        # Find CustomObjectTypeFields that reference this CustomObjectType
+        referencing_fields = CustomObjectTypeField.objects.filter(
+            related_object_type=obj.content_type
+        )
+
+        # Add the CustomObjectTypeFields that reference this CustomObjectType
+        if referencing_fields.exists():
+            dependent_objects[CustomObjectTypeField] = list(referencing_fields)
+
         return dependent_objects