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'))
def get_renderer(self, format):
return {
'corejson': CoreJSONRenderer(),
'openapi': OpenAPIRenderer(),
'openapi-json': JSONOpenAPIRenderer()
renderer_cls = {
'corejson': CoreJSONRenderer,
'openapi': OpenAPIRenderer,
'openapi-json': JSONOpenAPIRenderer,
}[format]
return renderer_cls()