通过POST请求在Django模型中创建记录

时间:2020-04-14 05:55:48

标签: django python-2.7 django-rest-framework

我是Django和Rest Framework的新手。所以我相信你们可以帮助我在这里解决我的问题。我使用的是python 2.7和Django 1.11。

我的模特-

class Item(models.Model):
    title = models.CharField(max_length=140)
    user = models.ForeignKey(User)
    created_date = models.DateTimeField(auto_now_add=True)
    due_date = models.DateTimeField(blank=True, null=True)
    completed = models.BooleanField()
    completed_date = models.DateTimeField(blank=True, null=True)

    def save(self,force_insert=False, force_update=False):
        i = datetime.datetime.now(pytz.utc)
        in_tz = pytz.timezone("Asia/Calcutta")
        i = i.astimezone(in_tz)
        # If Item is being marked complete, set the completed_date
        if self.completed:
            self.completed_date = i
        super(Item, self).save(force_insert, force_update)

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30, null=True, blank=True)
    last_name = models.CharField(max_length=30, null=True, blank=True)
    ... ... ...

我正在尝试将记录发布到Item模型,该模型应该从请求标头中获取用户电子邮件,并应该为该特定记录创建外键关系。

我的Item模型的序列化器-

class ItemSerializer(serializers.ModelSerializer):
    user = serializers.RelatedField(source='User',read_only=True)
    class Meta:
        model = Item
        fields = '__all__'

我的商品模型视图集-

class ItemViewset(viewsets.ModelViewSet):
    serializer_class = ItemSerializer
    queryset = Item.objects.all()

    def perform_create(self, serializer):
        print 'Im in here', self.request.user
        user_id = User.objects.get(email=self.request.user)
        serializer.save(user= user_id)

我已经检查了一些关于StackOverflow的博客和其他问题,并且到现在为止。

现在,当我尝试发布记录时,出现如下所示的TypeError

Got a `TypeError` when calling `Item.objects.create()`.
This may be because you have a writable field on the serializer class that is not a valid argument to `Item.objects.create()`.
You may need to make the field read-only, or override the ItemSerializer.create() method to handle this correctly.
Original exception was:
 Traceback (most recent call last):
  File "/../env/local/lib/python2.7/site-packages/rest_framework/serializers.py", line 943, in create
    instance = ModelClass._default_manager.create(**validated_data)
  File "/../env/local/lib/python2.7/site-packages/django/db/models/manager.py", line 85, in manager_method
    return getattr(self.get_queryset(), name)(*args, **kwargs)
  File "/../env/local/lib/python2.7/site-packages/django/db/models/query.py", line 394, in create
    obj.save(force_insert=True, using=self.db)
TypeError: save() got an unexpected keyword argument 'force_insert'

请帮助我解决我的用例。我没有时间完成这项工作。

非常感谢。

1 个答案:

答案 0 :(得分:1)

我认为您不需要覆盖perform_create。您所需要做的就是更改序列化器的create方法:

class ItemSerializer(serializers.ModelSerializer):
    class Meta:
        model = Item
        fields = '__all__'
        read_only_fields = ('user')

    def create(self, validated_data):
        validated_data['user'] = self.context["request"].user
        return super(ItemSerializer, self).create(validated_data)

这里,您应该能够在序列化器中获取请求对象,因为ViewSet通过extra context发送了request对象。我只是将user对象从序列化器添加到validated_data中。我还摆脱了user字段,而是将其添加到read_only_fields