encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-27 00:19:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle pagination

Browse Source
This commit is contained in:
Bill Collins 2022-03-20 16:39:48 +00:00
parent 764e399cfd
commit 8f85944fa7
2 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_framework/schemas/openapi.py
View File

@ -192,6 +192,9 @@ class AutoSchema(ViewInspector):
return component_name
def get_paginated_component_name(self, serializer):
return 'Paginated' + self.get_component_name(serializer, 'GET')
def get_error_component_name(self, serializer):
return self.get_component_name(serializer, 'GET') + 'Error'
@ -716,6 +719,9 @@ class AutoSchema(ViewInspector):
def _get_reference(self, serializer, method):
return {'$ref': '#/components/schemas/{}'.format(self.get_component_name(serializer, method))}
def _get_paginated_reference(self, serializer):
return {'$ref': '#/components/schemas/{}'.format(self.get_paginated_component_name(serializer))}
def _get_error_reference(self, serializer):
return {'$ref': '#/components/schemas/{}'.format(self.get_error_component_name(serializer))}
@ -763,7 +769,9 @@ class AutoSchema(ViewInspector):
}
paginator = self.get_paginator()
if paginator:
response_schema = paginator.get_paginated_response_schema(response_schema)
pagination_schema = paginator.get_paginated_response_schema(response_schema)
self.components[self.get_paginated_component_name(serializer)] = pagination_schema
response_schema = self._get_paginated_reference(serializer)
else:
response_schema = item_schema
status_code = '201' if method == 'POST' else '200'

17
tests/schemas/test_openapi.py
View File

@ -474,13 +474,7 @@ class TestOperationIntrospection(TestCase):
'content': {
'application/json': {
'schema': {
'type': 'object',
'item': {
'type': 'array',
'items': {
'$ref': '#/components/schemas/Item'
},
},
'$ref': '#/components/schemas/PaginatedItem',
},
},
},
@ -488,6 +482,15 @@ class TestOperationIntrospection(TestCase):
}
components = inspector.get_components(path, method)
assert components == {
'PaginatedItem': {
'type': 'object',
'item': {
'type': 'array',
'items': {
'$ref': '#/components/schemas/Item'
},
},
},
'Item': {
'type': 'object',
'properties': {