更新时如何将用户添加到组并创建用户

Question

我创建了一群堕落的链接的小组： 我订阅了用户类添加了一个名为“角色”的选择字段。基本上，这意味着用户的角色。这样，如果用户扮演“员工”的角色。因此，它将拥有团队工作人员的许可。问题是：我无法让Django按照我的预期做出回应。保存后，我发出了一个信号，该信号应根据他们的角色将用户添加到组中。该程序看起来像这样：

# project/app/model.py

class User(AbstractUser):
    class Roles(models.IntegerChoices):
        SUPER = 0, _('SuperAdmins')
        COMPANY = 1, _('Company')
        UNITY = 2, _('Unity')
        STAFF = 3, _('Staff')

   role: Roles = models.IntegerField(choices=Roles.choices, default=Roles.STAFF, verbose_name=_("Role"))

我的信号就像：

GROUPS = ['SuperAdmins', 'Company', 'Unity', 'Staff']
@receiver(post_save, sender=User)
def user(sender: User, instance: User, created: bool, **kwargs) -> None:
    """
    This receiver function will set every staff pages that is created to the group staff.

    :param sender: the model that will trigger this receiver
    :param instance: the instance
    :param created: if it was already created
    :param kwargs:
    :return: None
    """
    if created:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.add(group)
    else:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.add(group)
        print("update")

当我转到管理页面并创建一个新用户时，一切都按预期工作。但是，当我编辑现有用户的角色时，它只是在打印功能上打印“更新”，但没有任何变化。我在做什么错？

Answer 1

如果有人遇到同样的问题，我将在此处发布解决方案。以及@mtzd的贡献。我根据本文中的问题的描述找到了解决方案的提示： https://stackoverflow.com/a/ 1925784/18600263 。因此，pos_save信号将是：

from django.db import transaction

GROUPS = ['SuperAdmins', 'Company', 'Unity', 'Staff']


@receiver(post_save, sender=User)
def user(sender: User, instance: User, **kwargs) -> None:
    group = Group.objects.get(name=GROUPS[instance.role])
    transaction.on_commit(lambda: instance.groups.set([group], clear=True))

Answer 2

使用当前的代码，更改用户角色时，它将被添加到新组中（而不会从上一组中删除）。例如，如果用户最初是组的一部分员工，则其角色更改为 Company ，它将被添加到 Company 组中。由于 user 模型与组模型之间的关系是很多与许多关系的关系，因此不会从 staff 组中删除用户。为此，有一些方法（基本上是相同的）：

clear 然后 add ：

if created:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.add(group)
    else:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.clear() # Dissasociates any group the user belonged to
        instance.groups.add(group) # Adds the group

第二个选项：： set set </

if created:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.add(group)
    else:
        group = Group.objects.get(name=GROUPS[instance.role])
        instance.groups.set([group], clear=True) # Dissasociates any group the user belonged to and sets the new group. clear=True is necessary here because otherwise the Group instances would be deleted

code ：这，如果您决定使用第二个选项，则可以使用 set 也适用于正在创建的用户：

GROUPS = ['SuperAdmins', 'Company', 'Unity', 'Staff']
@receiver(post_save, sender=User)
def user(sender: User, instance: User, **kwargs) -> None:
    """
    This receiver function will set every staff pages that is created to the group staff.

    :param sender: the model that will trigger this receiver
    :param instance: the instance
    :param kwargs:
    :return: None
    """
    group = Group.objects.get(name=GROUPS[instance.role])
    instance.groups.set([group], clear=True)

因此，您不必使用创建的参数。有关 clear 和 set 的更多信息，您可以读取docs https://docs.djangoproject.com/en/4.0/ref/ref/ref/models/models/relations/relations/#django.db.models。 fields.recated.releatedmanager.clear 看看如何使用它们！