graphene/graphene/types/inputobjecttype.py

import six

from graphql import GraphQLInputObjectType

from .definitions import FieldsMeta, ClassTypeMeta, GrapheneGraphQLType
from .proxy import TypeProxy


class GrapheneInputObjectType(GrapheneGraphQLType, GraphQLInputObjectType):
    pass


class InputObjectTypeMeta(FieldsMeta, ClassTypeMeta):

    def get_options(cls, meta):
        return cls.options_class(
            meta,
            name=None,
            description=None,
            graphql_type=None,
            abstract=False
        )

    def construct(cls, bases, attrs):
        if not cls._meta.abstract:
            local_fields = cls._extract_local_fields(attrs)
            if not cls._meta.graphql_type:
                cls._meta.graphql_type = GrapheneInputObjectType(
                    graphene_type=cls,
                    name=cls._meta.name or cls.__name__,
                    description=cls._meta.description or cls.__doc__,
                    fields=cls._fields(bases, attrs, local_fields),
                )
            else:
                assert not local_fields, "Can't mount Fields in an InputObjectType with a defined graphql_type"
        return super(InputObjectTypeMeta, cls).construct(bases, attrs)


class InputObjectType(six.with_metaclass(InputObjectTypeMeta, TypeProxy)):
    class Meta:
        abstract = True