Question

我有一个具有多对多连接的模型。我想在Django REST中使用这个模型。默认情况下，这样的模型是只读的，但我也想写。此外，将通过连接的信息作为嵌套模型集成到GET中会很棒。

...
class KeyDateCase(models.Model):
    ...
    diagnoses_all_icd_10 = models.ManyToManyField(
        'ICD10', through='CaseICD10Connection')
...

class CaseICD10Connection(models.Model):
    case = models.ForeignKey('KeyDateCase', on_delete=models.CASCADE)
    icd_10 = models.ForeignKey('ICD10', on_delete=models.CASCADE)
    is_primary = models.BooleanField(default = False)
    certainty = models.CharField(
        max_length=1,
        choices=CERTAINTY_CHOICES,
        default='G',
    )

class ICD10(models.Model):

    primary_key_number = models.CharField(max_length=10, primary_key=True)

    star_key_number = models.CharField(max_length=10, blank=True, null=True)

    additional_key_number = models.CharField(
        max_length=10, blank=True, null=True)

    preferred_short_description = models.CharField(max_length=128, )
...

class KeyDateCaseViewSet(viewsets.ModelViewSet):
    ???

class KeyDateCaseSerializer(serializers.ModelSerializer):
    ???

我怎样才能做到这一点？我的视图和序列化器应该是什么样的？

Answer 1

通常我通过POST到through表的间接方式解决并实现nested-create（）。如果我的答案不准确，请提供更多信息。

models.py

from django.db import models


class ICD10(models.Model):
    primary_key_number = models.CharField(max_length=10, primary_key=True)
    star_key_number = models.CharField(max_length=10, blank=True, null=True)
    additional_key_number = models.CharField(max_length=10, blank=True, null=True)
    preferred_short_description = models.CharField(max_length=128, )

    def __str__(self):
        return f'{self.primary_key_number} {self.star_key_number}'


class CaseICD10Connection(models.Model):
    case = models.ForeignKey('KeyDateCase', related_name='connections', related_query_name='key_date_cases', on_delete=models.CASCADE)
    icd_10 = models.ForeignKey('ICD10', related_name='connections', related_query_name='icd_10s', on_delete=models.CASCADE)
    is_primary = models.BooleanField(default=False)
    certainty = models.CharField(max_length=1, default='G', )


class KeyDateCase(models.Model):
    name = models.CharField(max_length=20)
    diagnose_all_icd_10 = models.ManyToManyField(ICD10, related_name='icd10s', related_query_name='icd10s',
                                                 through=CaseICD10Connection)

serializers.py

from rest_framework import serializers

from keydatecases.models import KeyDateCase, ICD10, CaseICD10Connection


class KeyDateCaseSerializer(serializers.ModelSerializer):
    class Meta:
        model = KeyDateCase
        fields = [
            'id',
            'name',
            'diagnose_all_icd_10',
        ]
        read_only_fields = ['id', 'diagnose_all_icd_10']


class ICD10Serializer(serializers.ModelSerializer):
    class Meta:
        model = ICD10
        fields = [
            'primary_key_number',
            'star_key_number',
            'additional_key_number',
            'preferred_short_description',
        ]


class CaseICD10ConnectionSerializer(serializers.ModelSerializer):
    case = KeyDateCaseSerializer()
    icd_10 = ICD10Serializer()

    class Meta:
        model = CaseICD10Connection
        fields = [
            'case',
            'icd_10',
            'is_primary',
            'certainty',
        ]

    def create(self, validated_data) -> CaseICD10Connection:
        # import ipdb;
        # ipdb.set_trace()
        # create key_date_case
        key_date_case = KeyDateCase.objects.create(**validated_data.get('case'))

        # create icd10
        icd10 = ICD10.objects.create(**validated_data.get('icd_10'))

        # create connection
        conn = CaseICD10Connection.objects.create(
            case=key_date_case, icd_10=icd10, is_primary=validated_data.get('is_primary'),
            certainty=validated_data.get('certainty')
        )
        return conn

viewsets.py

from rest_framework import viewsets

from keydatecases.api.serializers import CaseICD10ConnectionSerializer
from keydatecases.models import CaseICD10Connection


class CaseICD10ConnectionViewSet(viewsets.ModelViewSet):
    permission_classes = ()
    queryset = CaseICD10Connection.objects.all()
    serializer_class = CaseICD10ConnectionSerializer

我的存储库：
我与许多问题共享我的存储库。请不要介意。
https://github.com/elcolie/tryDj2

Answer 2

关于创建或更新嵌套对象，the documentation actually has a great example。如果可以的话，我会为你提供一个更好的。如果示例中有任何令人困惑的内容，请在此处解释。

如果您遵循此方法，您的GET请求将自动为您扩展嵌套对象。

如何在Django REST中通过多对多模型发布模型

2 个答案: