![](/img/trans.png)
[英]How do I write a django-rest-framework serializer to save a nested hierarchy containing a generic model?
[英]How to write a django-rest-framework serializer / field to merge data from generic relations?
我有一个通用关系指向各种其他对象的对象,我需要它们合并(内联),因此序列化对象看起来像一个完整的对象。
例如:
class Enrollement(models.Model):
hq = models.ForeignKey(Hq)
enrollement_date = models.Datetime()
content_type = models.ForeignKey(ContentType)
object_id = models.PositiveIntegerField()
object = generic.GenericForeignKey('content_type', 'object_id')
class Nurse(models.Model):
hospital = models.ForeignKey(Hospital)
enrollement = GenericRelation(Enrollement)
class Pilot(models.Model):
plane = models.ForeignKey(plane)
enrollement = GenericRelation(Enrollement)
序列化时,我想得到这样的东西:
{
count: 50,
next: 'http...',
previous: null,
results: [
{
type: "nurse",
hq: 'http://url/to/hq-detail/view',
enrollement_date: '2003-01-01 01:01:01',
hospital: 'http://url/to/hospital-detail/view'
},
{
type: "pilot",
hq: 'http://url/to/hq-detail/view',
enrollement_date: '2003-01-01 01:01:01',
plante: 'http://url/to/plane-detail/view'
},
]
}
我可以这样做,如果可以,怎么做?
我可以嵌套一个通用关系,我可以发布处理serilizer.data以获得我想要的东西,但我想知道是否有更好的方法。
来自未来的亲朋好友:在撰写本文时,Django REST Framework团队似乎正在努力为通用关系添加更成熟的支持。 但它还没有完成。 在将此答案复制粘贴到您的代码库之前,请首先检查https://github.com/tomchristie/django-rest-framework/pull/755 ,看看它是否已合并到repo中。 可能有更优雅的解决方案等待着您。 - 你的古老祖先泰勒
鉴于您正在使用Django REST Framework ,如果您确实想要进行一些后处理(即使您似乎犹豫不决),您可以通过覆盖视图中的get_queryset
或list
来实现目标。 像这样的东西:
views.py:
from rest_framework.generics import ListAPIView
from rest_framework.response import Response
from models import *
from itertools import chain
class ResultsList(ListAPIView):
def list(self, request, *args, **kwargs):
nurses = Nurse.objects.all()
pilots = Pilot.objects.all()
results = list()
entries = list(chain(nurses, pilots)) # combine the two querysets
for entry in entries:
type = entry.__class__.__name__.lower() # 'nurse', 'pilot'
if isinstance(entry, Nurse):
serializer = NurseSerializer(entry)
hospital = serializer.data['hospital']
enrollement_date = serializer.data['enrollement.date']
hq = serializer.data['enrollement.hq']
dictionary = {'type': type, 'hospital': hospital, 'hq': hq, 'enrollement_date': enrollement_date}
if isinstance(entry, Pilot):
serializer = PilotSerializer(entry)
plane = serializer.data['plane']
enrollement_date = serializer.data['enrollement.date']
hq = serializer.data['enrollement.hq']
dictionary = {'type': type, 'plane': plane, 'hq': hq, 'enrollement_date': enrollement_date}
results.append(dictionary)
return Response(results)
serializers.py
class EnrollementSerializer(serializer.ModelSerializer):
class Meta:
model = Enrollement
fields = ('hq', 'enrollement_date')
class NurseSerializer(serializer.ModelSerializer):
enrollement = EnrollementSerializer(source='enrollement.get')
class Meta:
model = Nurse
fields = ('hospital', 'enrollement')
class PilotSerializer(serializer.ModelSerializer):
enrollement = EnrollementSerializer(source='enrollement.get')
class Meta:
model = Pilot
fields = ('plane', 'enrollement')
返回的响应看起来像:
[
{
type: "nurse",
hq: "http://url/to/hq-detail/view",
enrollement_date: "2003-01-01 01:01:01",
hospital: "http://url/to/hospital-detail/view"
},
{
type: "pilot",
hq: "http://url/to/hq-detail/view",
enrollement_date: "2003-01-01 01:01:01",
plane: "http://url/to/plane-detail/view"
},
]
值得注意的是:
source=enrollement.get
执行get
,因为否则如果我们不指定源,将返回GenericRelatedObjectManager对象。 那是因为这是一般关系所代表的。 使用.get
强制查询(如在QuerySet查询中),该查询访问您设置为泛型关系源的模型(在本例中为class Enrollement(models.Model)
。 list(chain())
而不是|
运算符,因为查询集来自不同的模型。 这就是为什么我们不能做entries = nurses | pilots
entries = nurses | pilots
。 for entry in entries
肯定会变得更加干燥。 GLHF。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.