Fix room querying

This commit is contained in:
Ramon Wenger 2018-09-17 17:53:14 +02:00
parent 5cb71cd877
commit 1927244f26
1 changed files with 13 additions and 2 deletions

View File

@ -39,12 +39,23 @@ class RoomNode(DjangoObjectType):
class RoomsQuery(object):
room = relay.Node.Field(RoomNode)
# room = relay.Node.Field(RoomNode)
room_entry = relay.Node.Field(RoomEntryNode)
room = graphene.Field(RoomNode, slug=graphene.String(), id=graphene.Int(), appearance=graphene.String())
all_rooms = DjangoFilterConnectionField(RoomNode)
all_room_entries = DjangoFilterConnectionField(RoomEntryNode)
def resolve_room(self, info, **kwargs):
slug = kwargs.get('slug')
room_id = kwargs.get('id')
if room_id is not None:
return Room.objects.get(pk=room_id)
if slug is not None:
return Room.objects.get(slug=slug)
return None
class RoomMutation(SerializerMutation):
class Meta:
@ -52,4 +63,4 @@ class RoomMutation(SerializerMutation):
class RoomMutations:
room = RoomMutation.Field()
add_room = RoomMutation.Field()