DRF动态ViewSet中如何将类实例数据传入装饰器？

针对你遇到的@swagger_auto_schema和@action装饰器在类加载阶段无法访问实例self数据的问题，这里提供几个实用的解决方案：

方案1：利用装饰器参数的可调用性延迟获取实例数据 #

Swagger的swagger_auto_schema允许参数接收可调用对象，这些对象会在视图实例化后执行，此时可以直接访问self相关属性。

示例代码：

from drf_yasg.utils import swagger_auto_schema
from rest_framework import viewsets

# 定义动态获取请求体的可调用函数
def dynamic_request_body(view_instance):
    # 这里可以访问view_instance的serializer_class、table_name等实例属性
    return view_instance.serializer_class

# 定义动态响应模型的可调用函数
def dynamic_response(view_instance):
    return {
        200: view_instance.serializer_class,
        400: openapi.Response("错误响应", openapi.Schema(
            type=openapi.TYPE_OBJECT,
            properties={"detail": openapi.Schema(type=openapi.TYPE_STRING)}
        ))
    }

class DynamicCrud(viewsets.ModelViewSet):
    @swagger_auto_schema(
        request_body=dynamic_request_body,
        responses=dynamic_response,
        operation_description=lambda view: f"创建{view.table_name}数据"
    )
    def create(self, request, *args, **kwargs):
        return super().create(request, *args, **kwargs)

    # 同理处理其他CRUD方法或自定义action
    @swagger_auto_schema(
        responses=dynamic_response,
        operation_description=lambda view: f"获取{view.table_name}详情"
    )
    def retrieve(self, request, *args, **kwargs):
        return super().retrieve(request, *args, **kwargs)

方案2：重写ViewSet的schema生成方法自定义配置 #

直接绕过装饰器的类加载限制，通过重写ViewSet的get_view_schema方法，在运行时动态生成Swagger schema：

from drf_yasg import openapi
from rest_framework import viewsets

class DynamicCrud(viewsets.ModelViewSet):
    def get_view_schema(self, request):
        # 调用父类方法获取基础schema
        schema = super().get_view_schema(request)
        
        # 动态修改schema属性，这里可直接访问self的所有实例数据
        schema.operation_id = f"{self.table_name}_{schema.operation_id.split('_')[-1]}"
        schema.description = f"{schema.description}（对应表：{self.table_name}）"
        
        # 动态设置请求/响应模型
        if schema.request_body:
            schema.request_body = openapi.Schema(
                type=openapi.TYPE_OBJECT,
                properties={field: field.schema for field in self.serializer_class().get_fields().values()}
            )
        
        return schema

# 子类正常继承即可，无需额外装饰
class TblMCategories(DynamicCrud):
    serializer_class = TblMCategoriesSerializer
    table_name = "tbl_m_categories"
    queryset = TblMCategories.objects.all()

方案3：用类装饰器给子类动态添加Swagger配置 #

如果父类不想绑定固定的装饰逻辑，可以给子类添加类装饰器，在子类加载时读取其类属性并生成对应装饰器：

from drf_yasg.utils import swagger_auto_schema

def dynamic_swagger(cls):
    # 读取子类的类属性
    serializer_cls = cls.serializer_class
    table_name = cls.table_name
    
    # 给create方法添加装饰器
    cls.create = swagger_auto_schema(
        request_body=serializer_cls,
        responses={200: serializer_cls},
        operation_description=f"创建{table_name}记录"
    )(cls.create)
    
    # 给自定义action添加装饰器
    if hasattr(cls, 'batch_update'):
        cls.batch_update = swagger_auto_schema(
            request_body=serializer_cls(many=True),
            operation_description=f"批量更新{table_name}记录"
        )(cls.batch_update)
    
    return cls

# 子类使用类装饰器
@dynamic_swagger
class TblMCategories(DynamicCrud):
    serializer_class = TblMCategoriesSerializer
    table_name = "tbl_m_categories"
    queryset = TblMCategories.objects.all()
    
    @action(detail=False, methods=['post'])
    def batch_update(self, request):
        # 批量更新逻辑
        pass

关键提示 #

• 如果你的serializer_class是通过get_serializer_class()动态返回的（而非类属性），优先选择方案1，因为它能在运行时获取实例实际使用的序列化器。
• 所有需要动态获取的实例数据，都要延迟到视图实例化后处理，避免在类加载阶段直接引用self。

内容的提问来源于stack exchange，提问作者Risky F