MongoEngine 在使用django restframework和多个数据库时，serializer.is_valid()总是使用默认数据库进行验证

在本文中，我们将介绍如何在使用django restframework和MongoEngine时，使用多个数据库并保证serializer.is_valid()方法使用正确的数据库进行验证。

为什么serializer.is_valid()总是使用默认数据库进行验证？

在使用django restframework和MongoEngine时，我们可以通过设置DATABASES配置来指定多个数据库。然而，在使用serializer.is_valid()方法进行验证时，默认情况下它会使用默认数据库进行验证。

这是因为MongoEngine将默认数据库配置为全局变量，而且django的数据库配置是在运行时确定的。因此，即使我们在视图中指定了其他数据库，serializer.is_valid()仍然会使用默认数据库进行验证。

考虑以下场景：

class MySerializer(serializers.ModelSerializer):
    class Meta:
        model = MyModel
        fields = '__all__'

    def create(self, validated_data):
        return MyModel.objects.using('other_db').create(**validated_data)

上述代码中，我们希望在创建对象时使用名为”other_db”的数据库。然而，当我们在视图中调用serializer.is_valid()时，它仍然会使用默认数据库进行验证。

如何指定正确的数据库进行验证？

为了解决这个问题，我们需要自定义serializer，重写is_valid()方法来手动指定要使用的数据库进行验证。

class MySerializer(serializers.ModelSerializer):
    class Meta:
        model = MyModel
        fields = '__all__'

    def is_valid(self, raise_exception=False):
        self._set_model_db()
        return super().is_valid(raise_exception)

    def _set_model_db(self):
        self.Meta.model.objects._collection._document._db = 'other_db'

在上述代码中，我们重写了is_valid()方法，并在其中手动指定了要使用的数据库。通过调用_set_model_db()方法，我们将要使用的数据库设置为”other_db”。

现在，当我们在视图中调用serializer.is_valid()时，它将使用”other_db”数据库进行验证。

示例说明

为了更好地理解在多个数据库情况下的使用，我们来看一个完整的示例。

假设我们有两个数据库：默认数据库和”other_db”数据库。我们的目标是在”other_db”数据库中创建一个对象，并使用自定义的数据库进行验证。

首先，我们需要在django的settings.py文件中配置两个数据库：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'default_db'
    },
    'other_db': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'other_db'
    }
}

接下来，我们定义一个model和对应的serializer：

from mongoengine import Document, fields

class MyModel(Document):
    name = fields.StringField()
    description = fields.StringField()

from rest_framework import serializers

class MySerializer(serializers.ModelSerializer):
    class Meta:
        model = MyModel
        fields = '__all__'

    def create(self, validated_data):
        return MyModel.objects.using('other_db').create(**validated_data)

注意，我们在create()方法中使用了objects.using()来指定我们要使用的数据库。

最后，我们在视图中使用自定义的serializer来创建对象并进行验证：

from rest_framework.views import APIView
from rest_framework.response import Response

class MyView(APIView):
    def post(self, request):
        serializer = MySerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=201)
        return Response(serializer.errors, status=400)

现在，当我们发送请求来创建一个对象时，serializer.is_valid()将会使用”other_db”数据库进行验证。

总结

在使用django restframework和MongoEngine时，通过重写serializer.is_valid()方法并手动指定要使用的数据库，我们可以确保serializer.is_valid()方法使用正确的数据库进行验证。这对于在多个数据库环境中使用MongoEngine非常重要，因为默认情况下它会使用默认数据库进行验证。通过指定正确的数据库，我们可以实现更高度的灵活性和数据管理。

希望本文对于理解如何在django restframework和MongoEngine中使用多个数据库，并保证serializer.is_valid()使用正确的数据库进行验证有所帮助。