Files
pretix_original/src/pretix/base/forms.py
2015-04-06 00:14:59 +02:00

77 lines
2.8 KiB
Python

from django.forms.models import ModelFormMetaclass, BaseModelForm
from django import forms
from django.utils import six
from pretix.base.i18n import I18nFormField
from versions.models import Versionable
from django.utils.translation import ugettext_lazy as _
class BaseI18nModelForm(BaseModelForm):
"""
This is a helperclass to construct I18nModelForm
"""
def __init__(self, *args, **kwargs):
event = kwargs.pop('event', None)
super().__init__(*args, **kwargs)
if event:
for k, field in self.fields.items():
if isinstance(field, I18nFormField):
field.widget.enabled_langcodes = event.settings.get('locales')
class VersionedBaseModelForm(BaseI18nModelForm):
"""
This is a helperclass to construct VersionedModelForm
"""
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()
return super().save(commit)
class VersionedModelForm(six.with_metaclass(ModelFormMetaclass, VersionedBaseModelForm)):
"""
This is a modified version of I18nModelForm which differs from I18nModelForm in
only one way: It executes the .clone() method of an object before saving it back to
the database, if the model is a sub-class of versions.models.Versionable. You can
safely use this as a base class for all your model forms, it will work out correctly
with both versioned and non-versioned models.
"""
pass
class I18nModelForm(six.with_metaclass(ModelFormMetaclass, BaseI18nModelForm)):
"""
This is a modified version of Django's ModelForm which differs from ModelForm in
only one way: The constructor takes one additional optional argument ``event``
which may be given an :pyclass:`pretix.base.models.Event` instance. If given, this
instance is used to select the visible languages in all I18nFormFields of the form. If
not given, all languages will be displayed.
"""
pass
class SettingsForm(forms.Form):
"""
This form is meant to be used for modifying Event- or OrganizerSettings
"""
BOOL_CHOICES = (
('True', _('enabled')),
('False', _('disabled')),
)
def __init__(self, *args, **kwargs):
self.obj = kwargs.pop('obj')
if 'initial' not in kwargs:
kwargs['initial'] = self.obj.settings
super().__init__(*args, **kwargs)
def save(self):
for name, field in self.fields.items():
value = self.cleaned_data[name]
if value is None:
del self.obj.settings[name]
elif self.obj.settings.get(name, as_type=type(value)) != value:
self.obj.settings.set(name, value)