mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-29 13:03:44 +03:00
5ce4553244
Without explicitly setting the encoding to "utf-8" I get the following error on windows (python 3.9)
```
File "D:\env\lib\site-packages\graphene_django\management\commands\graphql_schema.py", line 115, in handle
self.get_schema(schema, out, indent)
File "D:\env\lib\site-packages\graphene_django\management\commands\graphql_schema.py", line 72, in get_schema
self.save_graphql_file(out, schema)
File "D:\env\lib\site-packages\graphene_django\management\commands\graphql_schema.py", line 59, in save_graphql_file
outfile.write(print_schema(schema.graphql_schema))
File "C:\Users\u\AppData\Local\Programs\Python\Python39\lib\encodings\cp1252.py", line 19, in encode
return codecs.charmap_encode(input,self.errors,encoding_table)[0]
```
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
import os
|
|
import importlib
|
|
import json
|
|
import functools
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils import autoreload
|
|
|
|
from graphql import print_schema
|
|
from graphene_django.settings import graphene_settings
|
|
|
|
|
|
class CommandArguments(BaseCommand):
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--schema",
|
|
type=str,
|
|
dest="schema",
|
|
default=graphene_settings.SCHEMA,
|
|
help="Django app containing schema to dump, e.g. myproject.core.schema.schema",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--out",
|
|
type=str,
|
|
dest="out",
|
|
default=graphene_settings.SCHEMA_OUTPUT,
|
|
help="Output file, --out=- prints to stdout (default: schema.json)",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--indent",
|
|
type=int,
|
|
dest="indent",
|
|
default=graphene_settings.SCHEMA_INDENT,
|
|
help="Output file indent (default: None)",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--watch",
|
|
dest="watch",
|
|
default=False,
|
|
action="store_true",
|
|
help="Updates the schema on file changes (default: False)",
|
|
)
|
|
|
|
|
|
class Command(CommandArguments):
|
|
help = "Dump Graphene schema as a JSON or GraphQL file"
|
|
can_import_settings = True
|
|
requires_system_checks = False
|
|
|
|
def save_json_file(self, out, schema_dict, indent):
|
|
with open(out, "w") as outfile:
|
|
json.dump(schema_dict, outfile, indent=indent, sort_keys=True)
|
|
|
|
def save_graphql_file(self, out, schema):
|
|
with open(out, "w", encoding="utf-8") as outfile:
|
|
outfile.write(print_schema(schema.graphql_schema))
|
|
|
|
def get_schema(self, schema, out, indent):
|
|
schema_dict = {"data": schema.introspect()}
|
|
if out == "-" or out == "-.json":
|
|
self.stdout.write(json.dumps(schema_dict, indent=indent, sort_keys=True))
|
|
elif out == "-.graphql":
|
|
self.stdout.write(print_schema(schema))
|
|
else:
|
|
# Determine format
|
|
_, file_extension = os.path.splitext(out)
|
|
|
|
if file_extension == ".graphql":
|
|
self.save_graphql_file(out, schema)
|
|
elif file_extension == ".json":
|
|
self.save_json_file(out, schema_dict, indent)
|
|
else:
|
|
raise CommandError(
|
|
'Unrecognised file format "{}"'.format(file_extension)
|
|
)
|
|
|
|
style = getattr(self, "style", None)
|
|
success = getattr(style, "SUCCESS", lambda x: x)
|
|
|
|
self.stdout.write(
|
|
success("Successfully dumped GraphQL schema to {}".format(out))
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
options_schema = options.get("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
|
|
|
|
out = options.get("out") or graphene_settings.SCHEMA_OUTPUT
|
|
|
|
if not schema:
|
|
raise CommandError(
|
|
"Specify schema on GRAPHENE.SCHEMA setting or by using --schema"
|
|
)
|
|
|
|
indent = options.get("indent")
|
|
watch = options.get("watch")
|
|
if watch:
|
|
autoreload.run_with_reloader(
|
|
functools.partial(self.get_schema, schema, out, indent)
|
|
)
|
|
else:
|
|
self.get_schema(schema, out, indent)
|