如何将 InMemoryUploadedFile 对象复制到磁盘

Question

我试图捕获通过表单发送的文件，并在保存之前对其执行一些操作。所以我需要在临时目录中创建该文件的副本，但我不知道如何访问它。 Shutil 的函数无法复制该文件，因为没有该文件的路径。那么有没有办法以其他方式执行此操作？

我的代码：

    image = form.cleaned_data['image']
    temp = os.path.join(settings.PROJECT_PATH, 'tmp')
    sourceFile = image.name # without .name here it wasn't working either
    import shutil
    shutil.copy(sourceFile, temp)

引发：

异常类型：IOError at /
异常值：（2，'没有这样的文件或目录'）

和调试：

#  (..)\views.py in function

  67. sourceFile = image.name
  68. import shutil
  69. shutil.copy2(sourceFile, temp) ...

# (..)\Python26\lib\shutil.py in copy2

  92. """Copy data and all stat info ("cp -p src dst").
  93.
  94. The destination may be a directory.
  95.
  96. """
  97. if os.path.isdir(dst):
  98. dst = os.path.join(dst, os.path.basename(src))  
  99. copyfile(src, dst) ... 
 100. copystat(src, dst)
 101.

▼ Local vars
Variable    Value
dst     
u'(..)\\tmp\\myfile.JPG'
src     
u'myfile.JPG'
# (..)\Python26\lib\shutil.py in copyfile

  45. """Copy data from src to dst"""
  46. if _samefile(src, dst):
  47. raise Error, "`%s` and `%s` are the same file" % (src, dst)
  48.
  49. fsrc = None
  50. fdst = None
  51. try:
  52. fsrc = open(src, 'rb') ...
  53. fdst = open(dst, 'wb')
  54. copyfileobj(fsrc, fdst)
  55. finally:
  56. if fdst:
  57. fdst.close()
  58. if fsrc:

▼ Local vars
Variable    Value
dst     
u'(..)\\tmp\\myfile.JPG'
fdst    
None
fsrc    
None
src     
u'myfile.JPG'

Answer 1

这是类似的问题，可能会有所帮助。

import os
from django.core.files.storage import default_storage
from django.core.files.base import ContentFile
from django.conf import settings

data = request.FILES['image'] # or self.files['image'] in your form

path = default_storage.save('tmp/somename.mp3', ContentFile(data.read()))
tmp_file = os.path.join(settings.MEDIA_ROOT, path)

Answer 2

正如 @Sławomir Lenart 提到的，当上传大文件时，您不希望 data.read() 堵塞系统内存。

来自 Django 文档 ：

循环 UploadedFile.chunks() 而不是使用 read() 确保
大文件不会占用系统内存

from django.core.files.storage import default_storage

filename = "whatever.xyz" # received file name
file_obj = request.data['file']

with default_storage.open('tmp/'+filename, 'wb+') as destination:
    for chunk in file_obj.chunks():
        destination.write(chunk)

这会将文件保存在 MEDIA_ROOT/tmp/ 中，就像您的 default_storage 一样，除非另有说明。

Answer 3

这是使用 python 的 mkstemp 的另一种方法：

### get the inmemory file
data = request.FILES.get('file') # get the file from the curl

### write the data to a temp file
tup = tempfile.mkstemp() # make a tmp file
f = os.fdopen(tup[0], 'w') # open the tmp file for writing
f.write(data.read()) # write the tmp file
f.close()

### return the path of the file
filepath = tup[1] # get the filepath
return filepath

Answer 4

使用 default_storage （如在此答案中提到的）不是一个好的选择（至少对我来说） ！）。我宁愿使用 <直接代码>FileSystemStorage。

from django.core.files.storage import FileSystemStorage
from django.http import HttpResponse


def simple_view(request):
    in_memory_file_obj = request.FILES["file"]
    FileSystemStorage(location="/tmp").save(in_memory_file_obj.name, in_memory_file_obj)
    return HttpResponse("Success")

注意：

• 位置可以是适合目录路径的任何值。它甚至可以是 /tmp。

• 使用 Django 4.1 测试的解决方案

Answer 5

您最好的做法是编写自定义上传处理程序。请参阅文档 。如果添加“file_complete”处理程序，则无论有内存文件还是临时路径文件，都可以访问文件的内容。您还可以使用“receive_data_chunck”方法并在其中写入您的副本。

问候

Answer 6

这就是我尝试在本地保存文件的方法

    file_object = request.FILES["document_file"]
    file_name = str(file_object)
    print(f'[INFO] File Name: {file_name}')
    with open(file_name, 'wb+') as f:
        for chunk in file_object.chunks():
            f.write(chunk)