forms.py 2.4 KB

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