Django REST Framework 视图集每个操作权限

Django

使用Django REST Framework（简称DRF），我们可以轻松地构建出一个强大的Web API。在DRF中，视图集（ViewSet）是一种将相关操作组织在一起的方式，它允许我们在单个类中定义多个操作，并自动映射到合适的URL路由。除了定义操作之外，我们还可以为每个操作指定不同的权限。

什么是视图集（ViewSets）？

视图集是DRF中一种非常便捷的方式，用于组织和管理API视图。通过使用视图集，我们可以将相关操作（例如列表、创建、更新和删除）组织在一起，并且无需为每个操作都编写单独的视图函数或类。

在DRF中，视图集有两种类型：基于类的视图集（ViewSet）和基于集合的视图集（ModelViewSet）。基于类的视图集提供了一组基本的操作方法（例如列表、创建、更新和删除），而基于集合的视图集则提供了更多的默认操作方法（例如过滤、排序和搜索）。

视图集的权限控制

在API开发中，权限控制是非常重要的一环。DRF提供了丰富的权限类，可以轻松地为每个操作指定不同的权限。以下是一些常用的权限类：

1. AllowAny：允许所有用户访问。

2. IsAuthenticated：只允许认证用户访问。

3. IsAdminUser：只允许管理员用户访问。

4. IsAuthenticatedOrReadOnly：认证用户可以执行任意操作，非认证用户只能执行只读操作。

5. DjangoModelPermissions：基于Django模型的权限，允许通过设置模型的权限进行访问控制。

案例代码

下面是一个简单的示例，演示了如何在DRF中使用视图集和权限控制：

Python
from rest_framework import viewsets, permissions
from .models import Book
from .serializers import BookSerializer
class BookViewSet(viewsets.ModelViewSet):
    queryset = Book.objects.all()
    serializer_class = BookSerializer
    permission_classes = [permissions.IsAuthenticated]
    # 其他操作方法...

在上述代码中，我们定义了一个名为BookViewSet的视图集，它继承自viewsets.ModelViewSet类。我们指定了queryset和serializer_class属性，它们分别表示要操作的模型和使用的序列化器。然后，我们使用permission_classes属性指定了权限类，这里我们使用了IsAuthenticated类，表示只允许认证用户访问。

使用不同权限控制不同操作

在DRF中，我们可以为每个操作指定不同的权限。例如，我们可以将IsAdminUser权限类应用于创建操作，以确保只有管理员用户可以创建新的图书。我们可以按照以下方式修改代码：

Python
from rest_framework import viewsets, permissions
from .models import Book
from .serializers import BookSerializer
class BookViewSet(viewsets.ModelViewSet):
    queryset = Book.objects.all()
    serializer_class = BookSerializer
    permission_classes = {
        'list': [permissions.AllowAny],
        'create': [permissions.IsAdminUser],
        'retrieve': [permissions.AllowAny],
        'update': [permissions.IsAdminUser],
        'partial_update': [permissions.IsAdminUser],
        'destroy': [permissions.IsAdminUser],
    }
    # 其他操作方法...

在上述代码中，我们使用字典的方式为每个操作指定了不同的权限类。例如，list操作使用了AllowAny权限类，表示允许所有用户访问；create操作使用了IsAdminUser权限类，表示只允许管理员用户访问。

通过使用DRF中的视图集和权限控制，我们可以轻松地构建出一个功能强大的Web API。视图集提供了一种组织和管理API视图的便捷方式，而权限控制则确保只有具有合适权限的用户可以执行相应的操作。通过合理地配置视图集和权限类，我们可以满足不同操作的不同需求，从而提供更加安全和可靠的API服务。

参考资料：

- DRF官方文档：Django-rest-framework.org/">https://www.Django-rest-framework.org/

- DRF权限控制文档：Django-rest-framework.org/">https://www.Django-rest-framework.org/api-guide/permissions/