115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
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 u' \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)
|