Fix representation of connected_endpoint_type for non-connected components

This commit is contained in:
Jeremy Stretch 2018-12-06 16:39:03 -05:00
parent 703be259fd
commit 9689ba2c4f

View File

@ -28,12 +28,12 @@ class ConnectedEndpointSerializer(ValidatedModelSerializer):
connection_status = ChoiceField(choices=CONNECTION_STATUS_CHOICES, read_only=True)
def get_connected_endpoint_type(self, obj):
if obj.connected_endpoint is None:
return None
return '{}.{}'.format(
obj.connected_endpoint._meta.app_label,
obj.connected_endpoint._meta.model_name
)
if hasattr(obj, 'connected_endpoint') and obj.connected_endpoint is not None:
return '{}.{}'.format(
obj.connected_endpoint._meta.app_label,
obj.connected_endpoint._meta.model_name
)
return None
def get_connected_endpoint(self, obj):
"""