From 53f48d15641573eec189aa1102a4735206149d44 Mon Sep 17 00:00:00 2001
From: Aarni Koskela <akx@iki.fi>
Date: Tue, 18 Apr 2017 13:46:47 +0300
Subject: [PATCH] Allow customizing DefaultRouter's schema generator class

---
 rest_framework/routers.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rest_framework/routers.py b/rest_framework/routers.py
index 3d2649727..fce968aa0 100644
--- 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 @@ class DefaultRouter(SimpleRouter):
         """
         Return a schema root view.
         """
-        schema_generator = SchemaGenerator(
+        schema_generator = self.SchemaGenerator(
             title=self.schema_title,
             url=self.schema_url,
             patterns=api_urls