diff --git a/server/rooms/schema.py b/server/rooms/schema.py index 437dd393..77a581a9 100644 --- a/server/rooms/schema.py +++ b/server/rooms/schema.py @@ -47,6 +47,12 @@ class RoomsQuery(object): rooms = DjangoFilterConnectionField(RoomNode) all_room_entries = DjangoFilterConnectionField(RoomEntryNode) + def resolve_rooms(self, info, **kwargs): + user = info.context.user + if user.is_superuser: + return Room.objects.all() + return Room.objects.filter(school_class__in=user.school_classes.all()) + def resolve_room(self, info, **kwargs): slug = kwargs.get('slug') room_id = kwargs.get('id')