Wagtail：如何将按钮“添加子页面”添加到自定义列表视图 (IndexView) 的操作列表中？

Question

我正在基于每个用户构建管理菜单，没有资源管理器视图和自定义ModelAdmin实例。

@hooks.register('construct_main_menu')
def hide_page_explorer_menu_item(request, menu_items):
    if request.user.username == 'user1':
        menu_items[:] = [item for item in menu_items if item.name not in [
            'explorer',
            ]
        ]

class CustomAdmin(ModelAdmin):
    model = MyPage
    menu_label = 'custom ModelAdmin'

    list_display = ('title', "live")
    list_filter = ("date")
    search_fields = ("title")

modeladmin_register(CustomAdmin)

但是结果列表视图的操作列表（即上下文菜单）缺少资源管理器视图提供的某些功能。

我特别需要“添加儿童页面”。我知道钩子register_page_listing_buttons。但是您只能为资源管理器视图操作列表注册按钮。

是否可以将“添加子页面”按钮添加到自定义列表视图的操作列表中？

---

资源管理器列表视图：

自定义 ModelAdmin 列表视图：

Answer 1

回答

ModelAdmin 不使用相同的系统来构建在其视图中列出的按钮。

相反，它有一个“帮助类”的概念，它们用于生成 URL 和权限检查，而这些检查又用于生成出现在各种视图中的按钮。

https://docs.wagtail.io/en/stable/reference/contrib/modeladmin/primer.html?highlight=primer

您将需要在ModelAdmin实例上设置一个自定义 button_helper_class，该实例将在您的IndexView. 为此，您需要PageButtonHelper使用get_buttons_for_obj将返回按钮列表的自定义方法导入和扩展该类。

代码示例

# wagtail_hooks.py

from django.urls import reverse
from wagtail.contrib.modeladmin.options import (ModelAdmin, modeladmin_register)
from wagtail.contrib.modeladmin.helpers import (PageAdminURLHelper, PageButtonHelper)
# MyPage model import not included but will be needed

class CustomPageAdminURLHelper(PageAdminURLHelper):

    def get_action_url(self, action, *args, **kwargs):
        if action == 'add-child':
            # note: add_subpage is used as the internal name for adding child pages
            url_name = 'wagtailadmin_pages:add_subpage'
            target_url = reverse(url_name, args=args, kwargs=kwargs)
            return target_url

        # for every other case - just call the parent method
        return super().get_action_url(action, *args, **kwargs)


class CustomPageButtonHelperClass(PageButtonHelper):

    add_child_button_classnames = ['add-child'] # can be anything - just setting this to match convention

    # add a new button generator that mimics the 'copy_button' from existing code
    def add_child_button(self, pk, classnames_add=[], classnames_exclude=[]):
        classnames = self.add_child_button_classnames + classnames_add
        final_classnames = self.finalise_classname(classnames, classnames_exclude)

        return {
            # if we wanted - we can skip the creation of CustomPageAdminURLHelper & generate the URL here
            # but this may work against reusability later
            'url': self.url_helper.get_action_url('add-child', pk), # may need to wrap pk in quote if it could contain non-url-safe chars
            'label': 'Add Child',
            'classname': final_classnames,
            'title': 'Add Child uner this %s' % self.verbose_name
        }

    # override the PageButtonHelper method to 'add' custom buttons
    def get_buttons_for_obj(self, obj, exclude=[], classnames_add=[], classnames_exclude=[]):
        # call the parent class method to get the default set of buttons
        buttons = super().get_buttons_for_obj(obj, exclude, classnames_add, classnames_exclude)

        # set up some variables to do user checks and also get the primary key (id)
        permission_helper = self.permission_helper
        user = self.request.user
        pk = getattr(obj, self.opts.pk.attname)

        # many existing permission helpers are already available - see wagtail/contrib/modeladmin/helpers/permission.py
        if ('add-child' not in exclude and permission_helper.user_can_create(user)):
            # above we follow the convention set up where buttons can be included/excluded easier
            # we also do a permissions check
            # finally we build the button and add it to the end of the buttons list
            add_child_button = self.add_child_button(pk, classnames_add, classnames_exclude)
            buttons.append(add_child_button)

        return buttons

class CustomAdmin(ModelAdmin):
    model = MyPage
    menu_label = 'custom ModelAdmin'
    button_helper_class = CustomPageButtonHelperClass # added to enable custom button generation
    url_helper_class = CustomPageAdminURLHelper # added to enable custom url generation

    list_display = ('title', "live")
    list_filter = ("date")
    search_fields = ("title")

modeladmin_register(CustomAdmin)

代码说明

• https://github.com/wagtail/wagtail/tree/master/wagtail/contrib/modeladmin/helpers - 包含所有助手的代码，了解导入和被覆盖的现有方法很重要。

CustomPageButtonHelperClass

• get_buttons_for_obj通过附加 add_child_button覆盖。
• 使用现有的权限助手类做一些检查，这将避免需要重写容易出错的用户和身份验证逻辑
• 调用方法（我们将添加）以生成按钮内容， add_child_button
• 添加方法add_child_button- 必须返回一个带有 url、标签、类名和标题的对象。尝试保留现有的约定，以便可以在全局范围内附加/删除类名
• 最难的是 url 生成，我们将利用 URL 助手的现有结构，但您可以直接生成此 URL。

CustomPageAdminURLHelper

• 覆盖方法get_action_url并处理我们关心的“动作”在哪里的特定情况'add-child'
• 此方法必须简单地返回一个字符串（这将是一个相对 url），最好使用现有的 Django URL 反转系统。
• 添加子页面的url名称是 'wagtailadmin_pages:add_subpage'
• 最后使用传入的 args/kwargs 调用 Django reverse util，你将得到一个漂亮的 url 作为回报。

CustomAdmin

• 最后一步是在您的 CustomAdmin 实例上设置这些自定义 url 和按钮助手类。
• 这只需要button_helper_class和url_helper_class设置为它们各自的帮助程序覆盖。

额外注意事项和注意事项

• 仔细阅读 ModelAdmin 文档，它们的布局非常好，有很大帮助。https://docs.wagtail.io/en/stable/reference/contrib/modeladmin/index.html?highlight=modeladmin%20helpers#
• 缺少一些方面，但代码库中的 helpers 文件夹非常易于阅读，应该可以帮助您确定正在自定义的代码的流程。
• 这是假设您只想在 Page 模型上使用这些自定义帮助程序，ModelAdmin 处理页面和其他对象的方式之间存在细微差别，您可以进行额外检查，但这取决于您。
• 如果您不关心类名和遵循现有的模型管理约定，您可以缩短一些样板代码，这取决于您。
• 这假设您的主键是 URL 安全的，您将在 modeladmin 代码中看到有很多额外的 utils 用法来使所有 url 安全。
• 创建子页面后，用户将返回到父页面的非模型管理员列表。改变这一点可能需要一些额外的工作，但上面的代码应该可以满足您的需求。