import graphene
from graphene import InputObjectType

from books.schema.inputs import ContentElementInput
from user.inputs import UserGroupInput


class RoomInput(InputObjectType):
    title = graphene.String()
    description = graphene.String()
    user_group = UserGroupInput()
    appearance = graphene.String()


class AddRoomArgument(RoomInput):
    pass


class UpdateRoomArgument(RoomInput):
    id = graphene.ID(required=True)


class AddRoomEntryArgument(InputObjectType):
    title = graphene.String(required=True)
    subtitle = graphene.String()
    contents = graphene.List(ContentElementInput)
    room = graphene.ID(required=True)