如何将单个DRF ViewSet类拆分为多个类？

完全可以拆分，两种常用方案能帮你解决代码臃肿、导航困难的问题：

方案一：使用Mixin类拆分相关Action #

把功能相近的@action按业务模块分组，放到独立的Mixin类中，主ViewSet继承这些Mixin和基础的ModelViewSet。这种方式能保持ViewSet的逻辑关联性，同时拆分代码文件。

示例步骤： #

1. 创建mixins目录，按业务模块创建Mixin文件：

   • mixins/user_actions.py：存放用户相关的Action

     from rest_framework.decorators import action
     from rest_framework.response import Response
     
     class UserActionMixin:
         @action(detail=True, methods=['get'])
         def user_profile(self, request, pk=None):
             instance = self.get_object()
             # 编写用户信息查询逻辑
             return Response({'profile': instance.user_profile})
     
         # 添加其他用户相关Action...
     

   • mixins/order_actions.py：存放订单相关的Action

     from rest_framework.decorators import action
     from rest_framework.response import Response
     
     class OrderActionMixin:
         @action(detail=True, methods=['post'])
         def create_order(self, request, pk=None):
             instance = self.get_object()
             # 编写创建订单逻辑
             return Response({'order_id': 'ORD-12345'})
     
         # 添加其他订单相关Action...
     

2. 主ViewSet文件中继承这些Mixin：

   from rest_framework.viewsets import ModelViewSet
   from .models import YourModel
   from .serializers import YourSerializer
   from .mixins.user_actions import UserActionMixin
   from .mixins.order_actions import OrderActionMixin
   
   class YourModelViewSet(UserActionMixin, OrderActionMixin, ModelViewSet):
       queryset = YourModel.objects.all()
       serializer_class = YourSerializer
       # 保留原ViewSet的配置（如permission_classes、filter_backends等）
   

方案二：拆分为独立ViewSet #

把完全独立的业务Action分到单独的GenericViewSet中，通过路由注册时指定相同的basename，保证路由规则一致。这种方式代码隔离更彻底，适合业务模块完全独立的场景。

示例步骤： #

1. 按业务模块创建独立ViewSet文件：

   • views/user_actions_viewset.py

     from rest_framework.viewsets import GenericViewSet
     from rest_framework.decorators import action
     from rest_framework.response import Response
     from .models import YourModel
     from .serializers import YourSerializer
     
     class UserActionsViewSet(GenericViewSet):
         queryset = YourModel.objects.all()
         serializer_class = YourSerializer
     
         @action(detail=True, methods=['get'])
         def user_profile(self, request, pk=None):
             instance = self.get_object()
             return Response({'profile': instance.user_profile})
     

   • views/order_actions_viewset.py

     from rest_framework.viewsets import GenericViewSet
     from rest_framework.decorators import action
     from rest_framework.response import Response
     from .models import YourModel
     from .serializers import YourSerializer
     
     class OrderActionsViewSet(GenericViewSet):
         queryset = YourModel.objects.all()
         serializer_class = YourSerializer
     
         @action(detail=True, methods=['post'])
         def create_order(self, request, pk=None):
             instance = self.get_object()
             return Response({'order_id': 'ORD-12345'})
     

2. 路由中注册所有ViewSet，指定相同的basename：

   from rest_framework.routers import DefaultRouter
   from .views import YourModelViewSet, UserActionsViewSet, OrderActionsViewSet
   
   router = DefaultRouter()
   # 基础CRUD操作的ViewSet
   router.register(r'your-model', YourModelViewSet, basename='your-model')
   # 用户相关Action的ViewSet
   router.register(r'your-model', UserActionsViewSet, basename='your-model')
   # 订单相关Action的ViewSet
   router.register(r'your-model', OrderActionsViewSet, basename='your-model')
   

注意事项 #

• 拆分时确保每个Mixin或独立ViewSet能正确访问原ViewSet的核心属性（如queryset、serializer_class），自定义属性需在Mixin中显式定义或通过继承传递。
• 避免不同模块的Action命名重复，否则会出现覆盖问题。
• 调试时可直接定位到对应业务模块的文件，大幅提升代码导航效率。

内容的提问来源于stack exchange，提问作者Chandra Kiran Guntur