1 # -*- coding: utf-8 -*-
2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 from django import forms
6 from django.utils.translation import ugettext_lazy as _, get_language
7 from migdal.models import Entry
8 from migdal import app_settings
9 from slughifi import slughifi
10 from django.core.mail import mail_managers
11 from django import template
14 def get_submit_form(*args, **kwargs):
17 class SubmitForm(forms.ModelForm):
20 fields = ['title_%s' % lang, 'lead_%s' % lang,
21 'author', 'author_email', 'categories']
22 required = ['title_%s' % lang]
24 def __init__(self, *args, **kwargs):
25 super(SubmitForm, self).__init__(*args, **kwargs)
26 title = self.fields['title_%s' % lang]
28 title.label = _('Title')
29 lead = self.fields['lead_%s' % lang]
31 lead.label = _('Content')
34 data = super(SubmitForm, self).clean()
35 data['type'] = app_settings.TYPE_SUBMIT
36 orig_slug = slughifi(data.get('title_%s' % lang, ''))
39 while Entry.objects.filter(**{'slug_%s' % lang: slug}).exists():
40 slug = "%s-%s" % (orig_slug, number)
42 data['slug_%s' % lang] = slug
43 self.cleaned_data = data
46 def save(self, *args, **kwargs):
47 entry = super(SubmitForm, self).save(commit=False)
48 # Something's wrong with markup fields, they choke on None here.
49 for f in 'lead_en', 'lead_pl', 'body_en', 'body_pl':
50 if getattr(entry, f) is None:
52 for f in 'slug_%s' % lang, 'type':
53 setattr(entry, f, self.cleaned_data[f])
55 entry = super(SubmitForm, self).save(*args, **kwargs)
56 mail_managers(u"Nowy wpis",
57 template.loader.get_template(
58 'migdal/mail/manager_new_entry.txt').render(
59 template.Context({'object': entry})))
61 return SubmitForm(*args, **kwargs)