Compare commits

...

2 Commits

Author SHA1 Message Date
Smixi c3466cf226
Merge 832f7452f2 into b5bb732031 2024-04-17 10:24:51 -07:00
Sami Tahri 832f7452f2 fix: SerializedPKRelatedField schema now use nested serializer or response 2024-02-04 01:24:19 +01:00
1 changed files with 11 additions and 0 deletions

View File

@ -257,3 +257,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)