【发布时间】:2020-05-13 11:24:01
【问题描述】:
我想在我的 API 字段中输出来自两个不同(相关)模型的输出。
我有这个models.py:
class User(AbstractBaseUser):
first_name = models.CharField(max_length=30, blank=True, null=True)
surname = models.CharField(max_length=30, blank=True, null=True)
def __str__(self):
return "%s %s" % (self.first_name, self.surname) # I don't want to change this.
class Profile(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile', null=True, blank=True)
first_nameUnregistered = models.CharField(max_length=120, null=True, blank=True)
surnameUnregistered = models.CharField(max_length=120, null=True, blank=True)
这是我的序列化器:
class ProfileSerialiser(serializers.HyperlinkedModelSerializer):
user = serializers.StringRelatedField(many=False)
class Meta:
model = Profile
fields = ('first_nameUnregistered', 'surnameUnregistered', 'user')
我的看法:
class ProfileViewSet(viewsets.ModelViewSet):
queryset = Profile.objects.all()
serializer_class = ProfileSerialiser
现在当我转到 /api/profliles 端点时看到的内容:
[
{
"first_nameUnregistered": "",
"surnameUnregistered": "",
"user": "The full user name, coming from def __str__(self):"
},
[...]
我想得到类似的东西:
[
{
"first_nameUnregistered": "",
"surnameUnregistered": "",
"user.first_name": "first name"
"user.surnamee": "surname"
},
[...]
诸如profile__user.first_name 之类的查找似乎不起作用,我无法合并这两个查询集,因为它们来自不同的模型,并且我无法使用序列化程序的fields 定义中的属性。我怎样才能实现我想要的?
解决方案
感谢@Jan Giacomelli 将我指向source 关键字。我必须添加一个默认值,因为如果一个字段为空,Django Rest Framework 会抛出一个错误:
Got AttributeError when attempting to get a value for field `user__first_name` on serializer `ProfileSerialiser`.
The serializer field might be named incorrectly and not match any attribute or key on the `Profile` instance.
Original exception text was: 'NoneType' object has no attribute 'first_name'.
因此,我的完整工作代码是:
class ProfileSerialiser(serializers.HyperlinkedModelSerializer):
user__first_name = serializers.CharField(source='user.first_name', default='name')
user__surname = serializers.CharField(source='user.surname', default='name')
class Meta:
model = Profile
fields = ('id', 'first_nameUnregistered', 'surnameUnregistered', 'user__first_name', 'user__surname')
【问题讨论】:
标签: django django-models django-rest-framework django-serializer