from rest_framework import viewsets
from rest_framework.response import Response

from picampaign.feedback.serializers import FeedbackSerializer
from picampaign.feedback.models import Feedback, FeedbackCategory
from picampaign.campaign.models import CampaignContact


class FeedbackViewSet(viewsets.ViewSet):

    queryset = Feedback.objects.all()
    serializer_class = FeedbackSerializer

    def create(self, request, campaign_pk=None):
        serializer = FeedbackSerializer(request.DATA)
        callee = CampaignContact.objects.get(id=request.DATA['callee'])
        category = FeedbackCategory.objects.get(id=request.DATA['category'])
        serializer.data['callee'] = callee
        serializer.data['category'] = category
        feedback = Feedback.objects.create(**serializer.data)
        return Response(feedback.id)