lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Reformat using black

Browse source
This commit is contained in:
cuom1999 2022-05-14 12:57:27 -05:00
parent efee4ad081
commit a87fb49918
221 changed files with 19127 additions and 7310 deletions
Download patch file Download diff file Expand all files Collapse all files

312
judge/views/ticket.py
View file

@ -4,8 +4,18 @@ from itertools import chain
from django import forms
from django.contrib.auth.mixins import LoginRequiredMixin
from django.core.exceptions import ImproperlyConfigured, PermissionDenied, ValidationError
from django.http import Http404, HttpResponse, HttpResponseBadRequest, HttpResponseRedirect, JsonResponse
from django.core.exceptions import (
ImproperlyConfigured,
PermissionDenied,
ValidationError,
)
from django.http import (
Http404,
HttpResponse,
HttpResponseBadRequest,
HttpResponseRedirect,
JsonResponse,
)
from django.shortcuts import get_object_or_404
from django.template.defaultfilters import truncatechars
from django.template.loader import get_template
@ -26,90 +36,107 @@ from judge.utils.views import SingleObjectFormView, TitleMixin, paginate_query_c
from judge.views.problem import ProblemMixin
from judge.widgets import HeavyPreviewPageDownWidget
ticket_widget = (forms.Textarea() if HeavyPreviewPageDownWidget is None else
HeavyPreviewPageDownWidget(preview=reverse_lazy('ticket_preview'),
preview_timeout=1000, hide_preview_button=True))
ticket_widget = (
forms.Textarea()
if HeavyPreviewPageDownWidget is None
else HeavyPreviewPageDownWidget(
preview=reverse_lazy("ticket_preview"),
preview_timeout=1000,
hide_preview_button=True,
)
)
def add_ticket_notifications(users, author, link, ticket):
html = f"<a href=\"{link}\">{ticket.linked_item}</a>"
html = f'<a href="{link}">{ticket.linked_item}</a>'
users = set(users)
if author in users:
users.remove(author)
for user in users:
notification = Notification(owner=user,
html_link=html,
category='Ticket',
author=author)
notification = Notification(
owner=user, html_link=html, category="Ticket", author=author
)
notification.save()
class TicketForm(forms.Form):
title = forms.CharField(max_length=100, label=gettext_lazy('Ticket title'))
title = forms.CharField(max_length=100, label=gettext_lazy("Ticket title"))
body = forms.CharField(widget=ticket_widget)
def __init__(self, request, *args, **kwargs):
self.request = request
super(TicketForm, self).__init__(*args, **kwargs)
self.fields['title'].widget.attrs.update({'placeholder': _('Ticket title')})
self.fields['body'].widget.attrs.update({'placeholder': _('Issue description')})
self.fields["title"].widget.attrs.update({"placeholder": _("Ticket title")})
self.fields["body"].widget.attrs.update({"placeholder": _("Issue description")})
def clean(self):
if self.request is not None and self.request.user.is_authenticated:
profile = self.request.profile
if profile.mute:
raise ValidationError(_('Your part is silent, little toad.'))
raise ValidationError(_("Your part is silent, little toad."))
return super(TicketForm, self).clean()
class NewTicketView(LoginRequiredMixin, SingleObjectFormView):
form_class = TicketForm
template_name = 'ticket/new.html'
template_name = "ticket/new.html"
def get_assignees(self):
return []
def get_form_kwargs(self):
kwargs = super(NewTicketView, self).get_form_kwargs()
kwargs['request'] = self.request
kwargs["request"] = self.request
return kwargs
def form_valid(self, form):
ticket = Ticket(user=self.request.profile, title=form.cleaned_data['title'])
ticket = Ticket(user=self.request.profile, title=form.cleaned_data["title"])
ticket.linked_item = self.object
ticket.save()
message = TicketMessage(ticket=ticket, user=ticket.user, body=form.cleaned_data['body'])
message = TicketMessage(
ticket=ticket, user=ticket.user, body=form.cleaned_data["body"]
)
message.save()
ticket.assignees.set(self.get_assignees())
link = reverse('ticket', args=[ticket.id])
link = reverse("ticket", args=[ticket.id])
add_ticket_notifications(ticket.assignees.all(), ticket.user, link, ticket)
if event.real:
event.post('tickets', {
'type': 'new-ticket', 'id': ticket.id,
'message': message.id, 'user': ticket.user_id,
'assignees': list(ticket.assignees.values_list('id', flat=True)),
})
event.post(
"tickets",
{
"type": "new-ticket",
"id": ticket.id,
"message": message.id,
"user": ticket.user_id,
"assignees": list(ticket.assignees.values_list("id", flat=True)),
},
)
return HttpResponseRedirect(link)
class NewProblemTicketView(ProblemMixin, TitleMixin, NewTicketView):
template_name = 'ticket/new_problem.html'
template_name = "ticket/new_problem.html"
def get_assignees(self):
return self.object.authors.all()
def get_title(self):
return _('New ticket for %s') % self.object.name
return _("New ticket for %s") % self.object.name
def get_content_title(self):
return mark_safe(escape(_('New ticket for %s')) %
format_html('<a href="{0}">{1}</a>', reverse('problem_detail', args=[self.object.code]),
self.object.translated_name(self.request.LANGUAGE_CODE)))
return mark_safe(
escape(_("New ticket for %s"))
% format_html(
'<a href="{0}">{1}</a>',
reverse("problem_detail", args=[self.object.code]),
self.object.translated_name(self.request.LANGUAGE_CODE),
)
)
def form_valid(self, form):
if not self.object.is_accessible_by(self.request.user):
@ -127,7 +154,7 @@ class TicketMixin(object):
def get_object(self, queryset=None):
ticket = super(TicketMixin, self).get_object(queryset)
profile_id = self.request.profile.id
if self.request.user.has_perm('judge.change_ticket'):
if self.request.user.has_perm("judge.change_ticket"):
return ticket
if ticket.user_id == profile_id:
return ticket
@ -141,39 +168,55 @@ class TicketMixin(object):
class TicketView(TitleMixin, LoginRequiredMixin, TicketMixin, SingleObjectFormView):
form_class = TicketCommentForm
template_name = 'ticket/ticket.html'
context_object_name = 'ticket'
template_name = "ticket/ticket.html"
context_object_name = "ticket"
def form_valid(self, form):
message = TicketMessage(user=self.request.profile,
body=form.cleaned_data['body'],
ticket=self.object)
message = TicketMessage(
user=self.request.profile,
body=form.cleaned_data["body"],
ticket=self.object,
)
message.save()
link = '%s#message-%d' % (reverse('ticket', args=[self.object.id]), message.id)
link = "%s#message-%d" % (reverse("ticket", args=[self.object.id]), message.id)
notify_list = list(chain(self.object.assignees.all(), [self.object.user]))
add_ticket_notifications(notify_list, message.user, link, self.object)
if event.real:
event.post('tickets', {
'type': 'ticket-message', 'id': self.object.id,
'message': message.id, 'user': self.object.user_id,
'assignees': list(self.object.assignees.values_list('id', flat=True)),
})
event.post('ticket-%d' % self.object.id, {
'type': 'ticket-message', 'message': message.id,
})
event.post(
"tickets",
{
"type": "ticket-message",
"id": self.object.id,
"message": message.id,
"user": self.object.user_id,
"assignees": list(
self.object.assignees.values_list("id", flat=True)
),
},
)
event.post(
"ticket-%d" % self.object.id,
{
"type": "ticket-message",
"message": message.id,
},
)
return HttpResponseRedirect(link)
def get_title(self):
return _('%(title)s - Ticket %(id)d') % {'title': self.object.title, 'id': self.object.id}
return _("%(title)s - Ticket %(id)d") % {
"title": self.object.title,
"id": self.object.id,
}
def get_context_data(self, **kwargs):
context = super(TicketView, self).get_context_data(**kwargs)
context['ticket_messages'] = self.object.messages.select_related('user__user')
context['assignees'] = self.object.assignees.select_related('user')
context['last_msg'] = event.last()
context["ticket_messages"] = self.object.messages.select_related("user__user")
context["assignees"] = self.object.assignees.select_related("user")
context["last_msg"] = event.last()
return context
@ -182,21 +225,32 @@ class TicketStatusChangeView(LoginRequiredMixin, TicketMixin, SingleObjectMixin,
def post(self, request, *args, **kwargs):
if self.open is None:
raise ImproperlyConfigured('Need to define open')
raise ImproperlyConfigured("Need to define open")
ticket = self.get_object()
if ticket.is_open != self.open:
ticket.is_open = self.open
ticket.save()
if event.real:
event.post('tickets', {
'type': 'ticket-status', 'id': ticket.id,
'open': self.open, 'user': ticket.user_id,
'assignees': list(ticket.assignees.values_list('id', flat=True)),
'title': ticket.title,
})
event.post('ticket-%d' % ticket.id, {
'type': 'ticket-status', 'open': self.open,
})
event.post(
"tickets",
{
"type": "ticket-status",
"id": ticket.id,
"open": self.open,
"user": ticket.user_id,
"assignees": list(
ticket.assignees.values_list("id", flat=True)
),
"title": ticket.title,
},
)
event.post(
"ticket-%d" % ticket.id,
{
"type": "ticket-status",
"open": self.open,
},
)
return HttpResponse(status=204)
@ -205,16 +259,16 @@ class TicketNotesForm(forms.Form):
class TicketNotesEditView(LoginRequiredMixin, TicketMixin, SingleObjectFormView):
template_name = 'ticket/edit-notes.html'
template_name = "ticket/edit-notes.html"
form_class = TicketNotesForm
context_object_name = 'ticket'
context_object_name = "ticket"
def get_initial(self):
return {'notes': self.get_object().notes}
return {"notes": self.get_object().notes}
def form_valid(self, form):
ticket = self.get_object()
ticket.notes = notes = form.cleaned_data['notes']
ticket.notes = notes = form.cleaned_data["notes"]
ticket.save()
if notes:
return HttpResponse(linebreaks(notes, autoescape=True))
@ -227,8 +281,8 @@ class TicketNotesEditView(LoginRequiredMixin, TicketMixin, SingleObjectFormView)
class TicketList(LoginRequiredMixin, ListView):
model = Ticket
template_name = 'ticket/list.html'
context_object_name = 'tickets'
template_name = "ticket/list.html"
context_object_name = "tickets"
paginate_by = 50
paginator_class = DiggPaginator
@ -242,32 +296,38 @@ class TicketList(LoginRequiredMixin, ListView):
@cached_property
def can_edit_all(self):
return self.request.user.has_perm('judge.change_ticket')
return self.request.user.has_perm("judge.change_ticket")
@cached_property
def filter_users(self):
return self.request.GET.getlist('user')
return self.request.GET.getlist("user")
@cached_property
def filter_assignees(self):
return self.request.GET.getlist('assignee')
return self.request.GET.getlist("assignee")
def GET_with_session(self, key):
if not self.request.GET:
return self.request.session.get(key, False)
return self.request.GET.get(key, None) == '1'
return self.request.GET.get(key, None) == "1"
def _get_queryset(self):
return Ticket.objects.select_related('user__user').prefetch_related('assignees__user').order_by('-id')
return (
Ticket.objects.select_related("user__user")
.prefetch_related("assignees__user")
.order_by("-id")
)
def get_queryset(self):
queryset = self._get_queryset()
if self.GET_with_session('own'):
if self.GET_with_session("own"):
queryset = queryset.filter(own_ticket_filter(self.profile.id))
elif not self.can_edit_all:
queryset = filter_visible_tickets(queryset, self.user, self.profile)
if self.filter_assignees:
queryset = queryset.filter(assignees__user__username__in=self.filter_assignees)
queryset = queryset.filter(
assignees__user__username__in=self.filter_assignees
)
if self.filter_users:
queryset = queryset.filter(user__user__username__in=self.filter_users)
return queryset.distinct()
@ -275,29 +335,41 @@ class TicketList(LoginRequiredMixin, ListView):
def get_context_data(self, **kwargs):
context = super(TicketList, self).get_context_data(**kwargs)
page = context['page_obj']
context['title'] = _('Tickets - Page %(number)d of %(total)d') % {
'number': page.number,
'total': page.paginator.num_pages,
page = context["page_obj"]
context["title"] = _("Tickets - Page %(number)d of %(total)d") % {
"number": page.number,
"total": page.paginator.num_pages,
}
context['can_edit_all'] = self.can_edit_all
context['filter_status'] = {
'own': self.GET_with_session('own'), 'user': self.filter_users, 'assignee': self.filter_assignees,
'user_id': json.dumps(list(Profile.objects.filter(user__username__in=self.filter_users)
.values_list('id', flat=True))),
'assignee_id': json.dumps(list(Profile.objects.filter(user__username__in=self.filter_assignees)
.values_list('id', flat=True))),
'own_id': self.profile.id if self.GET_with_session('own') else 'null',
context["can_edit_all"] = self.can_edit_all
context["filter_status"] = {
"own": self.GET_with_session("own"),
"user": self.filter_users,
"assignee": self.filter_assignees,
"user_id": json.dumps(
list(
Profile.objects.filter(
user__username__in=self.filter_users
).values_list("id", flat=True)
)
),
"assignee_id": json.dumps(
list(
Profile.objects.filter(
user__username__in=self.filter_assignees
).values_list("id", flat=True)
)
),
"own_id": self.profile.id if self.GET_with_session("own") else "null",
}
context['last_msg'] = event.last()
context["last_msg"] = event.last()
context.update(paginate_query_context(self.request))
return context
def post(self, request, *args, **kwargs):
to_update = ('own',)
to_update = ("own",)
for key in to_update:
if key in request.GET:
val = request.GET.get(key) == '1'
val = request.GET.get(key) == "1"
request.session[key] = val
else:
request.session.pop(key, None)
@ -306,38 +378,56 @@ class TicketList(LoginRequiredMixin, ListView):
class ProblemTicketListView(TicketList):
def _get_queryset(self):
problem = get_object_or_404(Problem, code=self.kwargs.get('problem'))
problem = get_object_or_404(Problem, code=self.kwargs.get("problem"))
if problem.is_editable_by(self.request.user):
return problem.tickets.order_by('-id')
return problem.tickets.order_by("-id")
elif problem.is_accessible_by(self.request.user):
return problem.tickets.filter(own_ticket_filter(self.profile.id)).order_by('-id')
return problem.tickets.filter(own_ticket_filter(self.profile.id)).order_by(
"-id"
)
raise Http404()
class TicketListDataAjax(TicketMixin, SingleObjectMixin, View):
def get(self, request, *args, **kwargs):
try:
self.kwargs['pk'] = request.GET['id']
self.kwargs["pk"] = request.GET["id"]
except KeyError:
return HttpResponseBadRequest()
ticket = self.get_object()
message = ticket.messages.first()
return JsonResponse({
'row': get_template('ticket/row.html').render({'ticket': ticket}, request),
'notification': {
'title': _('New Ticket: %s') % ticket.title,
'body': '%s\n%s' % (_('#%(id)d, assigned to: %(users)s') % {
'id': ticket.id,
'users': (_(', ').join(ticket.assignees.values_list('user__username', flat=True)) or _('no one')),
}, truncatechars(message.body, 200)),
},
})
return JsonResponse(
{
"row": get_template("ticket/row.html").render(
{"ticket": ticket}, request
),
"notification": {
"title": _("New Ticket: %s") % ticket.title,
"body": "%s\n%s"
% (
_("#%(id)d, assigned to: %(users)s")
% {
"id": ticket.id,
"users": (
_(", ").join(
ticket.assignees.values_list(
"user__username", flat=True
)
)
or _("no one")
),
},
truncatechars(message.body, 200),
),
},
}
)
class TicketMessageDataAjax(TicketMixin, SingleObjectMixin, View):
def get(self, request, *args, **kwargs):
try:
message_id = request.GET['message']
message_id = request.GET["message"]
except KeyError:
return HttpResponseBadRequest()
ticket = self.get_object()
@ -345,10 +435,14 @@ class TicketMessageDataAjax(TicketMixin, SingleObjectMixin, View):
message = ticket.messages.get(id=message_id)
except TicketMessage.DoesNotExist:
return HttpResponseBadRequest()
return JsonResponse({
'message': get_template('ticket/message.html').render({'message': message}, request),
'notification': {
'title': _('New Ticket Message For: %s') % ticket.title,
'body': truncatechars(message.body, 200),
},
})
return JsonResponse(
{
"message": get_template("ticket/message.html").render(
{"message": message}, request
),
"notification": {
"title": _("New Ticket Message For: %s") % ticket.title,
"body": truncatechars(message.body, 200),
},
}
)