A Python GraphQL JSON type for graphqlql-core-next v1.1.1

graphql_json_type.py

"""GraphQL JSON type"""

import json
from typing import List, Mapping, Union, Any

from graphql.type import GraphQLScalarType
from graphql.language.ast import StringValueNode
from graphql.error import INVALID

JSONType = Union[List[Mapping[str, Any]], Mapping[str, Any]]

def _serialize_json(value: Any) -> JSONType:
    if isinstance(value, (dict, list)):
        return value
    elif isinstance(value, str):
        return json.loads(value)
    else:
        raise TypeError('Unable to serialize value')

def _coerce_json(value: Any) -> JSONType:
    if isinstance(value, (dict, list)):
        return value
    raise TypeError('Unable to coerce type')

def _parse_json_literal(ast, _variables=None):
    if isinstance(ast, StringValueNode):
        result = json.loads(ast.value)
        return result
    return INVALID

# pylint: disable=invalid-name
GraphQLJson = GraphQLScalarType(
    name="JSON",
    description="The `JSON` scalar type",
    serialize=_serialize_json,
    parse_value=_coerce_json,
    parse_literal=_parse_json_literal,
)