Wagtail 使用一对一模型自定义用户帐户设置表单

如何解决Wagtail 使用一对一模型自定义用户帐户设置表单

我的应用中有一个名为“门户”的模型，位于 portal/models.py 中：

from django.db import models
from django.contrib.auth.models import User
from django.dispatch import receiver
from django.db.models.signals import post_save

from wagtail.snippets.models import register_snippet
from wagtail.admin.edit_handlers import FieldPanel

@register_snippet
class StaffRoles(models.Model):
    role = models.CharField(max_length=154,unique=True,help_text="Create new staff roles here,these roles an be assigned to 'staff' users.")
    panels = [
        FieldPanel('role'),]

    def __str__(self):
        return self.role

    class Meta:
        verbose_name = "Staff Role"
        verbose_name_plural = "Staff Roles"

@register_snippet
class Staff(models.Model):
    user = models.OneToOneField(User,on_delete=models.CASCADE)
    bio = models.TextField(max_length=1024,blank=True)
    roles = models.ManyToManyField(StaffRoles,blank=True)
    
    def __str__(self):
        return str(self.user.first_name) + " " + str(self.user.last_name)

    class Meta:
        verbose_name = "Staff"
        verbose_name_plural = "Staff"


@receiver(post_save,sender=User)
def create_user_profile(sender,instance,created,**kwargs):
    if created and instance.is_superuser:
        Staff.objects.create(user=instance)

任何新的超级用户都会自动获得 Staff 模型。任何现有（非超级用户）用户也可以添加为 Staff。

我希望工作人员能够设置 bio 和 role。可以通过 Wagtail 管理中的 Snippets 页面添加角色。

现在，Staff 已注册为片段，这意味着任何职员都可以编辑其他职员的职员属性。

我想通过向每个用户各自的员工属性添加一个表单来自定义 Wagtail 用户帐户设置。通过这种方式，我可以锁定 Staff 片段，以便每个拥有 Staff 对象链接到其 User 模型的用户只能更改他们自己的 Staff 字段/属性。

按照 here 的 Wagtail 文档，我首先创建了包含内容的文件 portal/forms.py：

from django import forms
from django.contrib.auth import get_user_model
from django.contrib.auth.models import User

from wagtail.users.models import UserProfile

from .models import Staff

class CustomStaffSettingsForm(forms.ModelForm):

    class Meta:
        model = Staff
        exclude = []

还有另一个包含内容的文件 portal/wagtail_hooks.py：

from wagtail.admin.views.account import BaseSettingsPanel
from wagtail.core import hooks

from .forms import CustomStaffSettingsForm

@hooks.register('register_account_settings_panel')
class CustomStaffSettingsPanel(BaseSettingsPanel):
    name = 'custom_staff'
    title = "Staff settings"
    order = 500
    form_class = CustomStaffSettingsForm
    form_object = 'profile'

这不符合预期。访问“帐户设置”并向下滚动到“员工设置”时，我可以按预期看到 CustomStaffSettingsForm。尽管在 User 字段中预先选择了正确的用户，但我仍然可以选择其他用户。此外，不会检索 bio 和 roles 的存储值，也不会保存新值。

以下是在代码段页面中注册的 Staff Roles：

以下是代码段页面中用户“sid”的 Staff 模型：

下面是“帐户设置”中的 Staff settings 部分的样子。

如您所见，虽然 User 下拉列表已选择给正确的用户，但我仍然可以单击并选择其他用户。 bio 和 roles 字段未填充片段页面中的正确值且未显示在图片中，填充和保存值不起作用。

我希望我的问题尽可能详细，我该如何解决？

解决方法

经过几个小时的挖掘，我设法回答了我自己的问题。

需要向 Staff.user 添加一个 related_name，其值为 profile。这可以防止 'User' object has no attribute 'profile' 错误消息

class Staff(models.Model):
    user = models.OneToOneField(User,on_delete=models.CASCADE,related_name='profile')
    ...

我们需要从 user 中排除 CustomStaffSettingsForm 字段。这是为了防止用户选择其他用户并更改其他用户的 Staff 字段。

class CustomStaffSettingsForm(forms.ModelForm):
    class Meta:
        model = Staff
        exclude = ["user"]

棘手的部分（至少对我而言）是呈现表单的正确实例，换句话说，如果名为“sid”的用户正在访问此表单，他们需要能够看到自己的 {{1 }} 存储的字段，也可以更改存储的值。为了实现这一点，我必须重写 Staff 从 Django 的 get_form 继承而来的 CustomStaffSettingsPanel 方法。我可以更轻松地将整个 BaseSettingsPanel 内容粘贴到下方。

wagtail_hooks.py

上面代码的关键部分是from wagtail.admin.views.account import BaseSettingsPanel from wagtail.core import hooks from .forms import CustomStaffSettingsForm @hooks.register('register_account_settings_panel') class CustomStaffSettingsPanel(BaseSettingsPanel): name = 'custom_staff' title = "Staff settings" order = 300 form_class = CustomStaffSettingsForm form_object = 'profile' def get_form(self): """ Returns an initialised form. """ kwargs = { 'instance': self.request.user.profile,'prefix': self.name } if self.request.method == 'POST': return self.form_class(self.request.POST,**kwargs) else: return self.form_class(**kwargs)。将此参数传递到 'instance': self.request.user.profile 会导致正确检索和存储正确的用户存储的 self.form_class 字段和值。

此解决方案让我可以确保也注册为 Staff 的超级用户可以编辑自己的 Staff 字段并可以访问其他人，这是使用 Snippets 无法轻松实现的功能。

由于 Staff 不再注册为片段，我们无法更改其他用户的 Staff 字段。这可以通过在 Staff 中添加以下内容来解决：

portal/admin.py

只有通过 admin.site.register(Staff) 创建或将 python manage.py createsuperuser 设置为 true 的超级用户才能在 staff_status 访问 Django 管理界面。与 Snippets 不同，这种方式可用于仅允许 Django 超级用户人员访问和修改人员字段。

注册为 http://localhost:8000/django-admin 的非超级用户呢？非超级用户无法访问 Wagtail 管理员，因此他们无法访问帐户设置页面。在大多数情况下，这应该不是问题，因为理想的用例是只有超级用户可以设置为 Staff。如果您想将非超级用户设置为 Staff，您可能需要创建自己的视图和表单，以便用户可以在 Wagtail 管理员之外更改他们的 Staff 字段。

Wagtail 使用一对一模型自定义用户帐户设置表单

如何解决Wagtail 使用一对一模型自定义用户帐户设置表单

解决方法

相关推荐