kopia lustrzana https://github.com/rtts/django-simplecms
131 wiersze
4.2 KiB
Python
131 wiersze
4.2 KiB
Python
from django import forms
|
|
from django.conf import settings
|
|
from django.db.models import Prefetch
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core.mail import EmailMessage
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from . import registry
|
|
|
|
class PageForm(forms.ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.label_suffix = ''
|
|
|
|
self.formsets = [forms.inlineformset_factory(
|
|
parent_model=registry.page_class,
|
|
model=registry.section_class,
|
|
form=SectionForm,
|
|
extra=1,
|
|
)(
|
|
data=self.data if self.is_bound else None,
|
|
files=self.files if self.is_bound else None,
|
|
instance=self.instance,
|
|
)]
|
|
self.formsets[0][0].empty_permitted = True
|
|
|
|
def is_valid(self):
|
|
return super().is_valid() and self.formsets[0].is_valid()
|
|
|
|
def has_changed(self):
|
|
return super().has_changed() or self.formsets[0].has_changed()
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
if not self.formsets[0].is_valid():
|
|
self.add_error(None, repr(self.formsets[0].errors))
|
|
|
|
def save(self, *args, **kwargs):
|
|
page = super().save()
|
|
formset = self.formsets[0]
|
|
formset.instance = page
|
|
formset.save()
|
|
return page
|
|
|
|
class Meta:
|
|
model = registry.page_class
|
|
fields = '__all__'
|
|
|
|
class SectionForm(forms.ModelForm):
|
|
type = forms.ChoiceField()
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.label_suffix = ''
|
|
self.fields['DELETE'] = forms.BooleanField(label=_('Delete'), required=False)
|
|
self.fields['type'].choices = registry.get_types()
|
|
self.fields['type'].initial = registry.get_types()[0][0]
|
|
|
|
# Populate the 'formsets' attribute if the Section was
|
|
# extendend with one_to_many fields
|
|
self.formsets = []
|
|
for field in self.instance._meta.get_fields():
|
|
if field.one_to_many:
|
|
formset = forms.inlineformset_factory(
|
|
parent_model=registry.section_class,
|
|
model=field.related_model,
|
|
fields='__all__',
|
|
extra=1,
|
|
)(
|
|
instance=self.instance,
|
|
data=self.data if self.is_bound else None,
|
|
files=self.files if self.is_bound else None,
|
|
prefix=f'{self.prefix}-{field.name}',
|
|
form_kwargs={'label_suffix': self.label_suffix},
|
|
)
|
|
formset.name = field.name
|
|
self.formsets.append(formset)
|
|
|
|
def is_valid(self):
|
|
result = super().is_valid()
|
|
for formset in self.formsets:
|
|
result = result and formset.is_valid() # AND
|
|
return result
|
|
|
|
def has_changed(self):
|
|
result = super().has_changed()
|
|
for formset in self.formsets:
|
|
result = result or formset.has_changed() # OR
|
|
return result
|
|
|
|
def save(self, commit=True):
|
|
section = super().save(commit=commit)
|
|
|
|
if self.cleaned_data['DELETE']:
|
|
section.delete()
|
|
if section.page.slug and not section.page.sections.exists():
|
|
section.page.delete()
|
|
return
|
|
elif commit:
|
|
section.save()
|
|
|
|
for formset in self.formsets:
|
|
formset.save(commit=commit)
|
|
|
|
return section
|
|
|
|
class Meta:
|
|
model = registry.section_class
|
|
exclude = ['page']
|
|
|
|
class ContactForm(forms.Form):
|
|
sender = forms.EmailField(label=_('Your email address'))
|
|
spam_protection = forms.CharField(label=_('Your message'), widget=forms.Textarea())
|
|
message = forms.CharField(label=_('Your message'), widget=forms.Textarea(), initial='Hi there!')
|
|
|
|
def save(self):
|
|
body = self.cleaned_data.get('spam_protection')
|
|
if len(body.split()) < 7:
|
|
return
|
|
spamcheck = self.cleaned_data.get('message')
|
|
if spamcheck != 'Hi there!':
|
|
return
|
|
|
|
email = EmailMessage(
|
|
to = [settings.DEFAULT_TO_EMAIL],
|
|
body = body,
|
|
subject = _('Contact form'),
|
|
headers = {'Reply-To': self.cleaned_data.get('sender')},
|
|
)
|
|
email.send()
|