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


class ExecutiveViewSet(viewsets.ModelViewSet):
    queryset = Executive.objects.all()
    serializer_class = ExecutiveSerializer


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