python Wagtail自定义用户帐户设置表单(一对一模型)

oxcyiej7  于 2022-12-28  发布在  Python
关注(0)|答案(1)|浏览(178)

我的应用程序中有一个名为“portal”的模型,使用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管理中的代码段页面添加。
现在,Staff被注册为一个代码段,这意味着任何Staff成员都可以编辑另一个Staff成员的Staff属性。
我想通过向每个用户各自的Staff属性添加一个表单来自定义Wagtail用户帐户设置。这样,我就可以锁定Staff代码段,这样每个拥有链接到User模型的Staff对象的用户就只能更改自己的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字段中预先选择了正确的用户,但我仍然可以选择其他用户。此外,bioroles的存储值不会被检索,新值也不会被保存。
以下是代码片段页面中注册的Staff Roles

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

以下是“帐户设置”中Staff settings部分的外观。

如您所见,虽然User下拉菜单被正确的用户选中,但我仍然可以单击并选择另一个用户。bioroles字段没有填充代码片段页面中的正确值,也没有显示在图片中,填充和保存值不起作用。
我希望我的问题尽可能详细,我如何解决这个问题?

ldfqzlk8

ldfqzlk81#

经过几个小时的挖掘,我终于回答了自己的问题。
1.需要将related_nameprofile值一起添加到Staff.user中。这样可以防止出现'User' object has no attribute 'profile'错误消息

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

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

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

1.棘手的是(至少对我来说)是呈现表单的正确示例,换句话说,如果一个名为“sid”的用户正在访问这个表单,他们需要能够看到他们自己存储的Staff字段,并且还能够更改存储的值。我必须重写CustomStaffSettingsPanel从Django的BaseSettingsPanel继承来的get_form方法。对我来说,将整个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的超级用户可以编辑他们自己的Staff字段,并访问其他字段,这是一个使用Snippet无法轻松实现的功能。
由于Staff不再注册为Snippet,我们无法更改其他用户的Staff字段。可以通过在portal/admin.py中添加以下内容来修复:

admin.site.register(Staff)

只有通过python manage.py createsuperuser创建的超级用户或staff_status设置为true的超级用户才能访问http://localhost:8000/django-admin的Django Admin界面,与Snippet不同,这种方式只允许Django超级用户staff访问和修改Staff字段。
注册为Staff的非超级用户怎么办?非超级用户无法访问Wagtail管理员,因此他们无法访问帐户设置页面。在大多数情况下,这不应该是一个问题,因为理想的用例是只有超级用户可以设置为Staff。如果您想设置非作为Staff的超级用户,您可能必须创建自己的视图和表单,以便用户可以在Wagtail管理之外更改其Staff字段。

相关问题