mirror of
https://github.com/pretix/pretix.git
synced 2026-05-06 15:24:02 +00:00
900 lines
32 KiB
Python
900 lines
32 KiB
Python
from itertools import product
|
|
from django.db import transaction
|
|
from django.forms import BooleanField, ModelForm
|
|
from django.utils.functional import cached_property
|
|
|
|
from django.views.generic import ListView
|
|
from django.views.generic.edit import CreateView, UpdateView, DeleteView
|
|
from django.views.generic.base import TemplateView
|
|
from django.views.generic.detail import SingleObjectMixin
|
|
from django.core.urlresolvers import resolve, reverse
|
|
from django.http import HttpResponseRedirect, HttpResponseForbidden
|
|
from django.shortcuts import redirect
|
|
from django.forms.models import inlineformset_factory
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from pretix.base.forms import VersionedModelForm
|
|
|
|
from pretix.base.models import (
|
|
Item, ItemCategory, Property, ItemVariation, PropertyValue, Question, Quota,
|
|
Versionable)
|
|
from pretix.control.permissions import EventPermissionRequiredMixin, event_permission_required
|
|
from pretix.control.views.forms import TolerantFormsetModelForm, VariationsField
|
|
from pretix.control.signals import restriction_formset
|
|
|
|
|
|
class ItemList(ListView):
|
|
model = Item
|
|
context_object_name = 'items'
|
|
template_name = 'pretixcontrol/items/index.html'
|
|
|
|
def get_queryset(self):
|
|
return Item.objects.current.filter(
|
|
event=self.request.event
|
|
).prefetch_related("category")
|
|
|
|
|
|
class CategoryForm(VersionedModelForm):
|
|
|
|
class Meta:
|
|
model = ItemCategory
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'name'
|
|
]
|
|
|
|
|
|
class CategoryDelete(EventPermissionRequiredMixin, DeleteView):
|
|
model = ItemCategory
|
|
form_class = CategoryForm
|
|
template_name = 'pretixcontrol/items/category_delete.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'category'
|
|
|
|
def get_object(self, queryset=None) -> ItemCategory:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.categories.current.get(
|
|
identity=url.kwargs['category']
|
|
)
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
for item in self.object.items.current.all():
|
|
# TODO: Clone!?
|
|
item.category = None
|
|
item.save()
|
|
success_url = self.get_success_url()
|
|
self.object.delete()
|
|
return HttpResponseRedirect(success_url)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.categories', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?deleted=true'
|
|
|
|
|
|
class CategoryUpdate(EventPermissionRequiredMixin, UpdateView):
|
|
model = ItemCategory
|
|
form_class = CategoryForm
|
|
template_name = 'pretixcontrol/items/category.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'category'
|
|
|
|
def get_object(self, queryset=None) -> ItemCategory:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.categories.current.get(
|
|
identity=url.kwargs['category']
|
|
)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.categories', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?updated=true'
|
|
|
|
|
|
class CategoryCreate(EventPermissionRequiredMixin, CreateView):
|
|
model = ItemCategory
|
|
form_class = CategoryForm
|
|
template_name = 'pretixcontrol/items/category.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'category'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.categories', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?created=true'
|
|
|
|
def form_valid(self, form):
|
|
form.instance.event = self.request.event
|
|
return super().form_valid(form)
|
|
|
|
|
|
class CategoryList(ListView):
|
|
model = ItemCategory
|
|
context_object_name = 'categories'
|
|
template_name = 'pretixcontrol/items/categories.html'
|
|
|
|
def get_queryset(self):
|
|
return self.request.event.categories.current.all()
|
|
|
|
|
|
def category_move(request, category, up=True):
|
|
"""
|
|
This is a helper function to avoid duplicating code in category_move_up and
|
|
category_move_down. It takes a category and a direction and then tries to bring
|
|
all categories for this event in a new order.
|
|
"""
|
|
category = request.event.categories.current.get(
|
|
identity=category
|
|
)
|
|
categories = list(request.event.categories.current.order_by("position"))
|
|
|
|
index = categories.index(category)
|
|
if index != 0 and up:
|
|
categories[index - 1], categories[index] = categories[index], categories[index - 1]
|
|
elif index != len(categories) - 1 and not up:
|
|
categories[index + 1], categories[index] = categories[index], categories[index + 1]
|
|
|
|
for i, cat in enumerate(categories):
|
|
if cat.position != i:
|
|
cat.position = i
|
|
cat.save() # TODO: Clone or document sloppiness?
|
|
|
|
|
|
@event_permission_required("can_change_items")
|
|
def category_move_up(request, organizer, event, category):
|
|
category_move(request, category, up=True)
|
|
return redirect(reverse('control:event.items.categories', kwargs={
|
|
'organizer': request.event.organizer.slug,
|
|
'event': request.event.slug,
|
|
}) + '?ordered=true')
|
|
|
|
|
|
@event_permission_required("can_change_items")
|
|
def category_move_down(request, organizer, event, category):
|
|
category_move(request, category, up=False)
|
|
return redirect(reverse('control:event.items.categories', kwargs={
|
|
'organizer': request.event.organizer.slug,
|
|
'event': request.event.slug,
|
|
}) + '?ordered=true')
|
|
|
|
|
|
class PropertyList(ListView):
|
|
model = Property
|
|
context_object_name = 'properties'
|
|
template_name = 'pretixcontrol/items/properties.html'
|
|
|
|
def get_queryset(self):
|
|
return Property.objects.current.filter(
|
|
event=self.request.event
|
|
)
|
|
|
|
|
|
class PropertyForm(VersionedModelForm):
|
|
class Meta:
|
|
model = Property
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'name',
|
|
]
|
|
|
|
|
|
class PropertyValueForm(TolerantFormsetModelForm):
|
|
class Meta:
|
|
model = PropertyValue
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'value',
|
|
]
|
|
|
|
|
|
class PropertyUpdate(EventPermissionRequiredMixin, UpdateView):
|
|
model = Property
|
|
form_class = PropertyForm
|
|
template_name = 'pretixcontrol/items/property.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'property'
|
|
|
|
def get_object(self, queryset=None) -> Property:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.properties.current.get(
|
|
identity=url.kwargs['property']
|
|
)
|
|
|
|
def get_success_url(self) -> str:
|
|
url = resolve(self.request.path_info)
|
|
return reverse('control:event.items.properties.edit', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
'property': url.kwargs['property']
|
|
}) + '?success=true'
|
|
|
|
def get_formset(self):
|
|
formsetclass = inlineformset_factory(
|
|
Property, PropertyValue,
|
|
form=PropertyValueForm,
|
|
can_order=True,
|
|
extra=0,
|
|
)
|
|
kwargs = self.get_form_kwargs()
|
|
kwargs['queryset'] = self.object.values.current.all()
|
|
formset = formsetclass(**kwargs)
|
|
return formset
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['formset'] = self.get_formset()
|
|
return context
|
|
|
|
@transaction.atomic()
|
|
def form_valid(self, form, formset):
|
|
for f in formset.deleted_forms:
|
|
f.instance.delete()
|
|
f.instance.pk = None
|
|
|
|
for i, f in enumerate(formset.ordered_forms):
|
|
if f.instance.pk is not None:
|
|
f.instance = f.instance.clone()
|
|
f.instance.position = i
|
|
f.instance.save()
|
|
return super().form_valid(form)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
form_class = self.get_form_class()
|
|
form = self.get_form(form_class)
|
|
formset = self.get_formset()
|
|
if form.is_valid() and formset.is_valid():
|
|
return self.form_valid(form, formset)
|
|
else:
|
|
return self.form_invalid(form)
|
|
|
|
|
|
class PropertyCreate(EventPermissionRequiredMixin, CreateView):
|
|
model = Property
|
|
form_class = PropertyForm
|
|
template_name = 'pretixcontrol/items/property.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'property'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.properties', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?created=true'
|
|
|
|
def get_formset(self):
|
|
formsetclass = inlineformset_factory(
|
|
Property, PropertyValue,
|
|
form=PropertyValueForm,
|
|
can_order=True,
|
|
extra=3,
|
|
)
|
|
formset = formsetclass(**self.get_form_kwargs())
|
|
return formset
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
self.object = None
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['formset'] = self.get_formset()
|
|
return context
|
|
|
|
@transaction.atomic()
|
|
def form_valid(self, form, formset):
|
|
form.instance.event = self.request.event
|
|
resp = super().form_valid(form)
|
|
for i, f in enumerate(formset.ordered_forms):
|
|
f.instance.position = i
|
|
f.instance.prop = form.instance
|
|
f.instance.save()
|
|
return resp
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
form_class = self.get_form_class()
|
|
form = self.get_form(form_class)
|
|
formset = self.get_formset()
|
|
if form.is_valid() and formset.is_valid():
|
|
return self.form_valid(form, formset)
|
|
else:
|
|
return self.form_invalid(form)
|
|
|
|
|
|
class PropertyDelete(EventPermissionRequiredMixin, DeleteView):
|
|
model = Property
|
|
form_class = PropertyForm
|
|
template_name = 'pretixcontrol/items/property_delete.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'property'
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['dependent'] = self.get_object().items.current.all()
|
|
context['possible'] = self.is_allowed()
|
|
return context
|
|
|
|
def is_allowed(self) -> bool:
|
|
return self.get_object().items.current.count() == 0
|
|
|
|
def get_object(self, queryset=None) -> Property:
|
|
if not hasattr(self, 'object') or not self.object:
|
|
url = resolve(self.request.path_info)
|
|
self.object = self.request.event.properties.current.get(
|
|
identity=url.kwargs['property']
|
|
)
|
|
return self.object
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
if self.is_allowed():
|
|
success_url = self.get_success_url()
|
|
self.get_object().delete()
|
|
return HttpResponseRedirect(success_url)
|
|
else:
|
|
return HttpResponseForbidden()
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.properties', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?deleted=true'
|
|
|
|
|
|
class QuestionList(ListView):
|
|
model = Question
|
|
context_object_name = 'questions'
|
|
template_name = 'pretixcontrol/items/questions.html'
|
|
|
|
def get_queryset(self):
|
|
return self.request.event.questions.current.all()
|
|
|
|
|
|
class QuestionForm(VersionedModelForm):
|
|
|
|
class Meta:
|
|
model = Question
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'question',
|
|
'type',
|
|
'required',
|
|
]
|
|
|
|
|
|
class QuestionDelete(EventPermissionRequiredMixin, DeleteView):
|
|
model = Question
|
|
template_name = 'pretixcontrol/items/question_delete.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'question'
|
|
|
|
def get_object(self, queryset=None) -> Question:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.questions.current.get(
|
|
identity=url.kwargs['question']
|
|
)
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['dependent'] = list(self.get_object().items.current.all())
|
|
return context
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
success_url = self.get_success_url()
|
|
self.object.delete()
|
|
return HttpResponseRedirect(success_url)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.questions', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?deleted=true'
|
|
|
|
|
|
class QuestionUpdate(EventPermissionRequiredMixin, UpdateView):
|
|
model = Question
|
|
form_class = QuestionForm
|
|
template_name = 'pretixcontrol/items/question.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'question'
|
|
|
|
def get_object(self, queryset=None) -> Question:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.questions.current.get(
|
|
identity=url.kwargs['question']
|
|
)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.questions', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?updated=true'
|
|
|
|
|
|
class QuestionCreate(EventPermissionRequiredMixin, CreateView):
|
|
model = Question
|
|
form_class = QuestionForm
|
|
template_name = 'pretixcontrol/items/question.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'question'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.questions', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?created=true'
|
|
|
|
def form_valid(self, form):
|
|
form.instance.event = self.request.event
|
|
return super().form_valid(form)
|
|
|
|
|
|
class QuotaList(ListView):
|
|
model = Quota
|
|
context_object_name = 'quotas'
|
|
template_name = 'pretixcontrol/items/quotas.html'
|
|
|
|
def get_queryset(self):
|
|
return Quota.objects.current.filter(
|
|
event=self.request.event
|
|
).prefetch_related("items")
|
|
|
|
|
|
class QuotaForm(ModelForm):
|
|
|
|
def __init__(self, **kwargs):
|
|
items = kwargs['items']
|
|
del kwargs['items']
|
|
super().__init__(**kwargs)
|
|
|
|
if hasattr(self, 'instance'):
|
|
active_items = set(self.instance.items.all())
|
|
active_variations = set(self.instance.variations.all())
|
|
else:
|
|
active_items = set()
|
|
active_variations = set()
|
|
|
|
for item in items:
|
|
if len(item.properties.all()) > 0:
|
|
self.fields['item_%s' % item.identity] = VariationsField(
|
|
item, label=_("Activate for"),
|
|
required=False,
|
|
initial=active_variations
|
|
)
|
|
self.fields['item_%s' % item.identity].set_item(item)
|
|
else:
|
|
self.fields['item_%s' % item.identity] = BooleanField(
|
|
label=_("Activate"),
|
|
required=False,
|
|
initial=(item in active_items)
|
|
)
|
|
|
|
def save(self, commit=True):
|
|
if self.instance.pk is not None and isinstance(self.instance, Versionable):
|
|
if self.has_changed():
|
|
self.instance = self.instance.clone_shallow()
|
|
# TODO: order_cache, lock_cache are emptied by that but you'll have
|
|
# to rebuild them anyway
|
|
return super().save(commit)
|
|
|
|
class Meta:
|
|
model = Quota
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'name',
|
|
'size',
|
|
]
|
|
|
|
|
|
class QuotaEditorMixin:
|
|
|
|
@cached_property
|
|
def items(self) -> "List[Item]":
|
|
return list(self.request.event.items.all().prefetch_related("properties", "variations"))
|
|
|
|
def get_form(self, form_class):
|
|
if not hasattr(self, '_form'):
|
|
kwargs = self.get_form_kwargs()
|
|
kwargs['items'] = self.items
|
|
self._form = form_class(**kwargs)
|
|
return self._form
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['items'] = self.items
|
|
for item in context['items']:
|
|
item.field = self.get_form(QuotaForm)['item_%s' % item.identity]
|
|
return context
|
|
|
|
@transaction.atomic()
|
|
def form_valid(self, form):
|
|
res = super().form_valid(form)
|
|
# The following commented-out checks are not necessary as both self.object.items
|
|
# and self.object.variations can be expected empty due to the performance
|
|
# optimization of pretixbase.models.Versionable.clone_shallow()
|
|
# items = self.object.items.all()
|
|
# variations = self.object.variations.all()
|
|
self.object = form.instance
|
|
for item in self.items:
|
|
field = form.fields['item_%s' % item.identity]
|
|
data = form.cleaned_data['item_%s' % item.identity]
|
|
if isinstance(field, VariationsField):
|
|
self.object.variations.add(*data)
|
|
# for v in data:
|
|
# if v not in variations:
|
|
# self.object.variations.add(v)
|
|
# for v in variations:
|
|
# if v not in data:
|
|
# self.object.variations.remove(v)
|
|
if data: # and item not in items:
|
|
self.object.items.add(item)
|
|
# elif not data and item in items:
|
|
# self.object.items.remove(item)
|
|
return res
|
|
|
|
|
|
class QuotaCreate(EventPermissionRequiredMixin, QuotaEditorMixin, CreateView):
|
|
model = Quota
|
|
form_class = QuotaForm
|
|
template_name = 'pretixcontrol/items/quota.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'quota'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.quotas', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?created=true'
|
|
|
|
def form_valid(self, form):
|
|
form.instance.event = self.request.event
|
|
return super().form_valid(form)
|
|
|
|
|
|
class QuotaUpdate(EventPermissionRequiredMixin, QuotaEditorMixin, UpdateView):
|
|
model = Quota
|
|
form_class = QuotaForm
|
|
template_name = 'pretixcontrol/items/quota.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'quota'
|
|
|
|
def get_object(self, queryset=None) -> Quota:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.quotas.current.get(
|
|
identity=url.kwargs['quota']
|
|
)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.quotas', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?updated=true'
|
|
|
|
|
|
class QuotaDelete(EventPermissionRequiredMixin, DeleteView):
|
|
model = Quota
|
|
template_name = 'pretixcontrol/items/quota_delete.html'
|
|
permission = 'can_change_items'
|
|
context_object_name = 'quota'
|
|
|
|
def get_object(self, queryset=None) -> Quota:
|
|
url = resolve(self.request.path_info)
|
|
return self.request.event.quotas.current.get(
|
|
identity=url.kwargs['quota']
|
|
)
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['dependent'] = list(self.get_object().items.current.all())
|
|
return context
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
success_url = self.get_success_url()
|
|
self.object.delete()
|
|
return HttpResponseRedirect(success_url)
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.items.quotas', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
}) + '?deleted=true'
|
|
|
|
|
|
class ItemDetailMixin(SingleObjectMixin):
|
|
model = Item
|
|
context_object_name = 'item'
|
|
|
|
def get_object(self, queryset=None) -> Item:
|
|
if not hasattr(self, 'object') or not self.object:
|
|
url = resolve(self.request.path_info)
|
|
self.item = self.request.event.items.current.get(
|
|
identity=url.kwargs['item']
|
|
)
|
|
self.object = self.item
|
|
return self.object
|
|
|
|
|
|
class ItemFormGeneral(VersionedModelForm):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields['category'].queryset = self.instance.event.categories.current.all()
|
|
self.fields['properties'].queryset = self.instance.event.properties.current.all()
|
|
self.fields['questions'].queryset = self.instance.event.questions.current.all()
|
|
|
|
class Meta:
|
|
model = Item
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'category',
|
|
'name',
|
|
'active',
|
|
'short_description',
|
|
'long_description',
|
|
'default_price',
|
|
'tax_rate',
|
|
'properties',
|
|
'questions',
|
|
]
|
|
|
|
|
|
class ItemCreate(EventPermissionRequiredMixin, CreateView):
|
|
form_class = ItemFormGeneral
|
|
template_name = 'pretixcontrol/item/index.html'
|
|
permission = 'can_change_items'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.item', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
'item': self.object.identity,
|
|
}) + '?success=true'
|
|
|
|
def get_form_kwargs(self):
|
|
"""
|
|
Returns the keyword arguments for instantiating the form.
|
|
"""
|
|
newinst = Item(event=self.request.event)
|
|
kwargs = super().get_form_kwargs()
|
|
kwargs.update({'instance': newinst})
|
|
return kwargs
|
|
|
|
|
|
class ItemUpdateGeneral(ItemDetailMixin, EventPermissionRequiredMixin, UpdateView):
|
|
form_class = ItemFormGeneral
|
|
template_name = 'pretixcontrol/item/index.html'
|
|
permission = 'can_change_items'
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.item', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
'item': self.get_object().identity,
|
|
}) + '?success=true'
|
|
|
|
|
|
class ItemVariationForm(VersionedModelForm):
|
|
|
|
class Meta:
|
|
model = ItemVariation
|
|
localized_fields = '__all__'
|
|
fields = [
|
|
'active',
|
|
'default_price',
|
|
]
|
|
|
|
|
|
class ItemVariations(ItemDetailMixin, EventPermissionRequiredMixin, TemplateView):
|
|
|
|
permission = 'can_change_items'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.item = None
|
|
|
|
def get_form(self, variation, data=None) -> ItemVariationForm:
|
|
"""
|
|
Return the dict for one given variation. Variations are expected to be
|
|
dictionaries in the format of Item.get_all_variations()
|
|
"""
|
|
# Values are all dictionary ite
|
|
values = [i[1] for i in sorted([it for it in variation.items() if it[0] != 'variation'])]
|
|
if 'variation' in variation:
|
|
form = ItemVariationForm(
|
|
data,
|
|
instance=variation['variation'],
|
|
prefix=",".join([str(i.identity) for i in values]),
|
|
)
|
|
else:
|
|
inst = ItemVariation(item=self.object)
|
|
inst.item_id = self.object.identity
|
|
inst.creation = True
|
|
form = ItemVariationForm(
|
|
data,
|
|
instance=inst,
|
|
prefix=",".join([str(i.identity) for i in values]),
|
|
)
|
|
form.values = values
|
|
return form
|
|
|
|
def get_forms(self) -> tuple:
|
|
"""
|
|
Returns one form per possible item variation. The forms are returned
|
|
twice: The first entry in the returned tuple contains a 1-, 2- or
|
|
3-dimensional list, depending on the number of properties associated
|
|
with this item (this is being used for form display), the second
|
|
contains all forms in one single list (this is used for processing).
|
|
|
|
The first, hierarchical list, is a list of dicts on all levels but the
|
|
last one, where the dict contains the two entries 'row' containing a
|
|
string describing this layer and 'forms' which contains the forms or
|
|
the next list of dicts.
|
|
"""
|
|
forms = []
|
|
forms_flat = []
|
|
variations = self.object.get_all_variations()
|
|
data = self.request.POST if self.request.method == 'POST' else None
|
|
|
|
if self.dimension == 1:
|
|
# For one-dimensional structures we just have a list of forms
|
|
for variation in variations:
|
|
form = self.get_form(variation, data)
|
|
forms.append(form)
|
|
forms_flat = forms
|
|
|
|
elif self.dimension >= 2:
|
|
# For 2 or more dimensional structures we display a list of grids
|
|
# of forms
|
|
|
|
# prop1 is the property on all the grid's y-axes
|
|
prop1 = self.properties[0]
|
|
# prop2 is the property on all the grid's x-axes
|
|
prop2 = self.properties[1]
|
|
|
|
def selector(values):
|
|
# Given an iterable of PropertyValue objects, this will return a
|
|
# list of their primary keys, ordered by the primary keys of the
|
|
# properties they belong to EXCEPT the value for the property prop2.
|
|
# We'll see later why we need this.
|
|
return [
|
|
v.identity for v in sorted(values, key=lambda v: v.prop.identity)
|
|
if v.prop.identity != prop2.identity
|
|
]
|
|
|
|
def sort(v):
|
|
# Given a list of variations, this will sort them by their position
|
|
# on the x-axis
|
|
return v[prop2.identity].sortkey
|
|
|
|
# We now iterate over the cartesian product of all the other
|
|
# properties which are NOT on the axes of the grid because we
|
|
# create one grid for any combination of them.
|
|
for gridrow in product(*[prop.values.current.all() for prop in self.properties[2:]]):
|
|
grids = []
|
|
for val1 in prop1.values.current.all():
|
|
formrow = []
|
|
# We are now inside one of the rows of the grid and have to
|
|
# select the variations to display in this row. In order to
|
|
# achieve this, we use the 'selector' lambda defined above.
|
|
# It gives us a normalized, comparable version of a set of
|
|
# PropertyValue objects. In this case, we compute the
|
|
# selector of our row as the selector of the sum of the
|
|
# values defining our grind and the value defining our row.
|
|
selection = selector(gridrow + (val1,))
|
|
# We now iterate over all variations who generate the same
|
|
# selector as 'selection'.
|
|
filtered = [v for v in variations if selector(v.relevant_values()) == selection]
|
|
for variation in sorted(filtered, key=sort):
|
|
form = self.get_form(variation, data)
|
|
formrow.append(form)
|
|
forms_flat.append(form)
|
|
|
|
grids.append({'row': val1, 'forms': formrow})
|
|
|
|
forms.append({'row': ", ".join([value.value for value in gridrow]), 'forms': grids})
|
|
|
|
return forms, forms_flat
|
|
|
|
def main(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
self.properties = list(self.object.properties.current.all().prefetch_related("values"))
|
|
self.dimension = len(self.properties)
|
|
self.forms, self.forms_flat = self.get_forms()
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.main(request, *args, **kwargs)
|
|
context = self.get_context_data(object=self.object)
|
|
return self.render_to_response(context)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
self.main(request, *args, **kwargs)
|
|
context = self.get_context_data(object=self.object)
|
|
with transaction.atomic():
|
|
for form in self.forms_flat:
|
|
if form.is_valid() and form.has_changed():
|
|
form.save()
|
|
if hasattr(form.instance, 'creation') and form.instance.creation:
|
|
# We need this special 'creation' field set to true in get_form
|
|
# for newly created items as cleanerversion does already set the
|
|
# primary key in its post_init hook
|
|
form.instance.values.add(*form.values)
|
|
# TODO: Redirect to success message
|
|
return self.render_to_response(context)
|
|
|
|
def get_template_names(self) -> "List[str]":
|
|
if self.dimension == 0:
|
|
return ['pretixcontrol/item/variations_0d.html']
|
|
elif self.dimension == 1:
|
|
return ['pretixcontrol/item/variations_1d.html']
|
|
elif self.dimension >= 2:
|
|
return ['pretixcontrol/item/variations_nd.html']
|
|
|
|
def get_context_data(self, **kwargs) -> dict:
|
|
context = super().get_context_data(**kwargs)
|
|
context['forms'] = self.forms
|
|
context['properties'] = self.properties
|
|
return context
|
|
|
|
|
|
class ItemRestrictions(ItemDetailMixin, EventPermissionRequiredMixin, TemplateView):
|
|
|
|
permission = 'can_change_items'
|
|
template_name = 'pretixcontrol/item/restrictions.html'
|
|
|
|
def get_formsets(self):
|
|
responses = restriction_formset.send(self.object.event, item=self.object)
|
|
formsets = []
|
|
for receiver, response in responses:
|
|
response['formset'] = response['formsetclass'](
|
|
self.request.POST if self.request.method == 'POST' else None,
|
|
instance=self.object,
|
|
prefix=response['prefix'],
|
|
)
|
|
formsets.append(response)
|
|
return formsets
|
|
|
|
def main(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
self.request = request
|
|
self.formsets = self.get_formsets()
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.main(request, *args, **kwargs)
|
|
context = self.get_context_data(object=self.object)
|
|
return self.render_to_response(context)
|
|
|
|
@transaction.atomic()
|
|
def post(self, request, *args, **kwargs):
|
|
self.main(request, *args, **kwargs)
|
|
valid = True
|
|
for f in self.formsets:
|
|
valid &= f['formset'].is_valid()
|
|
if valid:
|
|
for f in self.formsets:
|
|
for form in f['formset']:
|
|
if 'DELETE' in form.cleaned_data and form.cleaned_data['DELETE'] is True:
|
|
if form.instance.pk is None:
|
|
continue
|
|
form.instance.delete()
|
|
else:
|
|
form.instance.event = request.event
|
|
form.instance.item = self.object
|
|
form.save()
|
|
return redirect(self.get_success_url())
|
|
else:
|
|
context = self.get_context_data(object=self.object)
|
|
return self.render_to_response(context)
|
|
|
|
def get_context_data(self, *args, **kwargs) -> dict:
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context['formsets'] = self.formsets
|
|
return context
|
|
|
|
def get_success_url(self) -> str:
|
|
return reverse('control:event.item.restrictions', kwargs={
|
|
'organizer': self.request.event.organizer.slug,
|
|
'event': self.request.event.slug,
|
|
'item': self.object.identity
|
|
}) + '?success=true'
|