From 2656696a0f752a507b6f9f32cd2d2c159e8a3e5b Mon Sep 17 00:00:00 2001 From: Pavel White Date: Thu, 13 Jun 2019 03:38:17 +0300 Subject: [PATCH] generators: support custom ReferenceResolver class (#350) --- src/drf_yasg/generators.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/drf_yasg/generators.py b/src/drf_yasg/generators.py index 708b958..9011bba 100644 --- a/src/drf_yasg/generators.py +++ b/src/drf_yasg/generators.py @@ -160,6 +160,7 @@ class OpenAPISchemaGenerator(object): Method implementations shamelessly stolen and adapted from rest-framework ``SchemaGenerator``. """ endpoint_enumerator_class = EndpointEnumerator + reference_resolver_class = ReferenceResolver def __init__(self, info, version='', url=None, patterns=None, urlconf=None): """ @@ -238,7 +239,7 @@ class OpenAPISchemaGenerator(object): :rtype: openapi.Swagger """ endpoints = self.get_endpoints(request) - components = ReferenceResolver(openapi.SCHEMA_DEFINITIONS, force_init=True) + components = self.reference_resolver_class(openapi.SCHEMA_DEFINITIONS, force_init=True) self.consumes = get_consumes(api_settings.DEFAULT_PARSER_CLASSES) self.produces = get_produces(api_settings.DEFAULT_RENDERER_CLASSES) paths, prefix = self.get_paths(endpoints, components, request, public)