mirror of
https://github.com/graphql-python/graphene.git
synced 2024-12-02 06:24:01 +03:00
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import six
|
|
|
|
from graphql import GraphQLInterfaceType
|
|
|
|
from ..utils.copy_fields import copy_fields
|
|
from ..utils.get_fields import get_fields
|
|
from ..utils.is_base_type import is_base_type
|
|
from .definitions import GrapheneGraphQLType
|
|
from .field import Field
|
|
from .options import Options
|
|
|
|
|
|
class GrapheneInterfaceType(GrapheneGraphQLType, GraphQLInterfaceType):
|
|
pass
|
|
|
|
|
|
class InterfaceTypeMeta(type):
|
|
|
|
def __new__(cls, name, bases, attrs):
|
|
super_new = super(InterfaceTypeMeta, cls).__new__
|
|
|
|
# Also ensure initialization is only performed for subclasses of Model
|
|
# (excluding Model class itself).
|
|
if not is_base_type(bases, InterfaceTypeMeta):
|
|
return super_new(cls, name, bases, attrs)
|
|
|
|
options = Options(
|
|
attrs.pop('Meta', None),
|
|
name=None,
|
|
description=None,
|
|
graphql_type=None,
|
|
abstract=False
|
|
)
|
|
|
|
fields = get_fields(Interface, attrs, bases)
|
|
attrs = attrs_without_fields(attrs, fields)
|
|
cls = super_new(cls, name, bases, dict(attrs, _meta=options))
|
|
|
|
if not options.graphql_type:
|
|
fields = copy_fields(Field, fields, parent=cls)
|
|
options.graphql_type = GrapheneInterfaceType(
|
|
graphene_type=cls,
|
|
name=options.name or cls.__name__,
|
|
description=options.description or cls.__doc__,
|
|
fields=fields,
|
|
)
|
|
else:
|
|
assert not fields, "Can't mount Fields in an Interface with a defined graphql_type"
|
|
fields = copy_fields(options.graphql_type.get_fields(), parent=cls)
|
|
|
|
for name, field in fields.items():
|
|
setattr(cls, field.attname or name, field)
|
|
|
|
return cls
|
|
|
|
|
|
def attrs_without_fields(attrs, fields):
|
|
return {k: v for k, v in attrs.items() if k not in fields}
|
|
|
|
|
|
class Interface(six.with_metaclass(InterfaceTypeMeta)):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
from .objecttype import ObjectType
|
|
if not isinstance(self, ObjectType):
|
|
raise Exception("An interface cannot be intitialized")
|
|
super(Interface, self).__init__(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def implements(cls, object_type):
|
|
'''
|
|
We use this function for customizing when a ObjectType have this class as Interface
|
|
For example, if we want to check that the ObjectType have some required things
|
|
in it like Node.get_node
|
|
'''
|