处理唯一一起错误的最佳方法 - Django 2.2?

Question

我知道这个问题已经被问了很多次，但我仍然无法找到正确的解决方案。 假设我有像 Follow 这样的模型

class Student(models.Model):
number = models.IntegerField()
department = models.ForeignKey(Department, on_delete=models.CASCADE)
class Meta:
constraints = [
models.UniqueConstraint(fields=['department', 'number'])
]

我的序列化程序如下所示。

class StudentModelSerializer(serializers.ModelSerializer):
class Meta:
model = Student
fields = ("number",)

在这个模型中，department和number是unique together的，现在从 url 中获取pk传递部门。 我处理独特错误的方式如下所示。

class StudentViewSet(ModelViewSet):
queryset = Student.objects.all()
serializer_class = StudentModelSerializer
def perform_create(self, serializer):
department = Department.objects.get(pk=self.kwargs['pk'])
serializer.save(department=department)
def create(self, request, *args, **kwargs):
try:
return super().create(request, *args, **kwargs)
except IntegrityError as err:
if 'UNIQUE constraint' in err.message:
raise ValidationError({
'number': 'Number field should be unique.'
})
else:
raise IntegrityError(err)

如上所示，我调用了super().create()捕获异常，然后检查UNIQUE消息，如果存在，我将再次引发验证错误，因此rest framework's exception handler处理它。 如果没有，我再次引发错误。

这种方法的问题是我正在检查消息UNIQUE的唯一错误，该错误将来可能会更改，当然我可以在保存之前将部门添加到serializer context和validate，但这可能会导致race condition，那么如何处理这样的场景best practice是什么？

Answer 1

更好的方法是将异常的 pgcode 与 psycopg2 错误代码进行比较：

from psycopg2 import errorcodes
class StudentViewSet(ModelViewSet):
queryset = Student.objects.all()
serializer_class = StudentModelSerializer
def perform_create(self, serializer):
department = Department.objects.get(pk=self.kwargs['pk'])
serializer.save(department=department)
def create(self, request, *args, **kwargs):
try:
return super().create(request, *args, **kwargs)
except IntegrityError as err:
if err.__cause__.pgcode == errorcodes.UNIQUE_VIOLATION and 
"number" in err.args[0]
raise ValidationError({
'number': 'Number field should be unique.'
})
raise