Fixed a bug that was causing graphql_schmea command to not import properly

Author: BossGrand

graphene_django/management/commands/graphql_schema.py

@@ -57,8 +57,15 @@ def save_file(self, out, schema_dict):
 
     def handle(self, *args, **options):
         options_schema = options.get('schema')
-        if options_schema:
-            schema = importlib.import_module(options_schema)
+
+        if options_schema and type(options_schema) is str:
+            module_str, schema_name = options_schema.rsplit('.', 1)
+            mod = importlib.import_module(module_str)
+            schema = getattr(mod, schema_name)
+
+        elif options_schema:
+            schema = options_schema
+
         else:
             schema = graphene_settings.SCHEMA