Simplified code

This commit is contained in:
Syrus Akbary 2016-06-06 20:38:55 -07:00
parent ec67303594
commit 4c8f5367ba
6 changed files with 7 additions and 25 deletions

View File

@ -61,26 +61,8 @@ class ClassTypeMeta(type):
class GrapheneGraphQLType(object): class GrapheneGraphQLType(object):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.graphene_type = kwargs.pop('graphene_type') self.graphene_type = kwargs.pop('graphene_type')
self._name = None
self._description = None
super(GrapheneGraphQLType, self).__init__(*args, **kwargs) super(GrapheneGraphQLType, self).__init__(*args, **kwargs)
@property
def name(self):
return self._name or self.graphene_type.__name__
@name.setter
def name(self, name):
self._name = name
@property
def description(self):
return self._description or self.graphene_type.__doc__
@description.setter
def description(self, description):
self._description = description
class GrapheneFieldsType(GrapheneGraphQLType): class GrapheneFieldsType(GrapheneGraphQLType):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):

View File

@ -15,8 +15,8 @@ class GrapheneEnumType(GrapheneGraphQLType, GraphQLEnumType):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
graphene_type = kwargs.pop('graphene_type') graphene_type = kwargs.pop('graphene_type')
self.graphene_type = graphene_type self.graphene_type = graphene_type
self._name = None self.name = None
self._description = None self.description = None
self._values = None self._values = None
self._value_lookup = None self._value_lookup = None
self._name_lookup = None self._name_lookup = None
@ -47,7 +47,7 @@ class EnumTypeMeta(ClassTypeMeta):
cls._meta.graphql_type = GrapheneEnumType( cls._meta.graphql_type = GrapheneEnumType(
graphene_type=cls, graphene_type=cls,
name=cls._meta.name or cls.__name__, name=cls._meta.name or cls.__name__,
description=cls._meta.description, description=cls._meta.description or cls.__doc__,
) )
def construct(cls, bases, attrs): def construct(cls, bases, attrs):

View File

@ -33,7 +33,7 @@ class InputObjectTypeMeta(ClassTypeMeta):
cls._meta.graphql_type = GrapheneInputObjectType( cls._meta.graphql_type = GrapheneInputObjectType(
graphene_type=cls, graphene_type=cls,
name=cls._meta.name or cls.__name__, name=cls._meta.name or cls.__name__,
description=cls._meta.description, description=cls._meta.description or cls.__doc__,
fields=FieldMap(cls, bases=filter(None, inherited_types)), fields=FieldMap(cls, bases=filter(None, inherited_types)),
) )

View File

@ -29,7 +29,7 @@ class InterfaceTypeMeta(ClassTypeMeta):
cls._meta.graphql_type = GrapheneInterfaceType( cls._meta.graphql_type = GrapheneInterfaceType(
graphene_type=cls, graphene_type=cls,
name=cls._meta.name or cls.__name__, name=cls._meta.name or cls.__name__,
description=cls._meta.description, description=cls._meta.description or cls.__doc__,
fields=FieldMap(cls, bases=filter(None, inherited_types)), fields=FieldMap(cls, bases=filter(None, inherited_types)),
) )

View File

@ -68,7 +68,7 @@ class ObjectTypeMeta(ClassTypeMeta):
cls._meta.graphql_type = GrapheneObjectType( cls._meta.graphql_type = GrapheneObjectType(
graphene_type=cls, graphene_type=cls,
name=cls._meta.name or cls.__name__, name=cls._meta.name or cls.__name__,
description=cls._meta.description, description=cls._meta.description or cls.__doc__,
fields=FieldMap(cls, bases=filter(None, inherited_types)), fields=FieldMap(cls, bases=filter(None, inherited_types)),
interfaces=tuple(cls.get_interfaces()), interfaces=tuple(cls.get_interfaces()),
) )

View File

@ -29,7 +29,7 @@ class ScalarTypeMeta(ClassTypeMeta):
cls._meta.graphql_type = GrapheneScalarType( cls._meta.graphql_type = GrapheneScalarType(
graphene_type=cls, graphene_type=cls,
name=cls._meta.name or cls.__name__, name=cls._meta.name or cls.__name__,
description=cls._meta.description, description=cls._meta.description or cls.__doc__,
serialize=getattr(cls, 'serialize', None), serialize=getattr(cls, 'serialize', None),
parse_value=getattr(cls, 'parse_value', None), parse_value=getattr(cls, 'parse_value', None),