mirror of
https://github.com/graphql-python/graphene.git
synced 2024-09-22 03:49:02 +03:00
53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
import graphene
|
|
|
|
from .data import get_character, get_droid, get_hero, get_human
|
|
|
|
|
|
class Episode(graphene.Enum):
|
|
NEWHOPE = 4
|
|
EMPIRE = 5
|
|
JEDI = 6
|
|
|
|
|
|
class Character(graphene.Interface):
|
|
id = graphene.ID()
|
|
name = graphene.String()
|
|
friends = graphene.List(lambda: Character)
|
|
appears_in = graphene.List(Episode)
|
|
|
|
def resolve_friends(self, info):
|
|
# The character friends is a list of strings
|
|
return [get_character(f) for f in self.friends]
|
|
|
|
|
|
class Human(graphene.ObjectType):
|
|
class Meta:
|
|
interfaces = (Character,)
|
|
|
|
home_planet = graphene.String()
|
|
|
|
|
|
class Droid(graphene.ObjectType):
|
|
class Meta:
|
|
interfaces = (Character,)
|
|
|
|
primary_function = graphene.String()
|
|
|
|
|
|
class Query(graphene.ObjectType):
|
|
hero = graphene.Field(Character, episode=Episode())
|
|
human = graphene.Field(Human, id=graphene.String())
|
|
droid = graphene.Field(Droid, id=graphene.String())
|
|
|
|
def resolve_hero(self, info, episode=None):
|
|
return get_hero(episode)
|
|
|
|
def resolve_human(self, info, id):
|
|
return get_human(id)
|
|
|
|
def resolve_droid(self, info, id):
|
|
return get_droid(id)
|
|
|
|
|
|
schema = graphene.Schema(query=Query)
|