django中响应zip文件

class TestCaseDownloadView(StaffRequiredMixin, SingleObjectMixin, View):
    model = TestCase

    @staticmethod
    def zip_dir(path, output=None):
        output = output or os.path.basename(path) + '.zip'
        zip = zipfile.ZipFile(output, 'w', zipfile.ZIP_DEFLATED)
        for root, dirs, files in os.walk(path):
            relative_root = '' if root == path else root.replace(path, '') + os.sep
            for filename in files:
                zip.write(os.path.join(root, filename), relative_root + filename)
        zip.close()

    def get(self, request, *args, **kwargs):
        testcase = self.get_object()
        snapshots = testcase.detail.snapshots
        gestures = [item['url'] for item in snapshots]

        # 创建一个临时目录
        temp_dir = tempfile.mkdtemp()

        # 将要打包的图片放入临时目录中
        for index, gesture in enumerate(gestures):
            with open('%s/%s.jpg' % (temp_dir, index + 1), 'w') as f:
                resp = requests.get(url=gesture)
                f.write(resp.content)

        zip_buffer = io.BytesIO()

        self.zip_dir(temp_dir, zip_buffer)

        # 将临时目录删除
        # os.removedirs(temp_dir)  # 删除非空目录会抛错
        shutil.rmtree(temp_dir)
		
		# 构造文件流的响应
        response = HttpResponse(content_type='application/x-zip-compressed')
        response['Content-Disposition'] = 'attachment; filename=test.zip'
        response.write(zip_buffer.getvalue())
        return response
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值