models.py

from django.db import models
from django.contrib.auth.models import User
from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _
from django.utils import timezone
from django.core import validators
# from . import myfields


# 5MB - 5242880
__MAX_UPLOAD_SIZE = 5242880


def validate_deadline(date):
    if date <= timezone.now():
        raise ValidationError(_('Date must be greater than now'), code='invalid')


# def validate_file_size(file):
#     if file._size > __MAX_UPLOAD_SIZE:
#         raise ValidationError(_('Please keep filesize under' + __MAX_UPLOAD_SIZE))

class Task(models.Model):
    title = models.CharField(max_length=150)
    date = models.DateTimeField(auto_now_add=True, editable=False)
    deadline = models.DateTimeField(validators=[validate_deadline])
    text = models.TextField()
    created_by = models.ForeignKey(User, on_delete=models.DO_NOTHING)
    files = models.FileField(
        validators=[validators.FileExtensionValidator(
            'image/png',
            'image/jpeg',
            'application/zip',
        )],
        blank=True,
    )
    # files = myfields.RestrictedFileField(
    #     content_types=['image/png', 'image/jpeg', 'application/zip'],
    #     max_upload_size=MAX_UPLOAD_SIZE,
    #     blank=True,
    #     null=True,
    # )
#    solution_file = models.BooleanField()
#
#    def clean(self):
#        if self.deadline <= timezone.now():
#            raise ValidationError(_('Invalid date'), code='invalid')


class Solution(models.Model):
    task = models.ForeignKey(Task, related_name='task_solution', on_delete=models.CASCADE)
    # student = models.ForeignKey(account.models.Profile, related_name='student_solution',  on_delete=models.CASCADE)
    date = models.DateTimeField(auto_now_add=True, editable=False)
    ready = models.BooleanField(default=False)  # if(Soulution.date <= Task.deadline)
    accepted = models.BooleanField(default=False)
    files = models.FileField(
        validators=[validators.FileExtensionValidator(
            'image/png',
            'image/jpeg',
            'application/zip',
        )],
        blank=True,
    )
    created_by = models.ForeignKey(User, on_delete=models.CASCADE)
    # files = myfields.RestrictedFileField(
    #     content_types=['image/png', 'image/jpeg', 'application/zip'],
    #     max_upload_size=MAX_UPLOAD_SIZE,
    #     blank=True,
    # )