Fixes #3709: Prevent exception when importing an invalid cable definition

This commit is contained in:
Jeremy Stretch
2019-11-26 16:46:51 -05:00
parent fb3595a4c5
commit 43350136db
2 changed files with 5 additions and 0 deletions

View File

@@ -2838,6 +2838,8 @@ class Cable(ChangeLoggedModel):
def clean(self):
# Validate that termination A exists
if not hasattr(self, 'termination_a_type'):
raise ValidationError('Termination A type has not been specified')
try:
self.termination_a_type.model_class().objects.get(pk=self.termination_a_id)
except ObjectDoesNotExist:
@@ -2846,6 +2848,8 @@ class Cable(ChangeLoggedModel):
})
# Validate that termination B exists
if not hasattr(self, 'termination_b_type'):
raise ValidationError('Termination B type has not been specified')
try:
self.termination_b_type.model_class().objects.get(pk=self.termination_b_id)
except ObjectDoesNotExist: