Add EXCLUDED_MEDIA_TYPES setting

Closes #158.
This commit is contained in:
Cristi Vîjdea
2018-08-08 00:56:18 +03:00
parent 5cd642c9a0
commit a5eb3dfa91
4 changed files with 14 additions and 1 deletions
+2
View File
@@ -26,6 +26,8 @@ SWAGGER_DEFAULTS = {
'drf_yasg.inspectors.CoreAPICompatInspector',
],
'EXCLUDED_MEDIA_TYPES': ['html'],
'DEFAULT_INFO': None,
'DEFAULT_API_URL': None,
+3 -1
View File
@@ -11,6 +11,8 @@ from rest_framework.settings import api_settings as rest_framework_settings
from rest_framework.utils import encoders, json
from rest_framework.views import APIView
from drf_yasg.app_settings import swagger_settings
logger = logging.getLogger(__name__)
@@ -328,7 +330,7 @@ def get_produces(renderer_classes):
:rtype: list[str]
"""
media_types = [renderer.media_type for renderer in renderer_classes or []]
media_types = [encoding for encoding in media_types if 'html' not in encoding]
media_types = [encoding for encoding in media_types if not any(excluded in encoding for excluded in swagger_settings.EXCLUDED_MEDIA_TYPES)]
return media_types