Django多组织用户角色权限模型存在重复问题，求优化方案

1. 从模型层面杜绝重复记录（最核心优化） #

当前模型允许同一用户在同一组织下生成多条角色记录，这是根源问题。直接在数据库层面添加唯一约束，从源头阻止重复数据：

class OrgUserPerms(models.Model):
    organization = models.ForeignKey(Organization, on_delete=models.CASCADE)
    user = models.ForeignKey("accounts.CustomUser", on_delete=models.CASCADE)
    user_role = models.CharField(max_length=25, choices=[
        ("organization_admin","admin"), 
        ("organization_manager", "manager"),
        ("organization_member", "member")
    ])

    class Meta:
        # Django 2.2+推荐用UniqueConstraint，支持更灵活的约束规则
        constraints = [
            models.UniqueConstraint(
                fields=['organization', 'user'], 
                name='unique_org_user_role'
            )
        ]
        # 兼容旧版Django可改用：unique_together = ('organization', 'user')

添加约束后，数据库会直接拒绝重复的(organization, user)组合插入/更新，彻底避免重复记录问题。

2. 清理现有重复数据 #

针对已存在的重复记录，执行一次数据清理脚本（可写成Django迁移或独立脚本），保留每个用户-组织组的最新记录：

from django.db.models import Max
from your_app.models import OrgUserPerms

# 筛选出每个用户-组织组的最新记录ID
latest_record_ids = (
    OrgUserPerms.objects.values("organization", "user")
    .annotate(latest_id=Max("id"))
    .values_list("latest_id", flat=True)
)

# 删除非最新的重复记录
OrgUserPerms.objects.exclude(id__in=latest_record_ids).delete()

3. 优化角色更新逻辑 #

后续修改用户角色时，不要新增记录，直接更新已有条目。用Django内置的update_or_create方法一步完成：

# 更新用户在指定组织的角色
OrgUserPerms.objects.update_or_create(
    organization=target_org,
    user=target_user,
    defaults={"user_role": new_role}
)

该方法会先查询匹配记录，存在则更新角色，不存在则创建新记录，完全避免重复插入。

4. 提升权限验证性能 #

频繁查询OrgUserPerms会带来数据库压力，建议缓存用户的组织角色信息：

from django.core.cache import cache

def get_user_org_role(user, org):
    cache_key = f"user_{user.id}_org_{org.id}_role"
    role = cache.get(cache_key)
    if not role:
        try:
            role = OrgUserPerms.objects.get(user=user, organization=org).user_role
            cache.set(cache_key, role, 3600)  # 缓存1小时
        except OrgUserPerms.DoesNotExist:
            role = None
    return role

# 角色更新时清除对应缓存
def update_user_org_role(user, org, new_role):
    OrgUserPerms.objects.update_or_create(
        organization=org, user=user, defaults={"user_role": new_role}
    )
    cache.delete(f"user_{user.id}_org_{org.id}_role")

5. 整合到Django权限系统（可选） #

如果想让角色权限和Django原生user.has_perm()兼容，可以自定义权限后端：

from django.contrib.auth.backends import ModelBackend

class OrgRolePermissionBackend(ModelBackend):
    def has_perm(self, user_obj, perm, obj=None):
        # 从对象或请求中获取当前组织（需提前通过中间件把组织绑定到request）
        org = None
        if obj and hasattr(obj, "organization"):
            org = obj.organization
        else:
            # 假设你有中间件将当前组织存入request
            from django.contrib.auth.middleware import get_request
            request = get_request()
            org = getattr(request, "current_organization", None)

        if not org or user_obj.is_anonymous:
            return super().has_perm(user_obj, perm, obj)

        # 获取用户在该组织的角色
        try:
            role = OrgUserPerms.objects.get(user=user_obj, organization=org).user_role
        except OrgUserPerms.DoesNotExist:
            return False

        # 映射角色到权限列表，根据业务需求调整
        role_permissions = {
            "organization_admin": ["org:manage_users", "org:edit_settings", "org:view_data"],
            "organization_manager": ["org:edit_settings", "org:view_data"],
            "organization_member": ["org:view_data"]
        }
        return perm in role_permissions.get(role, [])

然后在settings.py中配置：

AUTHENTICATION_BACKENDS = [
    "your_app.backends.OrgRolePermissionBackend",
    "django.contrib.auth.backends.ModelBackend",
]

内容的提问来源于stack exchange，提问作者salem