Просмотр исходного кода

Fix representation of connected_endpoint_type for non-connected components

Jeremy Stretch 7 лет назад
Родитель
Сommit
9689ba2c4f
1 измененных файлов с 6 добавлено и 6 удалено
  1. 6 6
      netbox/dcim/api/serializers.py

+ 6 - 6
netbox/dcim/api/serializers.py

@@ -28,12 +28,12 @@ class ConnectedEndpointSerializer(ValidatedModelSerializer):
     connection_status = ChoiceField(choices=CONNECTION_STATUS_CHOICES, read_only=True)
     connection_status = ChoiceField(choices=CONNECTION_STATUS_CHOICES, read_only=True)
 
 
     def get_connected_endpoint_type(self, obj):
     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):
     def get_connected_endpoint(self, obj):
         """
         """