如何在Wagtail Streamfield中限制特定DocumentChooserBlock的可选文件类型

有两种可行的实现方案，均可实现仅针对单个StreamField区块的文档选择器过滤，不会影响全局其他区块的正常使用。

方案一：自定义专属DocumentChooserBlock子类（最为推荐） #

直接继承官方的DocumentChooserBlock类，重写其get_queryset方法添加过滤规则，使用时在需要限制的区块处替换为自定义的Block类即可。
示例代码如下：

from wagtail.documents.blocks import DocumentChooserBlock
from wagtail.documents.models import Document

# 自定义仅可选PDF文件的DocumentChooserBlock
class PdfOnlyDocumentChooserBlock(DocumentChooserBlock):
    def get_queryset(self, request):
        # 先获取父类默认的查询集
        qs = super().get_queryset(request)
        # 添加自定义过滤规则，示例为仅返回后缀为pdf的文档
        return qs.filter(file_extension="pdf")

在StreamField中引用示例：

from wagtail.models import Page
from wagtail.fields import StreamField
from wagtail.blocks import RichTextBlock

class YourPage(Page):
    body = StreamField([
        # 普通的文档选择器，不受影响
        ("normal_document", DocumentChooserBlock(label="通用文档")),
        # 仅可选PDF的自定义文档选择器
        ("pdf_document", PdfOnlyDocumentChooserBlock(label="PDF文件")),
        ("rich_text", RichTextBlock()),
    ], use_json_field=True)

该方案的优势：

• 逻辑完全独立，和全局的construct_document_chooser_queryset钩子互不干扰，不会影响其他区块的使用
• 可复用性强，多处需要同样过滤规则的区块可直接引用该自定义类
• 是Wagtail官方支持的标准扩展方式，兼容性强，支持Wagtail 3.0及以上所有版本

方案二：基于全局钩子加上下文判断 #

如果你需要保留钩子的使用方式，也可以在钩子逻辑中添加上下文判断，仅在命中目标区块时才触发过滤：

from wagtail import hooks

@hooks.register("construct_document_chooser_queryset")
def filter_pdf_documents_for_specific_block(documents, request):
    # 从请求参数中获取当前调用的区块类型/页面字段标识，判断是否需要过滤
    if request.GET.get("block_type") == "pdf_only_block":
        documents = documents.filter(file_extension="pdf")
    return documents

该方案适合过滤规则需要动态调整的场景，但需要额外处理请求参数的识别逻辑，维护成本高于第一种方案。

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