Fix crash with empty generator class name

2018-12-28 09:34:54 +02:00 · 2018-12-28 09:34:54 +02:00 · 2bc9addc99
parent 6df2362156
commit 2bc9addc99
1 changed files with 4 additions and 1 deletions
--- a/src/drf_yasg/management/commands/generate_swagger.py
+++ b/src/drf_yasg/management/commands/generate_swagger.py
@ -96,7 +96,10 @@ class Command(BaseCommand):
        return request

    def get_schema_generator(self, generator_class_name, api_info, api_version, api_url):
-        generator_class = import_string(generator_class_name) or swagger_settings.DEFAULT_GENERATOR_CLASS
+        generator_class = swagger_settings.DEFAULT_GENERATOR_CLASS
+        if generator_class_name:
+            generator_class = import_string(generator_class_name)
+
        return generator_class(
            info=api_info,
            version=api_version,