postgresql Django Rest Framework:如何在验证序列化程序时忽略“unique”主键约束?

velaa5lx  于 2023-04-29  发布在  PostgreSQL
关注(0)|答案(2)|浏览(132)

我试图在PostgreSQL数据库中存储大量的运输数据,客户端一次上传多条记录(数万条)。我只希望每个站点、每个路线、每个行程有一个到达时间,并且该站点、路线和行程的唯一标识符是我的表的主键(以及不同表中的外键)。我正在尝试在我的序列化器中使用Django的update_or_create来创建该到达时间的条目,或者使用最新的数据更新到达时间(如果它已经存在)。不幸的是,当在我的序列化器上调用is_valid()时,它识别出重复记录违反了主键的唯一性约束(这是有道理的),给了我这个错误消息:

'real_id': [ErrorDetail(string='actual with this real id already exists.', code='unique')]

我想覆盖这个行为,因为如果主键不是唯一的,它只会更新条目。我已经尝试过循环并删除验证器,如下所示:

def run_validators(self, value):
        for validator in self.validators:
            self.validators.remove(validator)
        super(ActualSerializer, self).run_validators(value)

我还尝试使用我的序列化器 meta类的extra_kwargs字段删除所有验证器,如下所示:

extra_kwargs = {
            'name': {'validators': []}
        }

我并不想在视图中进行插入操作,因为理想情况下,序列化程序解析出值并验证其他约束。
但这些解决方案都没有改变任何事情。我找不到任何其他的东西,所以,将解决我的问题(但如果有人找到一个答案,我错过了,这将是一个天赐良机)。我想我的问题有点类似于this question,但那个问题没有一个好的答案。
作为参考,我的序列化程序文件:

from .models import Future, Actual
from rest_framework import serializers

class ActualSerializer(serializers.ModelSerializer):

    def create(self, validated_data):
        actual, created = Actual.objects.update_or_create(
            real_id=validated_data['real_id'],
            defaults={'arrival': validated_data.get('arrival', None)}
        )
        return actual

    class Meta:
        model = Actual
        fields = ['real_id', 'arrival']

class FutureSerializer(serializers.ModelSerializer):

    class Meta:
        model = Future
        fields = ['row_id', 'stop_id', 'route_id', 'record_time', 'predicted_arrival', 'delay', 'real_id']

我的模特:

from django.db import models
import json

class Actual(models.Model):
    real_id = models.CharField(max_length=100, primary_key=True)
    arrival = models.BigIntegerField(null=True)

    def __str__(self):
        return json.dumps([self.real_id, self.arrival])

class Future(models.Model):
    row_id = models.BigAutoField(primary_key=True)
    stop_id = models.CharField(max_length=20)
    route_id = models.CharField(max_length=10)
    record_time = models.BigIntegerField()
    predicted_arrival = models.IntegerField(null=True)
    delay = models.IntegerField()
    real_id = models.ForeignKey(Actual, on_delete=models.CASCADE)

    def __str__(self):
        return json.dumps([self.row_id,
                           self.stop_id,
                           self.route_id,
                           self.record_time,
                           self.predicted_arrival,
                           self.delay,
                           self.real_id])

最后,我的观点:

class ArrivalData(APIView):

    queryset = Future.objects.all()
    permission_classes = [OperationAllowed]

    def post(self, request: Request, format=None) -> Response:
        parsed_data = json.loads(request.data['data'])
        actual_arrivals = []
        for datum in parsed_data:
            saved_time = None
            if datum['predicted_arrival'] is not None and -200 < datum['predicted_arrival'] - datum['record_time'] < 0:
                saved_time = datum['predicted_arrival']

            actual_arrivals.append({
                'real_id': datum['real_id'],
                'arrival': saved_time,
            })

        actual_serializer = ActualSerializer(data=actual_arrivals, many=True)
        if actual_serializer.is_valid(raise_exception=False):
            actual_serializer.save()
        future_serializer = FutureSerializer(data=parsed_data, many=True)

        if future_serializer.is_valid():

            future_serializer.save()
            return Response("Data saved")
        
        return Response(str(actual_serializer.errors))
fquxozlt

fquxozlt1#

所以,原来唯一约束是一个字段级验证器,这就是为什么试图在类级别删除它不起作用的原因。
我在序列化器类中显式声明了该字段,但没有任何验证器real_id = serializers.CharField(validators=[]),这解决了这个问题。
这个页面最终帮助了我,如果其他人有这个问题。

hsvhsicv

hsvhsicv2#

我也曾为同样的问题而挣扎,最后对我来说,这起了作用:

from rest_framework import serializers, validators

class MySerializer(serializers.ModelSerializer):
    def __init__(self, instance=None, data=..., **kwargs):
        super().__init__(instance, data, **kwargs)
        # Horrible hack - In order to reuse all other validation logic in the model field, we need to
        # remove the Uniqueness validation. This is ok because on create we'll first check
        # if the attribute already exists / is unique.
        new_validators = filter(
            lambda validator: not isinstance(validator, validators.UniqueValidator),
            self.fields["field_i_want_to_bypass"].validators
        )
        self.fields["field_i_want_to_bypass"].validators = new_validators

请注意,在我的解决方案中,我们只处理UniqueValidator,而不是故意处理UniqueTogetherValidator

相关问题