39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
import os, io
|
||
from typing import List
|
||
import zipfile
|
||
from pathlib import Path
|
||
from django.conf import settings
|
||
from utils.path_utils import project_path
|
||
from utils.chen_response import ChenResponse
|
||
from django.http import FileResponse, HttpResponse
|
||
|
||
main_download_path = Path(settings.BASE_DIR) / 'media'
|
||
|
||
def get_file_respone(id: int, file_name: str | List[str]):
|
||
"""将生成文档下载响应"""
|
||
# 1.如果传入的是str，直接是文件名
|
||
if isinstance(file_name, str):
|
||
file_name = "".join([file_name, '.docx'])
|
||
file_abs_path = main_download_path / project_path(id) / 'final_seitai' / file_name
|
||
if not file_abs_path.is_file():
|
||
return ChenResponse(status=404, code=404, message="文档未生成或生成错误！")
|
||
response = FileResponse(open(file_abs_path, 'rb'))
|
||
response['Content-Type'] = 'application/octet-stream'
|
||
response['Content-Disposition'] = f"attachment; filename={file_name}.docx"
|
||
return response
|
||
# 2.如果传入的是列表，多个文件名
|
||
elif isinstance(file_name, list):
|
||
file_name_list = file_name
|
||
zip_buffer = io.BytesIO()
|
||
with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
|
||
for file_name in file_name_list:
|
||
file_name = "".join([file_name, '.docx'])
|
||
file_abs_path = main_download_path / project_path(id) / 'final_seitai' / file_name
|
||
zip_file.write(file_abs_path, os.path.basename(file_abs_path))
|
||
zip_buffer.seek(0)
|
||
response = HttpResponse(zip_buffer, content_type='application/zip')
|
||
response['Content-Disposition'] = 'attachment; filename="回归测试说明文档.zip"'
|
||
return response
|
||
else:
|
||
return ChenResponse(code=500, status=500, message='下载文档出现错误，确认是否有多个轮次内容')
|