mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Rename FHRPGroupAssignment object to interface

Browse Source
This commit is contained in:
jeremystretch
2021-11-02 15:10:02 -04:00
parent 264652f2c3
commit 131e433880
14 changed files with 54 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/ipam/api/serializers.py
View File

@@ -110,24 +110,25 @@ class FHRPGroupSerializer(PrimaryModelSerializer):
class FHRPGroupAssignmentSerializer(PrimaryModelSerializer):
url = serializers.HyperlinkedIdentityField(view_name='tenancy-api:contactassignment-detail')
content_type = ContentTypeField(
interface_type = ContentTypeField(
queryset=ContentType.objects.all()
)
object = serializers.SerializerMethodField(read_only=True)
interface = serializers.SerializerMethodField(read_only=True)
class Meta:
model = FHRPGroupAssignment
fields = [
'id', 'url', 'display', 'content_type', 'object_id', 'object', 'priority', 'created', 'last_updated',
'id', 'url', 'display', 'interface_type', 'interface_id', 'interface', 'priority', 'created',
'last_updated',
]
@swagger_serializer_method(serializer_or_field=serializers.DictField)
def get_object(self, obj):
if obj.object is None:
def get_interface(self, obj):
if obj.interface is None:
return None
serializer = get_serializer_for_model(obj.object, prefix='Nested')
serializer = get_serializer_for_model(obj.interface, prefix='Nested')
context = {'request': self.context['request']}
return serializer(obj.object, context=context).data
return serializer(obj.interface, context=context).data
#

2
netbox/ipam/api/views.py
View File

@@ -130,7 +130,7 @@ class FHRPGroupViewSet(CustomFieldModelViewSet):
class FHRPGroupAssignmentViewSet(CustomFieldModelViewSet):
queryset = FHRPGroupAssignment.objects.prefetch_related('group', 'object')
queryset = FHRPGroupAssignment.objects.prefetch_related('group', 'interface')
serializer_class = serializers.FHRPGroupAssignmentSerializer
filterset_class = filtersets.FHRPGroupAssignmentFilterSet