Limit rooms for regular user to their classes

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')