首页 文章

用于M2M字段的Django Rest框架嵌套序列化器的更新方法

提问于
浏览
0

我的models.py中有三个模型如下:

class Service(models.Model):
     name = models.CharField(max_length=50, unique=True)
     port = models.PositiveSmallIntegerField()
     protocol = models.CharField(max_length=50)

class ServiceGroup(models.Model):
     name = models.CharField(max_length=50, unique=True)
     services = models.ManyToManyField(Service, through=ServiceToServiceGroup)

class ServiceToServiceGroup(models.Model):
    service = models.ForeignKey(Service)
    service_group = models.ForeignKey(ServiceGroup)

我创建新服务组的JSON有效负载如下:

{
    "name": "test_service_group1",
    "services":["service_1", "service_2"],
    }

由于我有一个M2M直通表,我创建新ServiceGroup的策略是首先弹出服务列表,创建名称为ServiceGroup,然后创建M2M实现 .

我的序列化程序创建一个新的ServiceGroup如下:

class ServiceGroupCreateUpdateSerializer(serializers.ModelSerializer):
    services = serializers.SlugRelatedField(queryset=Service.objects.all(), 
                                            slug_field='name', many=True)
    class Meta:
        model = ServiceGroup
        fields = ['id', 'name', 'services']

    def create(self, validated_data):
        # Pop the services list out
        services = validated_data.pop('services', None)
        # Create the ServiceGroup with the name 
        service_group = ServiceGroup.objects.create(name=validated_data['name'])
        #Create M2M associations
        for service in services:
            service_id = Service.objects.get(name=service)
            ServiceToServiceGroup.objects.create(service_id=service_id,
                                                 service_group_id= service_group.id)

我的问题是我现在如何编写更新方法?我的JSON有效负载保持不变,唯一的区别是我在URL中传递实例ID . 伪代码如下:

  • 弹出服务列表 .

  • 将名称保存到实例ID .

  • 查找链接到ServiceGroup的现有服务 .

  • 对于现有列表和JSON有效内容列表中常见的服务,不执行任何操作 .

  • 对于现有列表中的服务而不是有效负载中的服务,请删除M2M关联 .

  • 对于不在现有列表和有效负载中的服务,请创建M2M关联 .

这对于更新方法来说似乎有很多工作要做 . 有更简单的方法吗?

UPDATE

In [145]: instance = ServiceGroup.objects.get(pk=1)                                                                                                                                                                                                                      

In [146]: instance.services.all()                                                                                                                                                                                                                                        
Out[146]: <QuerySet [<Service: test-create442>]>

In [147]: new_services_list = ['test-create398']                                                                                                                                                                                                                         

In [148]: service_objects = 
Service.objects.filter(name__in=new_services_list).all()                                                                                                                                                                                     

In [149]: service_objects                                                                                                                                                                                                                                                
Out[149]: <QuerySet [<Service: test-create398>]>

In [150]: instance.service_set = service_objects                                                                                                                                                                                                                         

In [151]: instance.save()                                                                                                                                                                                                                                                

In [152]: instance.services.all()                                                                                                                                                                                                                                        
Out[152]: <QuerySet [<Service: test-create442>]>

所以,我尝试了上述内容并没有奏效 .

1 回答

  • 0

    您可以覆盖更新方法

    def update(self, instance, validated_data):
        # Pop the services list out
        services = validated_data.pop('services', None)
        instance = super().update(instance, validated_data)
        service_objects = Service.objects.filter(name__in=services).all()
        ServiceToServiceGroup.objects.filter(service_group=instance).delete()
        service_group_obj = []
        for service in service_objects:
            service_group_obj(ServiceToServiceGroup(service=service, service_group=instance))
        ServiceToServiceGroup.objects.bulk_create(service_group_obj)
        return instance
    

相关问题