NDOJ/judge/utils/views.py

from django.shortcuts import render
from django.utils.decorators import method_decorator
from django.views.generic import FormView
from django.views.generic.detail import SingleObjectMixin

from judge.utils.diggpaginator import DiggPaginator
from django.utils.html import mark_safe


def class_view_decorator(function_decorator):
    """Convert a function based decorator into a class based decorator usable
    on class based Views.

    Can't subclass the `View` as it breaks inheritance (super in particular),
    so we monkey-patch instead.
    """

    def simple_decorator(View):
        View.dispatch = method_decorator(function_decorator)(View.dispatch)
        return View

    return simple_decorator


def generic_message(request, title, message, status=None):
    return render(
        request,
        "generic-message.html",
        {
            "message": message,
            "title": title,
        },
        status=status,
    )


def paginate_query_context(request):
    query = request.GET.copy()
    query.setlist("page", [])
    query = query.urlencode()
    if query:
        return {
            "page_prefix": "%s?%s&page=" % (request.path, query),
            "first_page_href": "%s?%s" % (request.path, query),
        }
    else:
        return {
            "page_prefix": "%s?page=" % request.path,
            "first_page_href": request.path,
        }


class TitleMixin(object):
    title = "(untitled)"
    content_title = None

    def get_context_data(self, **kwargs):
        context = super(TitleMixin, self).get_context_data(**kwargs)
        context["title"] = self.get_title()
        content_title = self.get_content_title()
        if content_title is not None:
            context["content_title"] = content_title
        return context

    def get_content_title(self):
        return self.content_title

    def get_title(self):
        return self.title


class DiggPaginatorMixin(object):
    def get_paginator(
        self, queryset, per_page, orphans=0, allow_empty_first_page=True, **kwargs
    ):
        return DiggPaginator(
            queryset,
            per_page,
            body=6,
            padding=2,
            orphans=orphans,
            allow_empty_first_page=allow_empty_first_page,
            **kwargs
        )


class QueryStringSortMixin(object):
    all_sorts = None
    default_sort = None
    default_desc = ()

    def get_default_sort_order(self, request):
        return self.default_sort

    def get(self, request, *args, **kwargs):
        order = request.GET.get("order", "")
        if not (
            (not order.startswith("-") or order.count("-") == 1)
            and (order.lstrip("-") in self.all_sorts)
        ):
            order = self.get_default_sort_order(request)
        self.order = order

        return super(QueryStringSortMixin, self).get(request, *args, **kwargs)

    def get_sort_context(self):
        query = self.request.GET.copy()
        query.setlist("order", [])
        query = query.urlencode()
        sort_prefix = (
            "%s?%s&order=" % (self.request.path, query)
            if query
            else "%s?order=" % self.request.path
        )
        current = self.order.lstrip("-")

        links = {
            key: sort_prefix + ("-" if key in self.default_desc else "") + key
            for key in self.all_sorts
        }
        links[current] = (
            sort_prefix + ("" if self.order.startswith("-") else "-") + current
        )

        order = {key: "" for key in self.all_sorts}
        order[current] = " \u25BE" if self.order.startswith("-") else " \u25B4"
        return {"sort_links": links, "sort_order": order}

    def get_sort_paginate_context(self):
        return paginate_query_context(self.request)


def short_circuit_middleware(view):
    view.short_circuit_middleware = True
    return view


class SingleObjectFormView(SingleObjectMixin, FormView):
    def post(self, request, *args, **kwargs):
        self.object = self.get_object()
        return super().post(request, *args, **kwargs)

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        return super().get(request, *args, **kwargs)