更新时间:2021年05月13日16时10分 来源:传智教育 浏览次数:
(1)问题分析
面试官主要想考察的是Django项目大文件下载,大多数浏览器不支持,容易报错,该怎么解决。
(2)核心问题讲解
基本的文件下载功能
def file_download(request):
# do something
with open('file_name.txt') as file:
c = file.read()
return HttpResponse(c)
以上这种方式简单粗暴,适合小文件的下载,但文件非常大,这种方式会占用大量的内存,甚至导致服务器崩溃,因此需要采用更加合理的方式。
Django的HttpResponse对象允许将迭代器作为传入参数,将上面代码中的传入参数c换成一个迭代器,便可以将上述下载功能优化为对大小文件均适合;而Django更进一步,推荐使用 StreamingHttpResponse对象取代HttpResponse对象,StreamingHttpResponse对象用于将文件流发送给浏览器,与HttpResponse对象非常相似,对于文件下载功能,使用StreamingHttpResponse对象更合理。
因此,更加合理的文件下载功能,应该先写一个迭代器,用于处理文件,然后将这个迭代器作为参数传递给StreaminghttpResponse对象,代码如下:
from django.http import StreamingHttpResponse
def big_file_download(request):
# do something
def file_iterator(file_name, chunk_size=512):
whth open(file_name) as file:
while True:
c = file.read(chunk_size)
if c:
yield c
else:
break
the_file_name = "file_name.txt"
response = StreamingHttpResponse(file_iterator(the_file_name))
return response
上面的代码已经完成了将服务器上的文件,通过文件流传输到浏览器,但文件流通常会以乱码形式显示到浏览器中,而非下载到硬盘上,因此还要进行一些优化,让文件流写入硬盘。以上代码的优化方式很简单,直接给StreamingHttpResponse对象的Content-Type和Content-Disposition字段赋下面的值即可,改后的代码如下:
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="test.pdf"'
完整代码如下:
from django.http import StreamingHttpResponse
def big_file_download(request):
# do something
def file_iterator(file_name, chunk_size=512):
whth open(file_name) as file:
while True:
c = file.read(chunk_size)
if c:
yield c
else:
break
the_file_name = "big_file.pdf"
response = StreamingHttpResponse(file_iterator(the_file_name))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{0}"'.format(the_file_name)
return response
(3)问题扩展
无
(4)结合项目中使用
Django项目大文件下载,大多数浏览器不支持,容易报错。