Django Rest Framework:如何使用request.user为登录用户发布到foreignkey字段



我正在使用Django rest Framework制作一个api,供android应用程序使用。我有一个活动和地点的模型:

  class Event(models.Model):
     user=models.ForeignKey(User,related_name="%(class)s")
     title=models.CharField(max_length=256)
     venue=models.ForeignKey(Venue,related_name="%(class)s_venue")
     description=models.CharField(max_length=256)
     time=models.DateTimeField(max_length=256)
     event_pic=models.ImageField(upload_to="photos/event_pic/%Y/%m/%d",null=True,blank=True)
     album=models.ForeignKey(Album,null=True,blank=True,related_name="%(class)s_album")
     photo=models.ForeignKey(Photo,null=True,blank=True,related_name="%(class)s_photo")
     likes=GenericRelation(Like,null=True,blank=True)
     comments=GenericRelation(Comments,null=True,blank=True)

这是我的场地模型。

class Venue(models.Model):
     name=models.CharField(max_length=256)
     longitude=models.FloatField(null=True,blank=True)
     latitude=models.FloatField(null=True,blank=True)
     def __unicode__(self):
         return self.name

我想使用request.user发布到用户字段,使用name发布到场地字段。有可能做到这一点吗?如果有,我该怎么做。以下是序列化程序和视图。

class WeddingList(generics.ListCreateAPIView):    
     queryset=Wedding.objects.all()
     serializer_class=WeddingSerializer

class WeddingDetail(generics.RetrieveUpdateDestroyAPIView):
     queryset=Wedding.objects.all()
     serializer_class=WeddingSerializer


class VenueList(generics.ListCreateAPIView):
     queryset=Venue.objects.all()
     serializer_class=VenueSerializer
class VenueDetail(generics.RetrieveUpdateDestroyAPIView):
     queryset=Venue.objects.all()
     serializer_class=VenueSerializer
class VenueSerializer(serializers.ModelSerializer):
      class Meta:
           model=Venue
           fields=('name')
class WeddingSerializer(serializers.ModelSerializer):

      class Meta:
           model = Wedding
           excludes=('id',)

由于Django REST Framework 3,pre_save钩子不再可用。相反,覆盖perform_create:

def perform_create(self, serializer):
    serializer.save(user=self.request.user, ...)

在视图中添加以下代码:

def pre_save(self, obj):
    # Set here all fields that you need and that aren't already set
    obj.user = self.request.user
    obj.name = ...

并根据您的需要对其进行修改:)

最新更新