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

[OpenAPI] Schemas Generator minor improvements, better generalization for third-party integrations

Browse Source
This commit is contained in:
peppelinux 2020-08-05 11:33:58 +02:00
parent 559088463b
commit eb1e2703de
4 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rest_framework/schemas/__init__.py
View File

@ -32,7 +32,7 @@ def get_schema_view(
public=False, patterns=None, generator_class=None,
authentication_classes=api_settings.DEFAULT_AUTHENTICATION_CLASSES,
permission_classes=api_settings.DEFAULT_PERMISSION_CLASSES,
version=None):
version=None, **kwargs):
"""
Return a schema view.
"""
@ -44,7 +44,7 @@ def get_schema_view(
generator = generator_class(
title=title, url=url, description=description,
urlconf=urlconf, patterns=patterns, version=version
urlconf=urlconf, patterns=patterns, version=version, **kwargs
)
# Avoid import cycle on APIView

7
rest_framework/schemas/generators.py
View File

@ -151,7 +151,9 @@ class BaseSchemaGenerator:
# Set by 'SCHEMA_COERCE_PATH_PK'.
coerce_path_pk = None
def __init__(self, title=None, url=None, description=None, patterns=None, urlconf=None, version=None):
def __init__(self, title=None, url=None, description=None,
patterns=None, urlconf=None, version=None, **kwargs):
if url and not url.endswith('/'):
url += '/'
@ -165,6 +167,9 @@ class BaseSchemaGenerator:
self.url = url
self.endpoints = None
for k, v in kwargs.items():
setattr(self, k, v)
def _initialise_endpoints(self):
if self.endpoints is None:
inspector = self.endpoint_inspector_cls(self.patterns, self.urlconf)

7
rest_framework/schemas/openapi.py
View File

@ -113,9 +113,8 @@ class SchemaGenerator(BaseSchemaGenerator):
return schema
# View Inspectors
class AutoSchema(ViewInspector):
def __init__(self, tags=None, operation_id_base=None, component_name=None):
@ -471,13 +470,15 @@ class AutoSchema(ViewInspector):
if isinstance(field, serializers.FloatField):
content = {
'type': 'number',
'format': 'float'
}
self._map_min_max(field, content)
return content
if isinstance(field, serializers.IntegerField):
content = {
'type': 'integer'
'type': 'integer',
'format': 'int64'
}
self._map_min_max(field, content)
# 2147483647 is max for int32_size, so we use int64 for format

2
tests/schemas/test_openapi.py
View File

@ -54,7 +54,7 @@ class TestFieldMapping(TestCase):
cases = [
(serializers.ListField(), {'items': {}, 'type': 'array'}),
(serializers.ListField(child=serializers.BooleanField()), {'items': {'type': 'boolean'}, 'type': 'array'}),
(serializers.ListField(child=serializers.FloatField()), {'items': {'type': 'number'}, 'type': 'array'}),
(serializers.ListField(child=serializers.FloatField()), {'items': {'type': 'number', 'format': 'float'}, 'type': 'array'}),
(serializers.ListField(child=serializers.CharField()), {'items': {'type': 'string'}, 'type': 'array'}),
(serializers.ListField(child=serializers.IntegerField(max_value=4294967295)),
{'items': {'type': 'integer', 'maximum': 4294967295, 'format': 'int64'}, 'type': 'array'}),