Django Rest框架关系

时间:2020-01-09 03:54:26

标签: python django django-rest-framework

(之前的问题:Django Rest Framework nested relationship

我已经这样制作了序列化器:

serializers.py

from rest_framework import serializers, fields

from .models import Pegawai,Barang


class BarangSerializer(serializers.ModelSerializer):
    class Meta:
        model = Barang
        fields = (
                'pegawai', 
                'nama_barang',
                'harga_barang',
                )

    def to_representation(self, instance):
        rep = super().to_representation(instance)
        rep['pegawai'] = instance.pegawai.name
        return rep

class PegawaiSerializer(serializers.ModelSerializer):
    barangs = BarangSerializer(read_only=True, many=True)
    class Meta:
        model = Pegawai
        fields = (
                'id',
                'name', 
                'alias',
                'barangs',
                )

结果:

{  
"pegawai": "Ryan",
    "nama_barang": "burjo",
    "harga_barang": "1234"
},

以及在发布数据时如何在barang API中获得如下结果:

{
    "pegawai": {"id" : 1,
        "name" : "Ryan",
        "alias" : "R"}
    "nama_barang": "burjo",
    "harga_barang": "1234"
},

请帮助,加油。

2 个答案:

答案 0 :(得分:2)

您尝试过:

rep = super().to_representation(instance)

pegawai_obj = instance.pegawai
pegawai_data = {"id":pegawai_obj.id, "name":pegawai_obj.name, "alias":pegawai_obj.alias}
rep['pegawai'] = pegawai_data

return rep

但是我认为这不是最好的解决方案。

答案 1 :(得分:2)

编写额外的序列化器,并以 to_representation(..) 方法进行连接,

class PegawaiShortSerializer(serializers.ModelSerializer):
    class Meta:
        model = Pegawai
        fields = (
            'id',
            'name',
            'alias',
        )


class BarangSerializer(serializers.ModelSerializer):
    class Meta:
        model = Barang
        fields = (
            'pegawai',
            'nama_barang',
            'harga_barang',
        )

    def to_representation(self, instance):
        rep = super().to_representation(instance)
        rep['pegawai'] = PegawaiShortSerializer(instance.pegawai).data
        return rep