Source code for oioioi.mailsubmit.forms

from django import forms
from django.forms import ValidationError
from django.utils.translation import gettext_lazy as _

from oioioi.contests.forms import SubmissionForm
from oioioi.contests.models import ProblemInstance
from oioioi.contests.utils import visible_problem_instances
from oioioi.default_settings import MAILSUBMIT_CONFIRMATION_HASH_LENGTH
from oioioi.mailsubmit.models import MailSubmission
from oioioi.mailsubmit.utils import is_mailsubmit_allowed, mail_submission_hashes


class MailSubmissionForm(SubmissionForm):
    problem_instance_id = forms.ChoiceField(label=_("Problem"))

    def __init__(self, request, *args, **kwargs):
        self.kind = 'MAILSUBMIT'
        super(MailSubmissionForm, self).__init__(
            request, *args, add_kind_and_user_fields=False, **kwargs
        )

    def clean(self):
        cleaned_data = super(MailSubmissionForm, self).clean(
            check_submission_limit=False, check_round_times=False
        )

        decision = is_mailsubmit_allowed(self.request)
        if not decision:
            raise ValidationError(getattr(decision, 'exc', _("Permission denied")))
        return cleaned_data

    def get_problem_instances(self):
        return visible_problem_instances(self.request)


class AcceptMailSubmissionForm(forms.Form):
    mailsubmission_id = forms.IntegerField(label=_("ID"))
    submission_hash = forms.RegexField(
        label=_("Confirmation code"),
        regex=r'^[0-9a-fA-F]*$',
        min_length=MAILSUBMIT_CONFIRMATION_HASH_LENGTH,
        max_length=MAILSUBMIT_CONFIRMATION_HASH_LENGTH,
        error_messages={
            'invalid': _("Value must consist of numbers or letters from A to F")
        },
    )

    def __init__(self, request, *args, **kwargs):
        super(AcceptMailSubmissionForm, self).__init__(*args, **kwargs)
        self.request = request

    def clean(self):
        cleaned_data = super(AcceptMailSubmissionForm, self).clean()
        if (
            'mailsubmission_id' not in cleaned_data
            or 'submission_hash' not in cleaned_data
        ):
            return cleaned_data

        mailsubmission_id = cleaned_data['mailsubmission_id']
        submission_hash = cleaned_data['submission_hash'].lower()

        try:
            mailsubmission = MailSubmission.objects.get(id=mailsubmission_id)
        except MailSubmission.DoesNotExist:
            raise ValidationError(
                _("Postal submission number %s does not exist")
                % (mailsubmission_id,)
            )

        if mailsubmission.problem_instance.contest != self.request.contest:
            raise ValidationError(
                _("Postal submission number %s is for a different contest")
                % (mailsubmission_id,)
            )

        _source_hash, mailsubmission_hash = mail_submission_hashes(mailsubmission)
        if submission_hash != mailsubmission_hash:
            raise ValidationError(_("Invalid confirmation code"))

        cleaned_data['mailsubmission'] = mailsubmission

        return cleaned_data