Source code for oioioi.oisubmit.admin

from django.template.loader import render_to_string
from django.utils.translation import gettext_lazy as _

from oioioi.contests.admin import SubmissionAdmin
from oioioi.oisubmit.err_dict import SUSPICION_REASONS


class OISubmitSubmissionAdminMixin(object):
    """Adds oisubmit specific information (e.g. suspected submissions) to
    an admin panel.
    """

    def __init__(self, *args, **kwargs):
        super(OISubmitSubmissionAdminMixin, self).__init__(*args, **kwargs)

    def get_list_display(self, request):
        return super(OISubmitSubmissionAdminMixin, self).get_list_display(request) + [
            'received_suspected'
        ]

    def get_list_filter(self, request):
        return super(OISubmitSubmissionAdminMixin, self).get_list_filter(request) + [
            'oisubmitextradata__received_suspected'
        ]

    def received_suspected(self, instance):
        received_suspected = getattr(
            instance.oisubmitextradata, 'received_suspected', None
        )
        is_oisubmit = received_suspected is not None
        comments = getattr(instance.oisubmitextradata, 'comments', '')
        comments = [SUSPICION_REASONS[c] for c in comments.split(',') if c]
        comments = '<br />'.join(map(str, comments))
        return render_to_string(
            'received-suspected.html',
            {
                'received_suspected': received_suspected,
                'is_oisubmit': is_oisubmit,
                'comments': comments,
            },
        )

    received_suspected.short_description = _("Received suspected")

    def get_custom_list_select_related(self):
        return super(
            OISubmitSubmissionAdminMixin, self
        ).get_custom_list_select_related() + ['oisubmitextradata']


SubmissionAdmin.mix_in(OISubmitSubmissionAdminMixin)