from rest_framework import viewsets
from main.serializers import *
from main.models import *
from url_filter.integrations.drf import DjangoFilterBackend
from .models import *

# Create your views here.


class BranchViewSet(viewsets.ModelViewSet):

    """
    View set for manipulating Branch objects
    """

    queryset = Branch.objects.all()
    serializer_class = BranchSerializer


class AreaViewSet(viewsets.ModelViewSet):

    """
    View set for manipulating Area objects
    """

    queryset = Area.objects.all()
    serializer_class = AreaSerializer
    filter_backends = [DjangoFilterBackend]
    filter_fields = ["branch"]


class ItemViewSet(viewsets.ModelViewSet):

    """
    View set for manipulating Area objects
    """

    queryset = Item.objects.all()
    serializer_class = ItemSerializer


class InventoryTakingViewSet(viewsets.ModelViewSet):

    """
    View set for manipulating Area objects
    """

    queryset = InventoryTaking.objects.filter(
        begin__lte=date.today(), end__gte=date.today()
    )
    serializer_class = InventoryTakingSerializer


class UserViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer