Limit rooms for regular user to their classes
This commit is contained in:
parent
a7141c3ea0
commit
004ad8ca57
|
|
@ -47,6 +47,12 @@ class RoomsQuery(object):
|
||||||
rooms = DjangoFilterConnectionField(RoomNode)
|
rooms = DjangoFilterConnectionField(RoomNode)
|
||||||
all_room_entries = DjangoFilterConnectionField(RoomEntryNode)
|
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):
|
def resolve_room(self, info, **kwargs):
|
||||||
slug = kwargs.get('slug')
|
slug = kwargs.get('slug')
|
||||||
room_id = kwargs.get('id')
|
room_id = kwargs.get('id')
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue