Skip to content
Snippets Groups Projects
serializers.py 1.96 KiB
Newer Older
Bodor Máté's avatar
Bodor Máté committed
from rest_framework import serializers
from django.utils import timezone
from account.models import Profile
Bodor Máté's avatar
Bodor Máté committed
from . import models
Bodor Máté's avatar
Bodor Máté committed
from common import email
from common.middleware import CurrentUserMiddleware
Bodor Máté's avatar
Bodor Máté committed


class TaskSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Task
        read_only_fields = ('created_by', 'created_at', 'updated_at')
        fields = '__all__'
Bodor Máté's avatar
Bodor Máté committed

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

    def create(self, validated_data):
        emails = Profile.objects.filter(role="Student").exclude(user__email='').values_list('user__email', flat=True)
Bodor Máté's avatar
Bodor Máté committed
        email.new_homework(emails)
        return self.Meta.model.objects.create(**validated_data)

Bodor Máté's avatar
Bodor Máté committed

class SolutionSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Solution
        read_only_fields = ('created_by', 'created_at', 'updated_at', 'ready', 'files')
Bodor Máté's avatar
Bodor Máté committed
            '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.')
        profile = CurrentUserMiddleware.get_current_user_profile()
        if profile.role != 'Staff' and (data['accepted'] or data['corrected'] or data['note'] != ''):
            raise serializers.ValidationError("You don't have permission!")

    def update(self, instance, validated_data):
        if instance.corrected == False and validated_data.get('corrected', instance.corrected) == True:
            email.homework_corrected(instance.created_by.user.email)
        return super().update(instance, validated_data)