django genericviewset和viewset的区别|极客教程

django genericviewset和viewset的区别

在Django中，APIView是最基本的视图类，用于处理HTTP请求并返回响应。而在开发API时，使用ViewSet和GenericViewSet可以更方便地处理请求和响应，并且可以减少重复代码。但是，很多人对GenericViewSet和ViewSet之间的区别感到困惑。本文将对这两个类的特点和用法进行详细解释，帮助读者更好地理解它们之间的区别。

ViewSet

ViewSet是Django REST framework中定义的一个视图类，可以处理多个HTTP方法（GET、POST、PUT、PATCH、DELETE等），并将这些方法绑定到不同的处理函数上。使用ViewSet可以更好地组织代码，将相关的处理逻辑放在一起，提高代码的可读性和可维护性。

下面是一个简单的ViewSet示例：

from rest_framework.viewsets import ViewSet
from rest_framework.response import Response

class MyViewSet(ViewSet):
    def list(self, request):
        # 处理GET请求，返回所有对象
        return Response({"message": "List all objects"})

    def create(self, request):
        # 处理POST请求，创建一个新对象
        return Response({"message": "Create a new object"})

    def retrieve(self, request, pk=None):
        # 处理GET请求，返回特定对象
        return Response({"message": f"Retrieve object {pk}"})

    def update(self, request, pk=None):
        # 处理PUT请求，更新特定对象
        return Response({"message": f"Update object {pk}"})

    def partial_update(self, request, pk=None):
        # 处理PATCH请求，部分更新特定对象
        return Response({"message": f"Partial update object {pk}"})

    def destroy(self, request, pk=None):
        # 处理DELETE请求，删除特定对象
        return Response({"message": f"Delete object {pk}"})

在上面的示例中，MyViewSet继承自ViewSet类，并实现了不同的处理方法来处理对应的HTTP请求。通过定义这些方法，我们可以清晰地知道不同的HTTP请求应该由哪个函数处理。

GenericViewSet

GenericViewSet是ViewSet的一个子类，它继承了ViewSet并集成了一些通用的处理方法，比如list()、create()、retrieve()、update()、partial_update()和destroy()。使用GenericViewSet可以减少重复代码，简化视图类的编写过程。

下面是一个简单的GenericViewSet示例：

from rest_framework.viewsets import GenericViewSet
from rest_framework.mixins import ListModelMixin, CreateModelMixin, RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin
from rest_framework.response import Response

from .models import MyModel
from .serializers import MyModelSerializer

class MyGenericViewSet(ListModelMixin, CreateModelMixin, RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin, GenericViewSet):
    queryset = MyModel.objects.all()
    serializer_class = MyModelSerializer

在上面的示例中，MyGenericViewSet继承自GenericViewSet，并使用ListModelMixin、CreateModelMixin、RetrieveModelMixin、UpdateModelMixin和DestroyModelMixin来处理对应的HTTP请求。在视图类中通过serializer_class和queryset属性来指定使用的序列化器和查询集，进一步简化了代码编写过程。

区别和适用场景

ViewSet更灵活，可以根据需求自定义处理方法，适用于处理比较复杂的业务逻辑或对不同类型的请求做不同的处理。
GenericViewSet更适合处理简单的CRUD操作，通过继承预定义的混合类来实现常见的处理方法，简化了视图类的编写过程。
使用GenericViewSet可以减少重复代码，提高代码的复用性和可维护性，适合处理与数据库操作相关的请求。
需要自定义处理逻辑或在一个视图中处理多个HTTP方法时，使用ViewSet更合适；需要简化代码编写过程或进行常见的CRUD操作时，使用GenericViewSet更为便捷。

总结

本文介绍了GenericViewSet和ViewSet的区别和特点，并给出了相应的示例代码。ViewSet更灵活，适用于处理复杂的业务逻辑；GenericViewSet更适合处理简单的CRUD操作，通过继承预定义的混合类来简化代码编写过程。根据具体的需求，选择合适的视图类来开发API接口，可以提高开发效率和代码质量。