encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-28 00:49:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix generateschema command

Browse Source
This commit is contained in:
Dhaval Mehta 2020-05-26 02:06:44 +05:30
parent 36f423870b
commit 34820a1233
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_framework/management/commands/generateschema.py
View File

@ -28,7 +28,7 @@ class Command(BaseCommand):
parser.add_argument('--urlconf', dest="urlconf", default=None, type=str) 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('--generator_class', dest="generator_class", default=None, type=str)
parser.add_argument('--file', dest="file", 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) parser.add_argument('--tag_objects', dest="tag_objects", default=None, type=str)
def handle(self, *args, **options): def handle(self, *args, **options):
if options['generator_class']: if options['generator_class']:
@ -36,13 +36,17 @@ class Command(BaseCommand):
else: else:
generator_class = self.get_generator_class() generator_class = self.get_generator_class()
tag_objects = ast.literal_eval(options['tag_objects']) if options['tag_objects'] else None kwargs = {}
if issubclass(generator_class, SchemaGenerator):
tag_objects = ast.literal_eval(options['tag_objects']) if options['tag_objects'] else None
kwargs['tag_objects'] = tag_objects
generator = generator_class( generator = generator_class(
url=options['url'], url=options['url'],
title=options['title'], title=options['title'],
description=options['description'], description=options['description'],
urlconf=options['urlconf'], urlconf=options['urlconf'],
tag_objects=tag_objects, **kwargs,
) )
schema = generator.get_schema(request=None, public=True) schema = generator.get_schema(request=None, public=True)
renderer = self.get_renderer(options['format']) renderer = self.get_renderer(options['format'])