Cloned DMOJ

judge/views/select2.py

@@ -0,0 +1,138 @@
+from django.db.models import F, Q
+from django.http import Http404, JsonResponse
+from django.shortcuts import get_object_or_404
+from django.utils.encoding import smart_text
+from django.views.generic.list import BaseListView
+
+from judge.jinja2.gravatar import gravatar
+from judge.models import Comment, Contest, Organization, Problem, Profile
+
+
+def _get_user_queryset(term):
+    qs = Profile.objects
+    if term.endswith(' '):
+        qs = qs.filter(user__username=term.strip())
+    else:
+        qs = qs.filter(user__username__icontains=term)
+    return qs
+
+
+class Select2View(BaseListView):
+    paginate_by = 20
+
+    def get(self, request, *args, **kwargs):
+        self.request = request
+        self.term = kwargs.get('term', request.GET.get('term', ''))
+        self.object_list = self.get_queryset()
+        context = self.get_context_data()
+
+        return JsonResponse({
+            'results': [
+                {
+                    'text': smart_text(self.get_name(obj)),
+                    'id': obj.pk,
+                } for obj in context['object_list']],
+            'more': context['page_obj'].has_next(),
+        })
+
+    def get_name(self, obj):
+        return str(obj)
+
+
+class UserSelect2View(Select2View):
+    def get_queryset(self):
+        return _get_user_queryset(self.term).annotate(username=F('user__username')).only('id')
+
+    def get_name(self, obj):
+        return obj.username
+
+
+class OrganizationSelect2View(Select2View):
+    def get_queryset(self):
+        return Organization.objects.filter(name__icontains=self.term)
+
+
+class ProblemSelect2View(Select2View):
+    def get_queryset(self):
+        queryset = Problem.objects.filter(Q(code__icontains=self.term) | Q(name__icontains=self.term))
+        if not self.request.user.has_perm('judge.see_private_problem'):
+            filter = Q(is_public=True)
+            if self.request.user.is_authenticated:
+                filter |= Q(authors=self.request.profile) | Q(curators=self.request.profile)
+            queryset = queryset.filter(filter).distinct()
+        return queryset.distinct()
+
+
+class ContestSelect2View(Select2View):
+    def get_queryset(self):
+        queryset = Contest.objects.filter(Q(key__icontains=self.term) | Q(name__icontains=self.term))
+        if not self.request.user.has_perm('judge.see_private_contest'):
+            queryset = queryset.filter(is_visible=True)
+        if not self.request.user.has_perm('judge.edit_all_contest'):
+            q = Q(is_private=False, is_organization_private=False)
+            if self.request.user.is_authenticated:
+                q |= Q(is_organization_private=True,
+                       organizations__in=self.request.profile.organizations.all())
+                q |= Q(is_private=True, private_contestants=self.request.profile)
+            queryset = queryset.filter(q)
+        return queryset
+
+
+class CommentSelect2View(Select2View):
+    def get_queryset(self):
+        return Comment.objects.filter(page__icontains=self.term)
+
+
+class UserSearchSelect2View(BaseListView):
+    paginate_by = 20
+
+    def get_queryset(self):
+        return _get_user_queryset(self.term)
+
+    def get(self, request, *args, **kwargs):
+        self.request = request
+        self.kwargs = kwargs
+        self.term = kwargs.get('term', request.GET.get('term', ''))
+        self.gravatar_size = request.GET.get('gravatar_size', 128)
+        self.gravatar_default = request.GET.get('gravatar_default', None)
+
+        self.object_list = self.get_queryset().values_list('pk', 'user__username', 'user__email', 'display_rank')
+
+        context = self.get_context_data()
+
+        return JsonResponse({
+            'results': [
+                {
+                    'text': username,
+                    'id': username,
+                    'gravatar_url': gravatar(email, self.gravatar_size, self.gravatar_default),
+                    'display_rank': display_rank,
+                } for pk, username, email, display_rank in context['object_list']],
+            'more': context['page_obj'].has_next(),
+        })
+
+    def get_name(self, obj):
+        return str(obj)
+
+
+class ContestUserSearchSelect2View(UserSearchSelect2View):
+    def get_queryset(self):
+        contest = get_object_or_404(Contest, key=self.kwargs['contest'])
+        if not contest.can_see_scoreboard(self.request.user) or \
+                contest.hide_scoreboard and contest.is_in_contest(self.request.user):
+            raise Http404()
+
+        return Profile.objects.filter(contest_history__contest=contest,
+                                      user__username__icontains=self.term).distinct()
+
+
+class TicketUserSelect2View(UserSearchSelect2View):
+    def get_queryset(self):
+        return Profile.objects.filter(tickets__isnull=False,
+                                      user__username__icontains=self.term).distinct()
+
+
+class AssigneeSelect2View(UserSearchSelect2View):
+    def get_queryset(self):
+        return Profile.objects.filter(assigned_tickets__isnull=False,
+                                      user__username__icontains=self.term).distinct()