from django.shortcuts import render

from rest_framework import viewsets, filters

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):
    page_size_query_param = "size"
    queryset = CustomTable.objects.all()
    serializer_class = CustomTableSerializer


class TerritoryItemViewSet(viewsets.ModelViewSet):
    page_size_query_param = "size"
    queryset = TerritoryItem.objects.all()
    serializer_class = TerritoryItemSerializer
    
    filter_backends = [filters.SearchFilter]
    search_fields = ["name"]


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