(资料图片)

定制视图集操作

视图集提供了一些通用的操作，例如获取列表、创建对象、获取详情、更新对象、部分更新对象和删除对象等。如果需要定制这些操作的行为，可以在视图集中重写对应的方法。例如，在 BookViewSet中，我们可以定制 create()方法，以便在创建新书籍时自动将当前用户设置为书籍的作者：

class BookViewSet(viewsets.ModelViewSet):    queryset = Book.objects.all()    serializer_class = BookSerializer    def perform_create(self, serializer):        serializer.save(author=self.request.user)

复制

在这个例子中，我们在 BookViewSet中重写了 perform_create()方法，该方法会在创建新书籍时被调用。我们使用 self.request.user获取当前请求的用户，并将其设置为新书籍的作者。

另一个例子是定制 list()方法，以便在获取书籍列表时只返回当前用户的书籍：

class BookViewSet(viewsets.ModelViewSet):    serializer_class = BookSerializer    def get_queryset(self):        user = self.request.user        return Book.objects.filter(author=user)

复制

在这个例子中，我们在 BookViewSet中重写了 get_queryset()方法，该方法会在获取书籍列表时被调用。我们使用 self.request.user获取当前请求的用户，并将其用于过滤书籍列表，只返回当前用户的书籍。

视图集类型

DRF 中提供了多种视图集类型，可以根据不同的需求选择合适的视图集类型。以下是几种常见的视图集类型：

ModelViewSet: 提供了默认的 CRUD 操作，以及一些其他常见操作（例如过滤、分页、搜索等）。ReadOnlyModelViewSet: 提供了默认的读取操作，但不支持创建、更新和删除操作。GenericViewSet: 提供了多种操作，例如列表、创建、获取详情、更新、部分更新和删除等。ViewSet: 不提供任何操作，需要手动在子类中定义操作方法。

视图集类型的选择取决于 API 的需求和开发人员的编写习惯。