from rest_framework import generics, parsers
from rest_framework.response import Response

from core.apps.wherehouse.serializers import stock_movmend as serializers
from core.apps.wherehouse.models import StockMovemend, StockMovmendProduct
from core.apps.accounts.permissions.permissions import HasRolePermission


class StockMovemendCreateApiView(generics.GenericAPIView):
    serializer_class = serializers.StockMovmendCreateSerializer
    queryset = StockMovemend.objects.all()
    permission_classes = [HasRolePermission]
    required_permissions = []

    def post(self, request):
        serializer = self.serializer_class(data=request.data)
        if serializer.is_valid(raise_exception=True):
            serializer.save()
            return Response(
                {'success': True, 'message': 'stock movemend created'},
                status=200
            )
        return Response(
            {'success': True, 'error_message': serializer.errors}
        )


class StockMovemendListApiView(generics.GenericAPIView):
    serializer_class = serializers.StockMovemendListSerializer
    queryset = StockMovemend.objects.select_related(
        'wherehouse_to', 'wherehouse_from', 'recipient', 'project_folder', 'project'
    ).prefetch_related('movmend_products')
    permission_classes = [HasRolePermission]
    required_permissions = []

    def get(self, request):
        queryset = self.queryset
        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.serializer_class(page, many=True)
            return self.get_paginated_response(serializer.data)
        serializer = self.serializer_class(queryset, many=True)
        return Response(serializer.data, status=200)