繁体   English   中英

Django Rest Framework:序列化模型并按其他数据的外键分组

[英]Django Rest Framework: Serializing a model and grouping by a foreign key for other data

我有以下三个模型:

class Package(models.Model):
  name = models.CharField()

class Part(models.Model):
  name = models.CharField()
  sku = models.IntegerField()

class Member(models.Model):
  name = models.CharField()

class MemberPurchase(models.Model):
  member = models.ForeignKey(Member)
  package = model.ForeignKey(Package)
  part = models.ForeignKey(Part)
  purchase_date = models.DateField()

我想序列化 API 的 MemberPurchase 模型,以便我可以获得所需的所有信息,如下所示:

{
  "name": "John Doe",
  "id": 123,
  "purchases": [
     {
       "name": "Package Silver",
       "parts": [
          {
            "name": "Part Silver 1",
            "sku": "12345",
          },
          {
            "name": "Part Silver 2",
            "sku": "145678",
          }
       ]
     },
     {
       "name": "Package Gold",
       "parts": [
           {
            "name": "Part Gold 1",
            "sku": "1010101",
          }
       ]
     }
  ]
}

我不太确定如何构建序列化程序来给我上面的结果。 我可以编写一个序列化程序来序列化来自MemberPurchase但这将有很多重复的数据。

我猜这需要多个序列化程序:

class MemberSerializer(serializers.ModelSerilaizer):
  purchases = MemberPurchaseSerializer(source='memberpurchase_set', many=True)
  class Meta:
    model = Member
    fields = ['name', 'id']

class MemberPurchaseSerializer(serializers.Serializer):
  name = serializers.SerializerMethodField('get_package_name')
  parts = PackagePartSerializer(many=True)
  def get_pacakage_name(self, instance):
    return instance.package.name

class PackagePartSerializer(serializers.Serializer):
  name = serializers.SerializerMethodField('get_part_name')
  sku = serializers.SerializerMethodField('get_part_sku')
  def get_part_name(self, instance):
    return instance.part.name
  def get_part_sku(self, instance):
    return instance.part.sku

我认为以上不会产生我需要的东西。 有没有更好的方法来做到这一点并获得上述结果?

编辑:所以我决定走这条路,但不确定这是否是最有效的实现:

class PackagePartSerializer(serializers.Serializer):
   name = serializers.SerializerMethodField('get_part_name')
   sku = serializers.SerializerMethodField('get_part_sku')
   def get_part_name(self, instance):
     return instance.part.name
   def get_part_sku(self, instance):
     return instance.part.sku

class MemberSerializer(serializers.ModelSerilaizer):
  purchases = serializers.SerializerMethodField('get_member_purchases')

  def get_member_purchases(self, instance):
    mem_purchases = instance.memberpurchase_set.all()
    purchases = []
    for pur in mem_purchases:
      purchases.append({
        "name": pur.package.name,
        "id": pur.package.id,
        "parts": PackagePartSerializer(instance.memberpurchase_set.all(), many=True).data
      })
    return purchases

  class Meta:
     model = Member
     fields = ['name', 'id', 'purchases']

我正在提供我得到的解决方案作为答案。

class PackagePartSerializer(serializers.Serializer):
   name = serializers.SerializerMethodField('get_part_name')
   sku = serializers.SerializerMethodField('get_part_sku')
   def get_part_name(self, instance):
     return instance.part.name
   def get_part_sku(self, instance):
     return instance.part.sku

class MemberSerializer(serializers.ModelSerilaizer):
  purchases = serializers.SerializerMethodField('get_member_purchases')

  def get_member_purchases(self, instance):
    mem_purchases = instance.memberpurchase_set.all()
    purchases = []
    for pur in mem_purchases:
      purchases.append({
        "name": pur.package.name,
        "id": pur.package.id,
        "parts": PackagePartSerializer(instance.memberpurchase_set.all(), many=True).data
      })
    return purchases

  class Meta:
     model = Member
     fields = ['name', 'id', 'purchases']

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM