Source code for oioioi.contests.processors

from django.conf import settings
from django.utils.functional import lazy

from oioioi.base.utils import request_cached
from oioioi.contests.models import Contest, ContestView
from oioioi.contests.utils import visible_contests




def register_current_contest(request):
    """A template context processor which makes the current contest available
    to the templates.

    The current :class:`~oioioi.contests.models.Contest` instance is added
    to the template context as a ``contest`` variable.

    Must be used together with
    :class:`~oioioi.contests.middleware.CurrentContestMiddleware`.
    """
    if hasattr(request, "contest"):
        return {"contest": request.contest}
    else:
        return {}





@request_cached
def recent_contests(request):
    if request.real_user.is_anonymous:
        ids = request.session.get("recent_contests", [])
        mapping = Contest.objects.in_bulk(ids)
        return [c for c in (mapping.get(id) for id in ids) if c is not None and c != request.contest]
    else:
        c_views = ContestView.objects.filter(user=request.real_user).select_related("contest")
        c_views = c_views[: getattr(settings, "NUM_RECENT_CONTESTS", 5)]
        return [cv.contest for cv in c_views if cv.contest in visible_contests(request)]





def register_recent_contests(request):
    if not hasattr(request, "contest") or not hasattr(request, "session"):
        return {}

    def generator():
        return recent_contests(request)

    return {"recent_contests": lazy(generator, list)()}