from rest_framework import generics
from rest_framework.response import Response
from rest_framework import permissions

from core.apps.evaluation.serializers.auto.AvgCost import AvgCostSerializer
from core.services.grpc.auto import get_auto_avg_cost


class AvgCostView(generics.GenericAPIView):
    serializer_class = AvgCostSerializer
    permission_classes = [permissions.IsAuthenticated]

    def post(self):
        serializer = self.get_serializer(data=self.request.data)
        if serializer.is_valid():
            avg_cost = get_auto_avg_cost(serializer.validated_data)
            return Response(avg_cost, status=200)
        return Response(serializer.errors, status=400)