forked from CGM_Public/pretix_original
110 lines
3.8 KiB
Python
110 lines
3.8 KiB
Python
import logging
|
|
from collections import OrderedDict
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from django.urls import reverse
|
|
from django.utils.functional import cached_property
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.views import View
|
|
from django.views.generic import TemplateView
|
|
|
|
from pretix.base.models import CachedFile
|
|
from pretix.base.services.shredder import export, shred
|
|
from pretix.base.shredder import ShredError, shred_constraints
|
|
from pretix.base.views.tasks import AsyncAction
|
|
from pretix.control.permissions import EventPermissionRequiredMixin
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ShredderMixin:
|
|
|
|
@cached_property
|
|
def shredders(self):
|
|
return OrderedDict(
|
|
sorted(self.request.event.get_data_shredders().items(), key=lambda s: s[1].verbose_name)
|
|
)
|
|
|
|
|
|
class StartShredView(EventPermissionRequiredMixin, ShredderMixin, TemplateView):
|
|
permission = 'can_change_orders'
|
|
template_name = 'pretixcontrol/shredder/index.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
ctx = super().get_context_data(**kwargs)
|
|
ctx['shredders'] = self.shredders
|
|
ctx['constraints'] = shred_constraints(self.request.event)
|
|
return ctx
|
|
|
|
|
|
class ShredDownloadView(EventPermissionRequiredMixin, ShredderMixin, TemplateView):
|
|
permission = 'can_change_orders'
|
|
template_name = 'pretixcontrol/shredder/download.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
ctx = super().get_context_data(**kwargs)
|
|
ctx['shredders'] = self.shredders
|
|
ctx['file'] = get_object_or_404(CachedFile, pk=kwargs.get("file"))
|
|
return ctx
|
|
|
|
|
|
class ShredExportView(EventPermissionRequiredMixin, ShredderMixin, AsyncAction, View):
|
|
permission = 'can_change_orders'
|
|
task = export
|
|
known_errortypes = ['ShredError']
|
|
|
|
def get_success_message(self, value):
|
|
return None
|
|
|
|
def get_success_url(self, value):
|
|
return reverse('control:event.shredder.download', kwargs={
|
|
'event': self.request.event.slug,
|
|
'organizer': self.request.event.organizer.slug,
|
|
'file': str(value)
|
|
})
|
|
|
|
def get_error_url(self):
|
|
return reverse('control:event.shredder.start', kwargs={
|
|
'event': self.request.event.slug,
|
|
'organizer': self.request.event.organizer.slug
|
|
})
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
constr = shred_constraints(self.request.event)
|
|
if constr:
|
|
return self.error(ShredError(self.get_error_url()))
|
|
|
|
return self.do(self.request.event.id, request.POST.getlist("shredder"))
|
|
|
|
|
|
class ShredDoView(EventPermissionRequiredMixin, ShredderMixin, AsyncAction, View):
|
|
permission = 'can_change_orders'
|
|
task = shred
|
|
known_errortypes = ['ShredError']
|
|
|
|
def get_success_url(self, value):
|
|
return reverse('control:event.shredder.start', kwargs={
|
|
'event': self.request.event.slug,
|
|
'organizer': self.request.event.organizer.slug,
|
|
})
|
|
|
|
def get_success_message(self, value):
|
|
return _('The selected data was deleted successfully.')
|
|
|
|
def get_error_url(self):
|
|
return reverse('control:event.shredder.download', kwargs={
|
|
'event': self.request.event.slug,
|
|
'organizer': self.request.event.organizer.slug,
|
|
'file': self.request.POST.get("file")
|
|
})
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
constr = shred_constraints(self.request.event)
|
|
if constr:
|
|
return self.error(ShredError(self.get_error_url()))
|
|
|
|
if not self.request.user.check_password(request.POST.get("password")):
|
|
return self.error(ShredError(_("The current password you entered was not correct.")))
|
|
|
|
return self.do(self.request.event.id, request.POST.get("file"), request.POST.get("confirm_code"))
|