fix tag_objects parsing in generateschema.py

This commit is contained in:
Dhaval Mehta 2020-05-26 01:51:40 +05:30
parent b29812302f
commit 084b4d5351

View File

@ -1,3 +1,5 @@
import ast
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from django.utils.module_loading import import_string from django.utils.module_loading import import_string
@ -33,12 +35,14 @@ class Command(BaseCommand):
generator_class = import_string(options['generator_class']) generator_class = import_string(options['generator_class'])
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
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=options['tag_objects'] tag_objects=tag_objects,
) )
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'])