from django.shortcuts import render

from rest_framework import viewsets

from .models import BaseCustomField, CustomTable, Territory, TerritoryItem
from .serializers import (
    BaseCustomFieldSerializer,
    CustomTableSerializer,
    TerritoryItemSerializer,
    TerritorySerializer,
)


class BaseCustomFieldViewSet(viewsets.ModelViewSet):
    queryset = BaseCustomField.objects.all()
    serializer_class = BaseCustomFieldSerializer


class CustomTableViewSet(viewsets.ModelViewSet):
    queryset = CustomTable.objects.all()
    serializer_class = CustomTableSerializer


class TerritoryItemViewSet(viewsets.ModelViewSet):
    page_size_query_param = 'size'
    queryset = TerritoryItem.objects.all()
    serializer_class = TerritoryItemSerializer


class TerritoryViewSet(viewsets.ModelViewSet):
    queryset = Territory.objects.all()
    serializer_class = TerritorySerializer