将所有守护用户对象权限从一个Django用户重新分配给另一个Django用户

在这种情况下，我们可能会在"批量"中这样做，就像在UserObjectPermissionBase的一个查询per子类中一样.然而，有一个可能的警告:如果我们想将权限分配给新用户，并且用户已经获得了该权限.

单个模型的查询如下所示:

MyModelUserObjectPermission.objects.filter(user=old_user).update(user=new_user)

但是，如果新用户的权限已经存在，则此操作将失败，因为此时将有一个将失败的唯一性约束.

然而，我们可以通过以下方式达到look before we leap ^[wiki]:

from django.db.models import Exists, OuterRef

MyModelUserObjectPermission.objects.filter(
    ~Exists(
        MyModelUserObjectPermission.objects.filter(
            user=new_user,
            content_object=OuterRef('content_object'),
            permission=OuterRef('permission'),
        )
    ),
    user=old_user,
).update(user=new_user)
MyModelUserObjectPermission.objects.filter(user=old_user).delete()

因此，这将首先判断我们是否可以更改它，以便相同的对象、权限和新用户没有100.如果是这样的话，我们会更新它.在第二个查询中，我们随后删除了旧用户中仍然存在的权限.

现在我们唯一需要做的就是对all模型执行此操作，我们可以通过查找UserObjectPermissionBase的子类来完成此操作:

from django.db.models import Exists, OuterRef
from guardian.models.models import UserObjectPermissionBase


def change_perm_for_klass(klass, old_user, new_user):
    klass.objects.filter(
        ~Exists(
            klass.objects.filter(
                user=new_user,
                content_object=OuterRef('content_object'),
                permission=OuterRef('permission'),
            )
        ),
        user=old_user,
    ).update(user=new_user)
    klass.objects.filter(user=old_user).delete()


def update_permissions_for_user(old_user, new_user):
    subclasses = set()
    new_gen = {UserObjectPermissionBase}
    while new_gen:
        subclasses.update(new_gen)
        new_gen = {sc for k in new_gen for sc in k.__subclasses__()}
    for klass in subclasses:
        if not klass._meta.abstract:
            change_perm_for_klass(klass, old_user, new_user)

因此，在这里，我们首先划分类层次 struct ，以查找UserObjectPermissionBase的所有子类，接下来，我们将为所有非抽象的模型调用change_perm_for_klass(…).

当然，只有在您通过定义UserObjectPermissionBase的子类来实现django-guardian的情况下，这种逻辑才有效.