Skip to content
Snippets Groups Projects
serializers.py 1.62 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
from common.email import new_homework
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)
        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')
        fields = (
            'task',
            'created_at',
            'updated_at',
            'accepted',
            'files',
            'created_by',
            'corrected',
            'note',
        )

    def validate(self, data):
        if timezone.now() > data['task'].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!")