75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
|
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()
|