我需要将基于字符串的唯一 ID添加到 Wagtail 的图像模型中。这些 ID 是字母、数字和标点符号的相对较短的组合,例如“AS.M-1.001”。因此,我使用带有参数的 Django 标准的自定义图像模型。但遗憾的是,编辑表单不会检查 ID 是否唯一。所以我可以对多个图像使用相同的 ID。这种唯一性检查确实适用于 Wagtail 中的任何其他标准形式,例如模型。但不适用于图像模型。CharField
unique=True
Page
from django.db import models
from wagtail.images.models import Image, AbstractImage
class CustomImage(AbstractImage):
custom_id = models.CharField(max_length=32, unique=True, null=True, blank=True)
admin_form_fields = ( 'custom_id', ) + Image.admin_form_fields
我的方法是使用自定义表单覆盖编辑表单以显示更多警告和错误,就像您可以base_form_class
对 Wagtail 的Page
模型等执行的操作一样,如此处所述。我尝试了 wagtail.admin.forms.WagtailAdminModelForm 以及 wagtail.images.forms.BaseImageForm。
from wagtail.images.forms import BaseImageForm
from wagtail.admin.forms import WagtailAdminModelForm
class CustomImageForm(WagtailAdminModelForm):
# add more logic here
pass
class CustomImage(ArtButlerIDMixin, AbstractImage):
...
base_form_class = CustomImageForm
两者都导致相同的异常:
raise AppRegistryNotReady("Models aren't loaded yet.")
因此,我试图在我的设置中使用这些应用程序,但没有任何效果。如何在 Wagtail 中覆盖管理表单的标准方法是否适用于图像模型?还有什么其他方法可以让唯一的字符串标识符在这里工作?...或自定义此表单。
解决方案(更新)
按照@gasman 的建议,我将以下行添加到我的settings/base.py
:
WAGTAILIMAGES_IMAGE_MODEL = 'images.CustomImage'
WAGTAILIMAGES_IMAGE_FORM_BASE = 'images.forms.CustomImageForm' # NEW
forms.py
并在我的图像应用程序中添加了以下表单:
from django.core.exceptions import ValidationError
from wagtail.images.forms import BaseImageForm
from .models import CustomImage
class CustomImageForm(BaseImageForm):
def clean(self):
cleaned_data = super().clean()
custom_id = cleaned_data.get("custom_id")
if CustomImage.objects.filter(custom_id=custom_id).exists():
raise ValidationError(
"Custom ID already exists"
)
return cleaned_data