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


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