Python--django 实现文件下载功能
Python--django 实现文件下载功能
class DownloadTextObject(object):
def __init__(self, request):
self.params = None
self.request = request
self.msg = ''
self.query = self.request.query_params
self.body = self.request.data
self.page = self.request.GET.get('page', 1)
self.size = self.request.GET.get('size', 10)
self.txt_name = self.request.GET.get('txt_name', None)
self.userid = request.user.id
self.status = 200
self.result = {}
self.total = 0
def file_down(self):
"""
下载压缩文件
:return:
"""
file_name = self.txt_name # 文件名
file_path = os.path.join(FILEPATH, file_name) # 下载文件的绝对路径
if not os.path.isfile(file_path): # 判断下载文件是否存在
return HttpResponse("Sorry but Not Found the File")
def file_iterator(file_path, chunk_size=512):
"""
文件生成器,防止文件过大,导致内存溢出
:param file_path: 文件绝对路径
:param chunk_size: 块大小
:return: 生成器
"""
with open(file_path, mode='rb') as f:
while True:
c = f.read(chunk_size)
if c:
yield c
else:
break
try:
# 设置响应头
# StreamingHttpResponse将文件内容进行流式传输,数据量大可以用这个方法
response = StreamingHttpResponse(file_iterator(file_path))
# 以流的形式下载文件,这样可以实现任意格式的文件下载
response['Content-Type'] = 'application/octet-stream'
# Content-Disposition就是当用户想把请求所得的内容存为一个文件的时候提供一个默认的文件名
response['Content-Disposition'] = 'attachment;filename="{}"'.format(file_name)
except:
return HttpResponse("Sorry but Not Found the File")
return response
参考博客:https://blog.csdn.net/wen61995120/article/details/80311142
从小白到大神的蜕变~~