5

我想在一个安静的项目中使用 django-guardian 管理我的对象权限(使用 django-rest-framework)。

我想要什么:

  • 仅当他具有“add_modelname”权限时,才允许连接的用户创建对象。
  • 连接用户创建对象时,设置“delete_modelname”和“change_modelname”权限。
  • 仅当连接的用户具有“change_modelobject”权限时,才允许连接的用户编辑对象。
  • 仅当连接的用户具有“delete_modelobject”权限时才允许他删除对象。

我正在尝试使用以下代码管理这些案例:

视图.py

class ModelNameViewSet(viewsets.ModelViewSet):
    """
    This viewset automatically provides `list`, `create`, `retrieve`,
    `update` and `destroy` actions.

    Additionally we also provide an extra `highlight` action.
    """
    queryset = ModelName.objects.all()
    serializer_class = ModelNameSerializer
    permission_classes = (permissions.IsAuthenticatedOrReadOnly, ModelNamePermission)

    def create(self, request, *args, **kwargs):
        assign_perm("change_modelname", request.user, self)
        assign_perm("delete_modelname", request.user, self)
        return super().create(request, *args, **kwargs)

权限.py

class ModelNamePermission(permissions.BasePermission):
    """
    Custom permission to only allow owners of an object to edit it.
    """

    def has_permission(self, request, view):
        if request.method in ['GET']:
            return request.user.has_perm('view_modelname')        
        if request.method in ['POST']:
            return request.user.has_perm('add_modelname')
        if request.method in ['PUT', 'PATCH']:
            return request.user.has_perm('change_modelname')
        if request.method in ['DELETE']:
            return request.user.has_perm('delete_modelname')
        return False

    def has_object_permission(self, request, view, obj):         
        if request.method in ['GET']:
            return request.user.has_perm('view_modelname', obj)        
        if request.method in ['POST']:
            return request.user.has_perm('add_modelname', obj)
        if request.method in ['PUT', 'PATCH']:
            return request.user.has_perm('change_modelname', obj)
        if request.method in ['DELETE']:
            return request.user.has_perm('delete_modelname', obj)
        return False

我遇到的第一个问题是我在这一行有一个错误:

assign_perm("change_modelname", request.user, self)

错误 :

error: 'ModelNameViewSet' object has no attribute '_meta'

而且我认为其余代码将不起作用,但至少您可以看到我想要做什么。

我还没有看到任何关于这些具体案例的例子。

编辑:另一件事是这段代码:

request.user.has_perm('view_coachingrequest')

总是返回真。但是我从来没有给我的用户设置这个权限(只用管理员用户试过,也许这就是原因)。

4

1 回答 1

2

视图集的create方法将返回一个新创建的模型实例。您正在尝试将权限分配给视图集对象本身。您的方法的代码create应如下所示:

def create(self, request, *args, **kwargs):
    instance = super().create(request, *args, **kwargs)
    assign_perm("change_modelname", request.user, instance)
    assign_perm("delete_modelname", request.user, instance)
    return instance

这将创建模型实例,然后在返回之前为其分配所需的权限。

于 2017-05-22T12:10:09.687 回答