Django 权限按国家/地区分隔用户

Question

我有一个关于权限的问题。我想通过国家/地区分隔标记为员工（并且可以进入管理页面）的用户。例如，来自荷兰的管理员只能查看和编辑来自荷兰的用户。最好的方法是什么？
我是否应该以某种方式进行查询，以便用户只能看到来自他们自己国家/地区的用户，而不是我可以使用权限对其进行自定义？

我的 models.py 定义了一个国家

class UserProfile(models.Model):

    user = models.OneToOneField(User, related_name='profile', unique=True)
    country = models.CharField("Country", max_length=150, blank=False)

我在 admin.py 中想过这样的事情

def queryset(self, request):
    qs = super(ProfileAdmin, self).queryset(request)
    if request.user.is_superuser:
        return qs
    return qs.filter(country=request.user.country)

Answer 1

你的想法是正确的。这是过滤管理员可见的对象的正确方法。

小修正：方法名称是get_queryset()，要访问用户所在的国家/地区，您应该使用profile 关系：

def get_queryset(self, request):
    qs = super(ProfileAdmin, self).get_queryset(request)
    if request.user.is_superuser:
        return qs
    return qs.filter(country=request.user.profile.country)

更新：要过滤标准 User 管理员中的用户，您必须将 UserAdmin 替换为您自己的子类：

from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User

class MyUserAdmin(UserAdmin):

    def get_queryset(self, request):
        qs = super(MyUserAdmin, self).get_queryset(request)
        if request.user.is_superuser:
            return qs
        return qs.filter(profile__country=request.user.profile.country)

admin.site.unregister(User)
admin.site.register(User, MyUserAdmin)