from rest_framework import serializers

from core.apps.tasks.models.task import Task
from core.apps.accounts.serializers.user import ShortUserSerializer
from core.apps.tasks.serializers.label import LabelSerializer


class TaskSerializer(serializers.ModelSerializer):
    labels = LabelSerializer(many=True)

    class Meta:
        model = Task
        fields = [
            'id',
            'column',
            'name',
            'description',
            'priority',
            'from_date',
            'to_date',
            'labels',
            'assignees',
            'created_by'
        ]

    def get_assignees(self, obj):
        return ShortUserSerializer(obj.assignees.all(), many=True, context={"request": self.context['request']})

    def get_created_by(self, obj):
        return ShortUserSerializer(obj.created_by, context={"request": self.context['request']})