Fix crash with empty generator class name

axnsan12 · Dec 28, 2018 · 2bc9add · 2bc9add
1 parent 6df2362
commit 2bc9add
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/drf_yasg/management/commands/generate_swagger.py b/src/drf_yasg/management/commands/generate_swagger.py
@@ -96,7 +96,10 @@ def get_mock_request(self, url, format, user=None):
         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,