From 306e53461dca95b32fed77563c094d6ec60ebd70 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cristi=20V=C3=AEjdea?= Date: Tue, 11 Dec 2018 23:38:50 +0200 Subject: [PATCH] Fix lint errors --- src/drf_yasg/renderers.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/drf_yasg/renderers.py b/src/drf_yasg/renderers.py index 37edfa3..0369fed 100644 --- a/src/drf_yasg/renderers.py +++ b/src/drf_yasg/renderers.py @@ -1,7 +1,7 @@ -import django.template.loader import six from django.shortcuts import resolve_url +from django.template.loader import render_to_string from django.utils.functional import Promise from rest_framework.renderers import BaseRenderer, JSONRenderer, TemplateHTMLRenderer from rest_framework.utils import json @@ -32,6 +32,7 @@ class _SpecRenderer(BaseRenderer): # in that case, it's probably better to let the default ``JSONRenderer`` render it # see https://github.com/axnsan12/drf-yasg/issues/58 return JSONRenderer().render(data, media_type, renderer_context) + return codec.encode(data) @@ -68,8 +69,9 @@ class _UIRenderer(BaseRenderer): # in that case, it's probably better to let the default ``TemplateHTMLRenderer`` render it # see https://github.com/axnsan12/drf-yasg/issues/58 return TemplateHTMLRenderer().render(swagger, accepted_media_type, renderer_context) + self.set_context(renderer_context, swagger) - return django.template.loader.render_to_string(self.template, renderer_context, renderer_context['request']) + return render_to_string(self.template, renderer_context, renderer_context['request']) def set_context(self, renderer_context, swagger=None): renderer_context['title'] = swagger.info.title or '' if swagger else ''