diff --git a/netbox/secrets/api/serializers.py b/netbox/secrets/api/serializers.py index ea245e753..e0ab3bcc3 100644 --- a/netbox/secrets/api/serializers.py +++ b/netbox/secrets/api/serializers.py @@ -1,6 +1,7 @@ from rest_framework import serializers -from dcim.api.serializers import DeviceNestedSerializer +from dcim.models import Device +from ipam.api.serializers import IPAddressNestedSerializer from secrets.models import Secret, SecretRole @@ -25,8 +26,16 @@ class SecretRoleNestedSerializer(SecretRoleSerializer): # Secrets # +class SecretDeviceSerializer(serializers.ModelSerializer): + primary_ip = IPAddressNestedSerializer() + + class Meta: + model = Device + fields = ['id', 'name', 'primary_ip'] + + class SecretSerializer(serializers.ModelSerializer): - device = DeviceNestedSerializer() + device = SecretDeviceSerializer() role = SecretRoleNestedSerializer() class Meta: