178 lines
7.4 KiB
Python
178 lines
7.4 KiB
Python
from django.conf import settings
|
|
from django.db.models import Count, Max, Q
|
|
from django.http import Http404
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from django.utils.functional import lazy
|
|
from django.utils.translation import ugettext as _
|
|
from django.views.generic import ListView
|
|
|
|
from judge.comments import CommentedDetailView
|
|
from judge.models import BlogPost, Comment, Contest, Language, Problem, ProblemClarification, Profile, Submission, \
|
|
Ticket
|
|
from judge.utils.cachedict import CacheDict
|
|
from judge.utils.diggpaginator import DiggPaginator
|
|
from judge.utils.problems import user_completed_ids
|
|
from judge.utils.tickets import filter_visible_tickets
|
|
from judge.utils.views import TitleMixin
|
|
|
|
|
|
# General view for all content list on home feed
|
|
class FeedView(ListView):
|
|
template_name = 'blog/list.html'
|
|
title = None
|
|
|
|
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)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(FeedView, self).get_context_data(**kwargs)
|
|
context['has_clarifications'] = False
|
|
if self.request.user.is_authenticated:
|
|
participation = self.request.profile.current_contest
|
|
if participation:
|
|
clarifications = ProblemClarification.objects.filter(problem__in=participation.contest.problems.all())
|
|
context['has_clarifications'] = clarifications.count() > 0
|
|
context['clarifications'] = clarifications.order_by('-date')
|
|
if participation.contest.is_editable_by(self.request.user):
|
|
context['can_edit_contest'] = True
|
|
|
|
context['page_titles'] = CacheDict(lambda page: Comment.get_page_title(page))
|
|
|
|
context['user_count'] = lazy(Profile.objects.count, int, int)
|
|
context['problem_count'] = lazy(Problem.objects.filter(is_public=True).count, int, int)
|
|
context['submission_count'] = lazy(Submission.objects.count, int, int)
|
|
context['language_count'] = lazy(Language.objects.count, int, int)
|
|
|
|
now = timezone.now()
|
|
|
|
# Dashboard stuff
|
|
# if self.request.user.is_authenticated:
|
|
# user = self.request.profile
|
|
# context['recently_attempted_problems'] = (Submission.objects.filter(user=user)
|
|
# .exclude(problem__in=user_completed_ids(user))
|
|
# .values_list('problem__code', 'problem__name', 'problem__points')
|
|
# .annotate(points=Max('points'), latest=Max('date'))
|
|
# .order_by('-latest')
|
|
# [:settings.DMOJ_BLOG_RECENTLY_ATTEMPTED_PROBLEMS_COUNT])
|
|
|
|
visible_contests = Contest.get_visible_contests(self.request.user).filter(is_visible=True) \
|
|
.order_by('start_time')
|
|
|
|
context['current_contests'] = visible_contests.filter(start_time__lte=now, end_time__gt=now)
|
|
context['future_contests'] = visible_contests.filter(start_time__gt=now)
|
|
|
|
visible_contests = Contest.get_visible_contests(self.request.user).filter(is_visible=True)
|
|
|
|
context['top_rated'] = Profile.objects.filter(is_unlisted=False).order_by('-rating')[:10]
|
|
context['top_scorer'] = Profile.objects.filter(is_unlisted=False).order_by('-performance_points')[:10]
|
|
print(context['top_rated'])
|
|
return context
|
|
|
|
|
|
class PostList(FeedView):
|
|
model = BlogPost
|
|
paginate_by = 10
|
|
context_object_name = 'posts'
|
|
|
|
def get_queryset(self):
|
|
queryset = BlogPost.objects.filter(visible=True, publish_on__lte=timezone.now()) \
|
|
.order_by('-sticky', '-publish_on') \
|
|
.prefetch_related('authors__user', 'organizations')
|
|
if not self.request.user.has_perm('judge.edit_all_post'):
|
|
filter = Q(is_organization_private=False)
|
|
if self.request.user.is_authenticated:
|
|
filter |= Q(organizations__in=self.request.profile.organizations.all())
|
|
queryset = queryset.filter(filter)
|
|
return queryset
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(PostList, self).get_context_data(**kwargs)
|
|
context['title'] = self.title or _('Page %d of Posts') % context['page_obj'].number
|
|
context['first_page_href'] = reverse('home')
|
|
context['page_prefix'] = reverse('blog_post_list')
|
|
context['feed_type'] = 'blog'
|
|
context['post_comment_counts'] = {
|
|
int(page[2:]): count for page, count in
|
|
Comment.objects
|
|
.filter(page__in=['b:%d' % post.id for post in context['posts']], hidden=False)
|
|
.values_list('page').annotate(count=Count('page')).order_by()
|
|
}
|
|
|
|
return context
|
|
|
|
|
|
class TicketFeed(FeedView):
|
|
model = Ticket
|
|
context_object_name = 'tickets'
|
|
paginate_by = 30
|
|
|
|
def get_queryset(self, is_own=True):
|
|
profile = self.request.profile
|
|
if is_own:
|
|
if self.request.user.is_authenticated:
|
|
return (Ticket.objects.filter(Q(user=profile) | Q(assignees__in=[profile]), is_open=True).order_by('-id')
|
|
.prefetch_related('linked_item').select_related('user__user'))
|
|
else:
|
|
return []
|
|
else:
|
|
# Superusers better be staffs, not the spell-casting kind either.
|
|
if self.request.user.is_staff:
|
|
tickets = (Ticket.objects.order_by('-id').filter(is_open=True).prefetch_related('linked_item')
|
|
.select_related('user__user'))
|
|
return filter_visible_tickets(tickets, self.request.user, profile)
|
|
else:
|
|
return []
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(TicketFeed, self).get_context_data(**kwargs)
|
|
context['feed_type'] = 'ticket'
|
|
context['first_page_href'] = self.request.path
|
|
context['page_prefix'] = '?page='
|
|
context['title'] = _('Ticket feed')
|
|
|
|
return context
|
|
|
|
|
|
class CommentFeed(FeedView):
|
|
model = Comment
|
|
context_object_name = 'comments'
|
|
paginate_by = 50
|
|
|
|
def get_queryset(self):
|
|
return Comment.most_recent(self.request.user, 1000)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(CommentFeed, self).get_context_data(**kwargs)
|
|
context['feed_type'] = 'comment'
|
|
context['first_page_href'] = self.request.path
|
|
context['page_prefix'] = '?page='
|
|
context['title'] = _('Comment feed')
|
|
|
|
return context
|
|
|
|
|
|
class PostView(TitleMixin, CommentedDetailView):
|
|
model = BlogPost
|
|
pk_url_kwarg = 'id'
|
|
context_object_name = 'post'
|
|
template_name = 'blog/blog.html'
|
|
|
|
def get_title(self):
|
|
return self.object.title
|
|
|
|
def get_comment_page(self):
|
|
return 'b:%s' % self.object.id
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(PostView, self).get_context_data(**kwargs)
|
|
context['og_image'] = self.object.og_image
|
|
return context
|
|
|
|
def get_object(self, queryset=None):
|
|
post = super(PostView, self).get_object(queryset)
|
|
if not post.can_see(self.request.user):
|
|
raise Http404()
|
|
return post
|