from django.shortcuts import render

# Create your views here.
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework.viewsets import ModelViewSet

from workreport.models import DailyReport
from workreport.serializer import DailyReportSerializer


class DailyReportViews(APIView):
    def get(self, request):
        data = DailyReport.objects.values('title', 'content', 'create_at', 'create_by', 'id')
        return Response(data=data, status=status.HTTP_200_OK)

    def post(self, request):
        data = request.data
        report = DailyReport(title=data['title'], content=data['content'], create_by_id=data['create_by_id'])
        report.save()
        return Response(status=status.HTTP_201_CREATED)


class ReportViews(ModelViewSet):
    queryset = DailyReport.objects.all()
    serializer_class = DailyReportSerializer