serializers.py

from rest_framework import serializers
from django.utils import timezone
from account.models import Profile
from . import models
from common import email
from common.middleware import CurrentUserMiddleware


class TaskSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Task
        read_only_fields = ('created_by', 'created_at', 'updated_at')
        fields = '__all__'

    def validate(self, data):
        if timezone.now() >= data['deadline']:
            raise serializers.ValidationError(
                'Please, enter appropriate deadline.')
        return data

    def create(self, validated_data):
        profiles = Profile.objects.filter(
            role="Student").exclude(
            user__email='')
        for profile in profiles:
            email.new_homework(profile.user, validated_data.get('deadline'))
        return self.Meta.model.objects.create(**validated_data)


class SolutionSerializer_Student(serializers.ModelSerializer):
    class Meta:
        model = models.Solution
        read_only_fields = (
            'created_by',
            'created_at',
            'updated_at',
            'ready',
            'files')
        fields = (
            'id',
            'task',
            'created_at',
            'updated_at',
            'accepted',
            'files',
            'created_by',
            'corrected',
            'note',
        )

    def validate_task(self, value):
        if timezone.now() > value.deadline:
            raise serializers.ValidationError('You late.')
        return value

    def validate_accepted(self, value):
        raise serializers.ValidationError(
            "You don't have permission to modify accepted!")

    def validate_corrected(self, value):
        raise serializers.ValidationError(
            "You don't have permission to modify corrected!")

    def validate_note(self, value):
        if value != '':
            raise serializers.ValidationError(
                "You don't have permission to create note!")
        return value

    def update(self, instance, validated_data):
        if instance.corrected is not True and validated_data.get(
                'corrected', instance.corrected) is True:
            email.homework_corrected(
                instance.created_by.user,
                instance.task.title,
                validated_data.get('accepted', instance.accepted)
            )
        return super().update(instance, validated_data)

    def create(self, validated_data):
        profile = CurrentUserMiddleware.get_current_user_profile()
        models.Solution.objects.filter(
            created_by=profile,
            task=validated_data['task']).delete()
        return super().create(validated_data)


class SolutionSerializer_Staff(serializers.ModelSerializer):
    class Meta:
        model = models.Solution
        read_only_fields = (
            'created_by',
            'created_at',
            'updated_at',
            'ready',
            'files')
        fields = (
            'id',
            'task',
            'created_at',
            'updated_at',
            'accepted',
            'files',
            'created_by',
            'corrected',
            'note',
        )

    def validate_task(self, value):
        if timezone.now() > value.deadline:
            raise serializers.ValidationError('You late.')
        return value

    def validate_accepted(self, value):
        return value

    def validate_corrected(self, value):
        return value

    def validate_note(self, value):
        return value

    def update(self, instance, validated_data):
        if instance.corrected is not True and validated_data.get(
                'corrected', instance.corrected) is True:
            email.homework_corrected(
                instance.created_by.user,
                instance.task.title,
                validated_data.get('accepted', instance.accepted)
            )
        return super().update(instance, validated_data)

    def create(self, validated_data):
        profile = CurrentUserMiddleware.get_current_user_profile()
        models.Solution.objects.filter(
            created_by=profile,
            task=validated_data['task']).delete()
        return super().create(validated_data)