From 79b9ef7d0c26060989a78bfd44e041ac3eb68878 Mon Sep 17 00:00:00 2001 From: Sami Tahri Date: Sun, 4 Feb 2024 01:24:19 +0100 Subject: [PATCH] fix: SerializedPKRelatedField schema now use nested serializer or response --- netbox/core/api/schema.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/netbox/core/api/schema.py b/netbox/core/api/schema.py index 5f64dcc53..bcc49d3fc 100644 --- a/netbox/core/api/schema.py +++ b/netbox/core/api/schema.py @@ -255,3 +255,14 @@ class NetBoxAutoSchema(AutoSchema): if '{id}' in self.path: return f"{self.method.capitalize()} a {model_name} object." return f"{self.method.capitalize()} a list of {model_name} objects." + + +class FixSerializedPKRelatedField(OpenApiSerializerFieldExtension): + target_class = 'netbox.api.fields.SerializedPKRelatedField' + + def map_serializer_field(self, auto_schema, direction): + if direction == "response": + component = auto_schema.resolve_serializer(self.target.serializer, direction) + return component.ref if component else None + else: + return build_basic_type(OpenApiTypes.INT)