forms.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from django import forms
  2. from django.conf import settings
  3. from django.contrib.flatpages.models import FlatPage
  4. from django.utils.translation import ugettext, ugettext_lazy as _
  5. class FlatpageForm(forms.ModelForm):
  6. url = forms.RegexField(label=_("URL"), max_length=100, regex=r'^[-\w/\.~]+$',
  7. help_text=_("Example: '/about/contact/'. Make sure to have leading"
  8. " and trailing slashes."),
  9. error_message=_("This value must contain only letters, numbers,"
  10. " dots, underscores, dashes, slashes or tildes."))
  11. class Meta:
  12. model = FlatPage
  13. fields = '__all__'
  14. def clean_url(self):
  15. url = self.cleaned_data['url']
  16. if not url.startswith('/'):
  17. raise forms.ValidationError(
  18. ugettext("URL is missing a leading slash."),
  19. code='missing_leading_slash',
  20. )
  21. if (settings.APPEND_SLASH and
  22. 'django.middleware.common.CommonMiddleware' in settings.MIDDLEWARE_CLASSES and
  23. not url.endswith('/')):
  24. raise forms.ValidationError(
  25. ugettext("URL is missing a trailing slash."),
  26. code='missing_trailing_slash',
  27. )
  28. return url
  29. def clean(self):
  30. url = self.cleaned_data.get('url', None)
  31. sites = self.cleaned_data.get('sites', None)
  32. same_url = FlatPage.objects.filter(url=url)
  33. if self.instance.pk:
  34. same_url = same_url.exclude(pk=self.instance.pk)
  35. if sites and same_url.filter(sites__in=sites).exists():
  36. for site in sites:
  37. if same_url.filter(sites=site).exists():
  38. raise forms.ValidationError(
  39. _('Flatpage with url %(url)s already exists for site %(site)s'),
  40. code='duplicate_url',
  41. params={'url': url, 'site': site},
  42. )
  43. return super(FlatpageForm, self).clean()