import graphene
from graphene import relay
from graphene_django import DjangoObjectType
from graphene_django.filter import DjangoFilterConnectionField

from rooms.models import Room


class RoomNode(DjangoObjectType):
    pk = graphene.Int()

    class Meta:
        model = Room
        filter_fields = ['slug', 'user_group', 'appearance']
        interfaces = (relay.Node,)

    def resolve_pk(self, *args, **kwargs):
        return self.id


class RoomsQuery(object):
    room = relay.Node.Field(RoomNode)
    rooms = DjangoFilterConnectionField(RoomNode)