Allow customizing DefaultRouter's schema generator class

akx · akx · commit 53f48d156415 · 2017-04-18T14:14:04.000+03:00
diff --git a/rest_framework/routers.py b/rest_framework/routers.py
@@ -316,6 +316,7 @@ class DefaultRouter(SimpleRouter):
     default_schema_renderers = None
     APIRootView = APIRootView
     APISchemaView = SchemaView
+    SchemaGenerator = SchemaGenerator
 
     def __init__(self, *args, **kwargs):
         if 'schema_title' in kwargs:
@@ -342,7 +343,7 @@ def get_schema_root_view(self, api_urls=None):
         """
         Return a schema root view.
         """
-        schema_generator = SchemaGenerator(
+        schema_generator = self.SchemaGenerator(
             title=self.schema_title,
             url=self.schema_url,
             patterns=api_urls
