在Django应用中为指定APIView选择对应数据库的方案

针对你遇到的「多视图共用模型但需要指定不同数据库」的问题，以下是几个比逐个加.using()更简洁的方案：

1. 自定义视图基类（推荐） #

创建一个通用的视图基类，把数据库指定逻辑封装进去，后续需要指定数据库的视图直接继承即可：

from rest_framework.viewsets import ModelViewSet

class DB2ModelViewSet(ModelViewSet):
    def get_queryset(self):
        # 查询时指定数据库
        return super().get_queryset().using('db2')
    
    def perform_create(self, serializer):
        # 创建数据时指定数据库
        serializer.save(using='db2')
    
    def perform_update(self, serializer):
        # 更新数据时指定数据库
        serializer.save(using='db2')
    
    def perform_destroy(self, instance):
        # 删除数据时指定数据库
        instance.delete(using='db2')

使用示例：

class UserViewSet(DB2ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

所有继承这个基类的视图，增删改查都会自动使用db2，不用重复写.using()。

2. 自定义装饰器 #

如果不想修改视图继承关系，可以用装饰器给单个视图绑定数据库：

from django.db import connection

def use_database(db_name):
    def decorator(view_func):
        def wrapped_view(*args, **kwargs):
            original_db = connection.settings_dict['NAME']
            try:
                # 切换到目标数据库
                connection.settings_dict['NAME'] = db_name
                return view_func(*args, **kwargs)
            finally:
                # 执行完恢复原数据库连接
                connection.settings_dict['NAME'] = original_db
        return wrapped_view
    return decorator

使用示例：

from rest_framework.decorators import action

@use_database('db2')
class ProductViewSet(ModelViewSet):
    queryset = Product.objects.all()
    serializer_class = ProductSerializer

    # 也可以给单个action绑定数据库
    @action(detail=False, methods=['get'])
    @use_database('db1')
    def hot_products(self, request):
        queryset = self.get_queryset()
        # 这里的查询会自动使用db1
        return Response(...)

3. 序列化层指定数据库 #

如果主要是创建/更新操作需要指定数据库，可以在序列化器里封装逻辑：

from rest_framework import serializers

class OrderSerializer(serializers.ModelSerializer):
    class Meta:
        model = Order
        fields = '__all__'
    
    def save(self, **kwargs):
        # 强制指定保存时的数据库
        kwargs['using'] = 'db2'
        return super().save(**kwargs)

这个方案适合配合视图的get_queryset方法一起用，查询时在视图里加.using()，保存时由序列化器处理。

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