from django import forms from django.conf import settings from django.contrib.flatpages.models import FlatPage from django.core.exceptions import ValidationError from django.utils.translation import gettext from django.utils.translation import gettext_lazy as _ class FlatpageForm(forms.ModelForm): url = forms.RegexField( label=_("URL"), max_length=100, regex=r"^[-\w/\.~]+$", help_text=_( "Example: “/about/contact/”. Make sure to have leading and trailing " "slashes." ), error_messages={ "invalid": _( "This value must contain only letters, numbers, dots, " "underscores, dashes, slashes or tildes." ), }, ) class Meta: model = FlatPage fields = "__all__" def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) if not self._trailing_slash_required(): self.fields["url"].help_text = _( "Example: “/about/contact”. Make sure to have a leading slash." ) def _trailing_slash_required(self): return ( settings.APPEND_SLASH and "django.middleware.common.CommonMiddleware" in settings.MIDDLEWARE ) def clean_url(self): url = self.cleaned_data["url"] if not url.startswith("/"): raise ValidationError( gettext("URL is missing a leading slash."), code="missing_leading_slash", ) if self._trailing_slash_required() and not url.endswith("/"): raise ValidationError( gettext("URL is missing a trailing slash."), code="missing_trailing_slash", ) return url def clean(self): url = self.cleaned_data.get("url") sites = self.cleaned_data.get("sites") same_url = FlatPage.objects.filter(url=url) if self.instance.pk: same_url = same_url.exclude(pk=self.instance.pk) if sites and same_url.filter(sites__in=sites).exists(): for site in sites: if same_url.filter(sites=site).exists(): raise ValidationError( _("Flatpage with url %(url)s already exists for site %(site)s"), code="duplicate_url", params={"url": url, "site": site}, ) return super().clean()