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

generate_schema -> generateschema, and fix to OpenAPIRenderer

Browse Source
This commit is contained in:
Tom Christie 2018-10-03 14:39:42 +01:00
parent 90fb5ae2fd
commit e5cc846303
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_framework/management/commands/generate_schema.py → rest_framework/management/commands/generateschema.py
View File

@ -1,7 +1,10 @@
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from rest_framework.compat import coreapi from rest_framework.compat import coreapi
from rest_framework.renderers import CoreJSONRenderer, OpenAPIRenderer, JSONOpenAPIRenderer from rest_framework.renderers import (
CoreJSONRenderer, JSONOpenAPIRenderer, OpenAPIRenderer
)
from rest_framework.schemas.generators import SchemaGenerator
from rest_framework.settings import api_settings from rest_framework.settings import api_settings
@ -17,12 +20,11 @@ class Command(BaseCommand):
def handle(self, *args, **options): def handle(self, *args, **options):
assert coreapi is not None, 'coreapi must be installed.' assert coreapi is not None, 'coreapi must be installed.'
generator_class = api_settings.DEFAULT_SCHEMA_GENERATOR_CLASS( generator = SchemaGenerator(
url=options['url'], url=options['url'],
title=options['title'], title=options['title'],
description=options['description'] description=options['description']
) )
generator = generator_class()
schema = generator.get_schema(request=None, public=True) schema = generator.get_schema(request=None, public=True)
@ -36,4 +38,4 @@ class Command(BaseCommand):
'corejson': CoreJSONRenderer(), 'corejson': CoreJSONRenderer(),
'openapi': OpenAPIRenderer(), 'openapi': OpenAPIRenderer(),
'openapi-json': JSONOpenAPIRenderer() 'openapi-json': JSONOpenAPIRenderer()
} }[format]

2
rest_framework/renderers.py
View File

@ -1034,7 +1034,7 @@ class OpenAPIRenderer(_BaseOpenAPIRenderer):
def render(self, data, media_type=None, renderer_context=None): def render(self, data, media_type=None, renderer_context=None):
structure = self.get_structure(data) structure = self.get_structure(data)
return yaml.dumps(structure, default_flow_style=False) return yaml.dump(structure, default_flow_style=False)
class JSONOpenAPIRenderer(_BaseOpenAPIRenderer): class JSONOpenAPIRenderer(_BaseOpenAPIRenderer):