使用 Django Rest Framework 序列化程序为外键关系创建方法答案

【问题标题】：Create method for foreign key relationships with Django Rest Framework serializers使用 Django Rest Framework 序列化程序为外键关系创建方法
【发布时间】：2018-11-11 18:16:36
【问题描述】：

我的模型是这样的：

class FirewallPolicy(models.Model):
    name = models.CharField(max_length=100, unique=True)
    team = models.ForeignKey(Team)
    source_ip = models.ForeignKey(IP)
    destination_ip = models.ForeignKey(IP)

现在，为了创建新的防火墙策略，应该已经有一个现有的团队 source_ip 和 destination_ip。我创建新防火墙策略的有效负载如下：

{"name": "test-create-policy-911",
 "team": "avengers",
 "source_ip": "1.1.1.1",
 "destination_ip": "2.2.2.2",
}

我创建新防火墙策略的序列化程序如下：

class FirewallPolicyCreateSerializer(serializers.ModelSerializer):
    name = serializers.CharField(max_length=100)
    team = serializers.CharField(max_length=100)
    source_ip = serializers.CharField(max_length=100)
    destination_ip = serializers.CharField(max_length=100)

    class Meta:
        model = Policy
        fields = ['id', 'name', 'team', 'source_ip', 'destination_ip']

    def validate(self, data):
        try:
            Team.objects.get(name=data['team'])
            IP.objects.get(name=data['source_ip'])
            IP.objects.get(name=data['destination_ip'])
        except ObjectDoesNotExist:
            raise serializers.ValidationError("Entities must exist before you can associate it with a Firewall Policy")

    def create(self, validated_data):
        team = Team.objects.get(name=validated_data['team'])
        source_ip = IP.objects.get(name=validated_data['source_ip'])
        destination_ip = IP.objects.get(name=validated_data['destination_ip'])
        policy  = Policy.objects.create(name=validated_data['name'], 
                                        team_id=team.id, 
                                        source_ip_id = source_ip.id, 
                                        destination_ip_id = destination_ip.id )
        return policy

我不确定这是否是向模型添加外键的正确方法，因为它似乎工作量太大。序列化程序可以自动检查所有这些并添加外键，我是否遗漏了什么？

【问题讨论】：

标签： python django django-rest-framework serialization

【解决方案1】：

是的，您所做的工作比您需要的要多得多。

您应该使用SlugRelatedField 定义您的字段，以允许 DRF 从相关模型上的字段自动填充它们。所以：

class FirewallPolicyCreateSerializer(serializers.ModelSerializer):
    team = serializers.SlugRelatedField(queryset=Team.objects.all(), slug_field='name')
    source_ip = serializers.SlugRelatedField(queryset=IP.objects.all(), slug_field='source_ip')
    destination_ip = serializers.SlugRelatedField(queryset=IP.objects.all(), slug_field='destination_ip')

    class Meta:
        model = Policy
        fields = ['id', 'name', 'team', 'source_ip', 'destination_ip']

现在您根本不需要定义 validate 或 create，因为 DRF 将完成所有相关的验证和分配。

（请注意，您也不需要重新定义 name 字段，因为您不会更改基础模型字段的任何内容。）

【讨论】：

如何在序列化程序的create() 字典中实现此解决方案？