Add --api-version CLI option to generateschema (encode#8663)

* Add --version CLI option to generateschema * fix conflicting argument name
sigvef · Dec 3, 2022 · 045fd9c · 045fd9c
1 parent 0c40837
commit 045fd9c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rest_framework/management/commands/generateschema.py b/rest_framework/management/commands/generateschema.py
@@ -26,6 +26,7 @@ def add_arguments(self, parser):
         parser.add_argument('--urlconf', dest="urlconf", default=None, type=str)
         parser.add_argument('--generator_class', dest="generator_class", default=None, type=str)
         parser.add_argument('--file', dest="file", default=None, type=str)
+        parser.add_argument('--api_version', dest="api_version", default='', type=str)
 
     def handle(self, *args, **options):
         if options['generator_class']:
@@ -37,6 +38,7 @@ def handle(self, *args, **options):
             title=options['title'],
             description=options['description'],
             urlconf=options['urlconf'],
+            version=options['api_version'],
         )
         schema = generator.get_schema(request=None, public=True)
         renderer = self.get_renderer(options['format'])