mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-11 12:16:58 +03:00
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
from graphql.core.type import GraphQLEnumValue
|
|
import graphene
|
|
from graphene import resolve_only_args
|
|
|
|
from .data import getHero, getHuman, getCharacter, getDroid, Human as _Human, Droid as _Droid
|
|
|
|
Episode = graphene.Enum('Episode', dict(
|
|
NEWHOPE=GraphQLEnumValue(4),
|
|
EMPIRE=GraphQLEnumValue(5),
|
|
JEDI=GraphQLEnumValue(6)
|
|
))
|
|
|
|
|
|
def wrap_character(character):
|
|
if isinstance(character, _Human):
|
|
return Human(character)
|
|
elif isinstance(character, _Droid):
|
|
return Droid(character)
|
|
|
|
|
|
class Character(graphene.Interface):
|
|
id = graphene.IDField()
|
|
name = graphene.StringField()
|
|
friends = graphene.ListField('self')
|
|
appearsIn = graphene.ListField(Episode)
|
|
|
|
def resolve_friends(self, args, *_):
|
|
return [wrap_character(getCharacter(f)) for f in self.instance.friends]
|
|
|
|
|
|
class Human(Character):
|
|
homePlanet = graphene.StringField()
|
|
|
|
|
|
class Droid(Character):
|
|
primaryFunction = graphene.StringField()
|
|
|
|
|
|
class Query(graphene.ObjectType):
|
|
hero = graphene.Field(Character,
|
|
episode=graphene.Argument(Episode)
|
|
)
|
|
human = graphene.Field(Human,
|
|
id=graphene.Argument(graphene.String)
|
|
)
|
|
droid = graphene.Field(Droid,
|
|
id=graphene.Argument(graphene.String)
|
|
)
|
|
|
|
class Meta:
|
|
type_name = 'core_Query'
|
|
|
|
@resolve_only_args
|
|
def resolve_hero(self, episode=None):
|
|
return wrap_character(getHero(episode))
|
|
|
|
@resolve_only_args
|
|
def resolve_human(self, id):
|
|
return wrap_character(getHuman(id))
|
|
|
|
@resolve_only_args
|
|
def resolve_droid(self, id):
|
|
return wrap_character(getDroid(id))
|
|
|
|
|
|
Schema = graphene.Schema(query=Query)
|