如何将Wagtail后台MediaPage标签字段改为下拉选择框并关联同站点标签？

1. 自定义过滤站点的标签表单字段 #

先创建自定义表单字段，实现基于当前页面站点过滤标签的逻辑：

from django import forms
from .models import MediaTag

# 单选下拉框用这个字段
class MediaTagChoiceField(forms.ModelChoiceField):
    def __init__(self, page=None, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.page = page

    def _get_queryset(self):
        queryset = super()._get_queryset()
        # 仅保留与当前页面同站点的标签
        if self.page and self.page.site:
            queryset = queryset.filter(site=self.page.site)
        return queryset

# 多选下拉框用这个字段（如果需要支持多选）
class MediaTagMultipleChoiceField(forms.ModelMultipleChoiceField):
    def __init__(self, page=None, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.page = page

    def _get_queryset(self):
        queryset = super()._get_queryset()
        if self.page and self.page.site:
            queryset = queryset.filter(site=self.page.site)
        return queryset

2. 结合Wagtail钩子与字段覆盖实现动态过滤 #

通过Wagtail的编辑钩子传递当前页面实例，再用register_form_field_override替换默认组件：

from django import forms
from wagtail.admin.panels import register_form_field_override
from wagtail.core.hooks import register
from .models import MediaPage, MediaTag

# 钩子：在页面编辑前，将当前页面实例传递给标签字段
@register('before_edit_page')
def pass_current_page_to_tag_field(request, page_instance, form):
    if isinstance(page_instance, MediaPage) and 'tags' in form.fields:
        form.fields['tags'].page = page_instance

# 注册字段覆盖，替换为下拉框组件
# 单选版本
register_form_field_override(
    MediaPage,
    'tags',
    override={
        'form_class': MediaTagChoiceField,
        'widget': forms.Select(attrs={'class': 'select2'}),
        'queryset': MediaTag.objects.all(),
        'required': True  # 匹配原字段blank=False的设置
    }
)

# 如果需要多选下拉框，替换上面的register_form_field_override为以下代码：
# register_form_field_override(
#     MediaPage,
#     'tags',
#     override={
#         'form_class': MediaTagMultipleChoiceField,
#         'widget': forms.SelectMultiple(attrs={'class': 'select2'}),
#         'queryset': MediaTag.objects.all(),
#         'required': True
#     }
# )

3. 核心逻辑说明 #

• 自定义字段通过_get_queryset方法，利用传入的页面实例过滤出同站点标签；
• before_edit_page钩子负责在页面编辑表单初始化时，把当前页面实例传递给标签字段，确保过滤逻辑能获取到站点信息；
• 单选/多选可根据需求切换对应的字段和组件，select2类是为了优化下拉框的交互体验。

内容的提问来源于stack exchange，提问作者Nikolay