From cae211ac0221e6a621a230f795b011e0d9185f49 Mon Sep 17 00:00:00 2001 From: Dhaval Mehta <20968146+dhaval-mehta@users.noreply.github.com> Date: Tue, 26 May 2020 01:33:21 +0530 Subject: [PATCH] add tag_objects to generateschema command --- rest_framework/management/commands/generateschema.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rest_framework/management/commands/generateschema.py b/rest_framework/management/commands/generateschema.py index 024306b65..110720bcf 100644 --- a/rest_framework/management/commands/generateschema.py +++ b/rest_framework/management/commands/generateschema.py @@ -26,6 +26,7 @@ class Command(BaseCommand): 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('--tag_objects', dest="tag_objects", default=None, type=dict) def handle(self, *args, **options): if options['generator_class']: @@ -37,6 +38,7 @@ class Command(BaseCommand): title=options['title'], description=options['description'], urlconf=options['urlconf'], + tag_objects=options['tag_objects'] ) schema = generator.get_schema(request=None, public=True) renderer = self.get_renderer(options['format'])