Deferred schema renderer creation to avoid requiring pyyaml.

Closes #6366.
This commit is contained in:
Carlton Gibson 2019-01-21 20:54:53 +01:00
parent 271c4c5920
commit 4f70b6b9b7

View File

@ -32,8 +32,10 @@ class Command(BaseCommand):
self.stdout.write(output.decode('utf-8')) self.stdout.write(output.decode('utf-8'))
def get_renderer(self, format): def get_renderer(self, format):
return { renderer_cls = {
'corejson': CoreJSONRenderer(), 'corejson': CoreJSONRenderer,
'openapi': OpenAPIRenderer(), 'openapi': OpenAPIRenderer,
'openapi-json': JSONOpenAPIRenderer() 'openapi-json': JSONOpenAPIRenderer,
}[format] }[format]
return renderer_cls()