use more precise exception

This commit is contained in:
Chris Meyers 2016-12-05 13:54:33 -05:00
parent f47f8abe42
commit a5b1c7b579
2 changed files with 3 additions and 4 deletions

View File

@ -609,7 +609,7 @@ class UnifiedJobSerializer(BaseSerializer):
summary_fields['source_workflow_job'] = {}
try:
summary_obj = obj.unified_job_node.workflow_job
except ObjectDoesNotExist:
except UnifiedJob.unified_job_node.RelatedObjectDoesNotExist:
return summary_fields
for field in SUMMARIZABLE_FK_FIELDS['job']:

View File

@ -20,7 +20,6 @@ from django.utils.translation import ugettext_lazy as _
from django.utils.timezone import now
from django.utils.encoding import smart_text
from django.apps import apps
from django.core.exceptions import ObjectDoesNotExist
# Django-Polymorphic
from polymorphic import PolymorphicModel
@ -783,7 +782,7 @@ class UnifiedJob(PolymorphicModel, PasswordFieldsModel, CommonModelNameNotUnique
if self.spawned_by_workflow:
try:
return self.unified_job_node.workflow_job.pk
except ObjectDoesNotExist:
except UnifiedJob.unified_job_node.RelatedObjectDoesNotExist:
pass
return None
@ -792,7 +791,7 @@ class UnifiedJob(PolymorphicModel, PasswordFieldsModel, CommonModelNameNotUnique
if self.spawned_by_workflow:
try:
return self.unified_job_node.pk
except ObjectDoesNotExist:
except UnifiedJob.unified_job_node.RelatedObjectDoesNotExist:
pass
return None