from rest_framework import viewsets
from apps.hotel.models import Hotel, TypeRoom, Room
from apps.hotel.serializer import HotelSerializer, TypeRoomSerializer, RoomSerializer
from rest_framework.decorators import action

from rest_framework.response import Response


class HotelViewSet(viewsets.ModelViewSet):
    queryset = Hotel.objects.all()
    serializer_class = HotelSerializer


class TypeRoomViewSet(viewsets.ModelViewSet):
    queryset = TypeRoom.objects.all()
    serializer_class = TypeRoomSerializer


class RoomViewSet(viewsets.ModelViewSet):
    queryset = Room.objects.all()
    serializer_class = RoomSerializer

    #@action(detail=True)
    #def availableroom(self, request, *kwargs):
    #    availableRoom = Room.objects.filter(active=True)
    #    serializer = RoomSerializer(availableRoom, many=True)

    #    return Response({'data': serializer.data})


class AvailableRoomViewSet(viewsets.ModelViewSet):
    queryset = Room.objects.all().filter(active=True)
    serializer_class = RoomSerializer