mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-11 20:27:03 +03:00
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
import graphene
|
|
from graphene import resolve_only_args, relay
|
|
|
|
from .data import (
|
|
getFaction,
|
|
getShip,
|
|
getRebels,
|
|
getEmpire,
|
|
)
|
|
|
|
class Ship(relay.Node):
|
|
'''A ship in the Star Wars saga'''
|
|
name = graphene.StringField(description='The name of the ship.')
|
|
|
|
@classmethod
|
|
def get_node(cls, id):
|
|
ship = getShip(id)
|
|
if ship:
|
|
return Ship(ship)
|
|
|
|
|
|
class Faction(relay.Node):
|
|
'''A faction in the Star Wars saga'''
|
|
name = graphene.StringField(description='The name of the faction.')
|
|
ships = relay.ConnectionField(Ship, description='The ships used by the faction.')
|
|
|
|
@resolve_only_args
|
|
def resolve_ships(self, **kwargs):
|
|
return [Ship(getShip(ship)) for ship in self.instance.ships]
|
|
|
|
@classmethod
|
|
def get_node(cls, id):
|
|
faction = getFaction(id)
|
|
if faction:
|
|
return Faction(faction)
|
|
|
|
|
|
class Query(graphene.ObjectType):
|
|
rebels = graphene.Field(Faction)
|
|
empire = graphene.Field(Faction)
|
|
node = relay.NodeField()
|
|
|
|
@resolve_only_args
|
|
def resolve_rebels(self):
|
|
return Faction(getRebels())
|
|
|
|
@resolve_only_args
|
|
def resolve_empire(self):
|
|
return Faction(getEmpire())
|
|
|
|
|
|
schema = graphene.Schema(query=Query, name='Starwars Relay Schema')
|