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