1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from drf_rw_serializers.generics import GenericAPIView as RWGenericAPIView
from drf_spectacular.openapi import AutoSchema
class CustomAutoSchema(AutoSchema):
""" Utilize custom drf_rw_serializers methods for directional serializers """
def get_request_serializer(self):
if isinstance(self.view, RWGenericAPIView):
return self.view.get_write_serializer()
return self._get_serializer()
def get_response_serializers(self):
if isinstance(self.view, RWGenericAPIView):
return self.view.get_read_serializer()
return self._get_serializer()
|
