Django中的权限类

Question

我使用Django组权限，可以为用户分配一些权限。 这是我的测试场景：

我有一个公司模型，

User1.hasPerm- > view_company, change_company, add_company, delete_company

Permissions.py：

class HasPermissions(permissions.BasePermission):
def has_permission(self, request, view):
    if request.user.has_perm('MYAPP.view_company'):
        return True
    else:
        return False
    if request.user.has_perm('MYAPP.change_company'):
        return True
    else:
        return False
    if request.user.has_perm('MYAPP.add_company'):
        return True
    else:
        return False
    if request.user.has_perm('MYAPP.delete_company'):
        return True
    else:
        return False

    return True

CompanyView.py：

class CompanyViewSet(ModelViewSet):
queryset = Company.objects.all()
filter_class = CompanyFilter
serializer_class = CompanySerializer
permission_classes = [IsAuthenticated, HasPermissions]

def get_queryset(self):
    if self.request.user.is_authenticated and self.request.user.is_active:
        company = Company.objects.filter(companyUser__exact=self.request.user)
        return company
    else:
        return Company.objects.all()

我编写了HasPermissions函数来控制用户权限，并且此函数仅适用于CompanyView。 我想使此功能成为全局控件，因为我可以控制所有视图。

HasPermissions函数就像硬编码一样，我想更改更可用的版本来控制所有视图。

我怎样才能做到这一点？

Answer 1

您无需自己编写。 Django已经有一个DjangoModelPermissions [drf-doc] ：

该权限类与Django的标准django.contrib.auth模型权限相关。 此权限只能应用于设置了.queryset属性的视图。 只有在对用户进行身份验证并分配了相关模型权限的情况下，才可以授予授权。

• POST请求要求用户对模型具有add权限。
• PUT和PATCH请求要求用户具有对模型的change权限。
• DELETE请求要求用户对模型具有delete权限。

因此，您可以使用：

class CompanyViewSet(ModelViewSet):
    queryset = Company.objects.all()
    filter_class = CompanyFilter
    serializer_class = CompanySerializer
    permission_classes = [IsAuthenticated, DjangoModelPermissions]