coderanger
diff --git a/‎docs/community/release-notes.md
Lines changed: 1 addition & 0 deletions b/‎docs/community/release-notes.md
Lines changed: 1 addition & 0 deletions
diff --git a/‎rest_framework/management/commands/generateschema.py
Lines changed: 6 additions & 1 deletion b/‎rest_framework/management/commands/generateschema.py
Lines changed: 6 additions & 1 deletion
diff --git a/‎tests/schemas/test_managementcommand.py
Lines changed: 17 additions & 0 deletions b/‎tests/schemas/test_managementcommand.py
Lines changed: 17 additions & 0 deletions
@@ -45,6 +45,7 @@ You can determine your currently installed version using `pip show`:
 **Date**: [Unreleased][3.10.0-milestone]
 
 * Resolve DeprecationWarning with markdown. [#6317][gh6317]
+* Add `generateschema --generator_class` CLI option
 
 
 ## 3.9.x series
 
@@ -1,4 +1,5 @@
 from django.core.management.base import BaseCommand
+from django.utils.module_loading import import_string
 
 from rest_framework import renderers
 from rest_framework.schemas import coreapi
@@ -22,9 +23,13 @@ def add_arguments(self, parser):
             parser.add_argument('--format', dest="format", choices=['openapi', 'openapi-json', 'corejson'], default='openapi', type=str)
         else:
             parser.add_argument('--format', dest="format", choices=['openapi', 'openapi-json'], default='openapi', type=str)
+        parser.add_argument('--generator_class', dest="generator_class", default=None, type=str)
 
     def handle(self, *args, **options):
-        generator_class = self.get_generator_class()
+        if options['generator_class']:
+            generator_class = import_string(options['generator_class'])
+        else:
+            generator_class = self.get_generator_class()
         generator = generator_class(
             url=options['url'],
             title=options['title'],
 
@@ -22,6 +22,16 @@ def get(self, request):
 ]
 
 
+class CustomSchemaGenerator:
+    SCHEMA = {"key": "value"}
+
+    def __init__(self, *args, **kwargs):
+        pass
+
+    def get_schema(self, **kwargs):
+        return self.SCHEMA
+
+
 @override_settings(ROOT_URLCONF=__name__)
 @pytest.mark.skipif(not uritemplate, reason='uritemplate is not installed')
 class GenerateSchemaTests(TestCase):
@@ -56,6 +66,13 @@ def test_renders_openapi_json_schema(self):
         out_json = json.loads(self.out.getvalue())
         assert out_json['openapi'] == '3.0.2'
 
+    def test_accepts_custom_schema_generator(self):
+        call_command('generateschema',
+                     '--generator_class={}.{}'.format(__name__, CustomSchemaGenerator.__name__),
+                     stdout=self.out)
+        out_json = yaml.safe_load(self.out.getvalue())
+        assert out_json == CustomSchemaGenerator.SCHEMA
+
     @pytest.mark.skipif(yaml is None, reason='PyYAML is required.')
     @override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema'})
     def test_coreapi_renders_default_schema_with_custom_title_url_and_description(self):