DRF上传多个文件

时间:2016-12-14 11:07:56

标签: python django django-rest-framework multiple-file-upload

我正在使用DRF创建具有多个附件的用户。 创建用户时,您必须上传一个或多个文件。 当我更新您加载新文件的用户(没有其他修改过的字段)时,响应旧实例。

我通过在返回实例之前强制序列化程序中的'_prefetched_objects_cache'属性来解决。

setattr(instance, '_prefetched_objects_cache', True)

这是对的吗? 你有其他解决方案吗? 感谢

有我的代码

models.py 中的

class User(models.Model):
   #field of user model

class Attachment(models.Model):
    class Meta:
        db_table = 'attachment'

    path = models.FileField()
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='attachments')
    dt_created = models.DateTimeField(auto_now_add=True, verbose_name='Created')
在serializer.py

class AttachmentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Attachment
        fields = '__all__'

class UserSerializer(serializers.ModelSerializer):
  attachments = AttachmentSerializer(many=True, read_only=True)


  def create(self, validated_data):

      user = User.objects.create(**validated_data)

      for file_item in self.initial_data.getlist('attachments'):
         c = Attachment(path=file_item, user=user)
         c.save()
      return user

  def update(self, instance, validated_data):
      for item in validated_data:
          if User._meta.get_field(item):
              setattr(instance, item, validated_data[item])

       c = Attachment(path=self.context['request'].FILES['attachments'], user=instance)
       c.save()

       instance.save()
       setattr(instance, '_prefetched_objects_cache', True)
       return instance
test.py

中的

io = StringIO.StringIO()
io.write('foo')
file = InMemoryUploadedFile(io, None, 'foo.txt', 'text', io.len, None)
file.seek(0)
self.user['attachments'] = [file, file]
data = self.user
response = self.client.post(url, data, format='multipart')
file = InMemoryUploadedFile(io, None, 'foo2.txt', 'text', io.len, None)
file.seek(0)
#url = url of user detail for update
local_user['attachments'].extend(response.data['attachments'])
local_user['attachments'].append(file)
data = local_user
response = self.client.put(path=url, data=data, format='multipart')

2 个答案:

答案 0 :(得分:0)

一段时间以后,我遇到的问题更少了。 this answer解释了我是如何解决它的。

答案 1 :(得分:0)

如果您使用表单发布到DRF,如果您使用表单发布数据,则可以将FormParserMultiPartParser一起使用

例如。

...
from rest_framework.parsers import MultiPartParser, FormParser
...

class UserView(APIView):
    parser_classes = (FormParser, MultiPartParser)

    def post(self, request):
        ...

文件将在request.datarequest.FILES下作为InMemoryUploadedFile实例可用。 您可以继续使用序列化程序中的代码,也可以继续使用更适合您的逻辑视图。

不要忘记在表单中将enctype设置为multipart/form-data,否则文件将无法正确解析。

<form action="..." method="POST" enctype="multipart/form-data">
  <input type="file" name="file-1" />
  <input type="file" name="file-2" />
</form>