我有一个Person
模型和Address
模型。一个person
可以有多个地址,因此Address
模型看起来像:
Address(models.Model):
person = models.ForeignKey(Person, related_name="addresses")
...
我想序列化一个person
实例,我做了最简单的事情:
PersonSerializer(serializers.ModelSerializer):
class Meta:
fields = (..., "addresses", ...)
但是,假设我想在当前request.user
的基础上序列化addresses
,以便不同的当前user
查看不同的地址集,我应该如何修改代码
如果您要为Detail View执行此操作,那么也应该为Address模型实现一个序列化程序。有关嵌套序列化程序关系,请参见此示例。此外,在您的api视图中,您必须根据request.user.进行查询
如果您是为列表视图执行此操作,那么PersonSerializer应该有一个SerializerMethodField。这样,用户将能够看到自己的地址,而不能看到其他人的地址。
class PersonSerializer(serializers.ModelSerializer):
addresses = serializers.SerializerMethodField('get_addresses')
class Meta:
fields = (..., "addresses", ...)
def get_addresses(self, obj):
if obj.pk != self.context['request'].user.pk:
return ""
else:
return obj.addresses
p.S:我假设你的自定义用户模型是Person,因为你没有提供太多信息。
我假设您想要一个像/user/current/这样的静态端点,它为您提供当前请求用户的Person对象+相关的地址记录。添加这样的视图:
class CurrentUser(GenericAPIView):
model = Person
serializer_class = PersonSerializer
permission_classes = (permissions.IsAuthenticated,)
def get(self, request):
return Response(self.get_serializer(request.user).data)
只有当用户不是匿名的并且将使用您的PersonSerializer来转储请求用户时,才可以访问它。如果您相应地设置了"addresses"序列化程序字段,那么相关的地址记录就会出现,如下所述:
http://www.django-rest-framework.org/api-guide/relations/#nested-关系