d:\python\demo3
目标
- 简介
- 文件上传完整实例
- 文件下载关键代码
- 文件下载完整实例
简介
Django在处理文件上传的时候,文件数据被保存在request.FILES。
可以通过request.FILES['file']来读取数据
注意request.FILES 只有在请求方法为POST,并且发送请求的<form>拥有enctype="multipart/form-data" 属性时,才会包含数据。否则request.FILES 为空。
废话少说,开始编码
简单文件上传(略)
参考:http://www.ywnds.com/?p=11834
基于表单上传文件
-
form.py
class UploadFileForm(forms.Form): title = forms.CharField(max_length=50, label='标题') file = forms.FileField(label='文件')
-
views.py
from demo3.settings import BASE_DIR from demo3.form import RegisterForm, RegisterFormChinese,UploadFileForm import uuid FILE_DIR = BASE_DIR + '/files/' def upload(request): if request.method == 'POST': form = UploadFileForm(request.POST, request.FILES) if form.is_valid(): handle_uploaded_file(request.FILES['file']) return render(request, 'success.html') else: form = UploadFileForm() return render(request, 'upload.html', {'form': form}) def handle_uploaded_file(file): # file_path = FILE_DIR + str(uuid.uuid1()) file_path = FILE_DIR + file.name with open(file_path, 'wb+') as destination: for chunk in file.chunks(): destination.write(chunk)
-
urls.py里新增路径映射
path('upload/',views.upload())
-
upload.html
<form action="/forms/upload/" method="post" enctype="multipart/form-data"> {% csrf_token %} <table>{{ form }}</table> <input type="submit" value="Submit" /> </form>
-
工程里新建files目录
image.png
测试地址:http://127.0.0.1:8000/forms/upload/
-
成功后看一下目录下的文件
image.png
文件下载
基于Django建立的网站,如果提供文件下载功能,最简单的方式莫过于将静态文件交给Nginx等处理,但有些时候,由于网站本身逻辑,需要通过Django提供下载功能,如页面数据导出功能(下载动态生成的文件)、先检查用户权限再下载文件等。因此,有必要研究一下文件下载功能在Django中的实现。
- 最简单的文件下载功能的实现
将文件流放入HttpResponse对象即可,如:
def download_file(request):
# do something...
with open('/tmp/file_name.txt', 'rb') as f:
c = f.read()
return HttpResponse(c)
这种方式简单粗暴,适合小文件的下载,但如果这个文件非常大,这种方式会占用大量的内存,甚至导致服务器崩溃。
- 合理的文件下载功能(未完待续)
Django的HttpResponse对象允许将迭代器作为传入参数,将上面代码中的传入参数c换成一个迭代器,便可以将上述下载功能优化为对大小文件均适合;而Django更进一步,推荐使用 StreamingHttpResponse对象取代HttpResponse对象,StreamingHttpResponse对象用于将文件流发送给浏览器,与HttpResponse对象非常相似,对于文件下载功能,使用StreamingHttpResponse对象更合理。
因此,更加合理的文件下载功能,应该先写一个迭代器,用于处理文件,然后将这个迭代器作为参数传递给StreaminghttpResponse对象,如:
from django.http import StreamingHttpResponse
def download_file(request):
def file_iterator(file, chunk_size=512):
with open(file) as f:
while True:
c = f.read(chunk_size)
if c:
yield c
else:
break
file = "file_name.txt"
response = StreamingHttpResponse(file_iterator(file))
return response
文件下载功能再次优化
上述的代码,已经完成了将服务器上的文件,通过文件流传输到浏览器,但文件流通常会以乱码形式显示到浏览器中,而非下载到硬盘上,因此,还要在做点优化,让文件流写入硬盘。优化很简单,给StreamingHttpResponse对象的Content-Type和Content-Disposition字段赋下面的值即可,如:
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="test.pdf"'
所以,文件下载views.py完整代码如下:
from django.http import StreamingHttpResponse
def download_file(request):
def file_iterator(file, chunk_size=512):
with open(file) as f:
while True:
c = f.read(chunk_size)
if c:
yield c
else:
break
file = "big_file.pdf"
response = StreamingHttpResponse(file_iterator(file))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{0}"'.format(file)
return response
代码实例
创建一个download.html,当访问/download/地址时,将上传后的文件在页面中以列表的形式显示出来,每一行列出上传文件的文件名、上传时间,可以点击文件名下载该文件。
-
views.py
from django.http import StreamingHttpResponse
def download(request, filename):
def file_iterator(file_name, chunk_size=512):
with open(file_name) as f:
while True:
c = f.read(chunk_size)
if c:
yield c
else:
break
the_file_name = FILE_DIR + filename
response = StreamingHttpResponse(file_iterator(the_file_name))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{0}"'.format(filename)
return responsedef download_page(request):
return render(request, 'download.html') -
download.html
<body>
文件下载: test.txt <a href="/forms/test.txt">点我下载</a>
</body>
-
urls.py新增
path('download/', views.download_page),
path('<filename>', views.download) -
测试,进入以入地址,点击下载