DRF-сериализатор с общим внешним ключом - проверьте, существует ли данный идентификатор объекта перед сохранением
Я пытаюсь найти способ проверить, является ли данный идентификатор объекта в IntegerField
существует (который используется для родового отношения в сериализаторе), как есть для PrimaryKeyRelatedField
,
Пока что я пришел с таким подходом:
models.py:
class Comment(models.Model):
person = models.ForeignKey(User, on_delete=models.CASCADE)
text = models.TextField()
created = models.DateTimeField(auto_now_add=True)
content_type = models.ForeignKey(ContentType, limit_choices_to={'pk__in': CONTENT_TYPES_PK})
object_id = models.PositiveIntegerField()
content_object = GenericForeignKey('content_type', 'object_id')
serializers.py:
class CommentSerializer(serializers.ModelSerializer):
person = UserSerializer(read_only=True, default=serializers.CurrentUserDefault())
content_type = serializer.PrimaryKeyRelatedField(queryset=ContentType.objects.filter(pk__in=CONTENT_TYPES_PK), write_only=True)
object_id = IntegerField(write_only=True)
class Meta:
model = Comment
extra_kwargs = {'created': {'read_only': True}}
exclude = ('content_object',)
def create(self, validated_data):
obj = Comment(**validated_data)
if not obj.content_object:
raise serializers.ValidationError({'object_id': ['Invalid pk "'+str(obj.object_id)+'" - object does not exist.']})
obj.save()
return obj
Но это не надежный способ сделать это, потому что он на самом деле не вызывает ошибку поля - он просто имитирует ее, и поэтому в браузере API поле не выделяется. Интересно, есть ли лучшее решение для этого? Заранее спасибо!
2 ответа
Я сделал это как-то - снял подкласс create
метод и добавил это вместо:
def validate(self, attrs):
try:
attrs['content_object'] = attrs['content_type'].model_class().objects.get(pk=attrs['object_id'])
except:
raise serializers.ValidationError({'object_id': ['Invalid pk "'+str(attrs['object_id'])+'" - object does not exist.']})
return attrs
Теперь он выделяет поле, но внешний вид отличается от PrimaryKeyRelatedField
, Я думаю, что есть некоторый код, который проверяет, откуда выбрасывается исключение, и если это из этого поля, то он показывает это по-другому, иначе это происходит по умолчанию. Я не мог придумать другого объяснения этого поведения, потому что оно вызывает то же самое ValidationError
,
Пользователь extra_kwargs просто и легко применить Я делюсь, пожалуйста, примените его После полей сериализатора.
class TestSerializer(serializers.ModelSerializer):
class Meta:
model = testModel
fields = ('id','state','city','first_name',)
extra_kwargs = {
"state": {
'allow_null': False, 'required': True,
"error_messages" : {
"null" : "State is mandatory.",
"invalid": "State should be valid id",
"incorrect_type": "State should be valid id",
"does_not_exist": "State should be valid id",
"required":"State is mandatory.",
"blank":"State is mandatory."
}
},
}