django之处理execl和zip

接收excel

@csrf_exempt
def importExecl(request):
    # Receiving excel
    if request.method != 'POST':
        return response.failResponse('Request method exception')
    try:
        excel = request.FILES.get('excel', '')

        if not excel:
            return response.failResponse('params exception')
        # open file
        data = xlrd.open_workbook(filename=None, file_contents=excel.read())

        # Get worksheet by file name, get sheet 1
        table = data.sheet_by_name('Sheet1')

        params = []
        n = table.nrows
        for i in range(1, n):
            params.append(table.row_values(i))

        return response.responseWithData('Upload Execl Successful', params)
    except Exception as e:
        # record error log
         return response.failResponse('Failed to Upload execl')

导出zip

# Export the personal zip package
@csrf_exempt
def importZip(request):
    if request.method != 'POST':
        return response.failResponse('Request method exception')
    try:
        marname = str(request.GET.POST('marname', ''))

        if not marname:
            return response.failResponse('params exception')
        
         # md5 Encrypted storage
        p_m = hashlib.md5()
        p_m.update(marname.encode())

        # storage path
        path = settings.IMAGE_ROOT + '/' + str(p_m.hexdigest()) + '/'
        
         # 判断下载文件是否存在
        if not os.path.isdir(path):
            return response.failResponse('dirs is not exist')

        zipname = marname + '.zip'  # 压缩后文件夹的path

        z = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)

        for dir_path, dir_names, file_names in os.walk(path):
            f_path = dir_path.replace(path, '')  # 这一句很重要,不replace的话,就从根目录开始复制
            f_path = f_path and f_path + os.sep or ''  # 实现当前文件夹以及包含的所有文件的压缩
            for filename in file_names:
                z.write(os.path.join(dir_path, filename), f_path + filename)
       
        z = open(zipname, 'rb')
        data = z.read()
        z.close()
        os.remove(z.name)

        res = HttpResponse(data, content_type='application/zip')
        from urllib import parse
        res['Content-Disposition'] = 'attachment;filename=' + parse.quote(zipname)

        return res
    except Exception as e:
        return response.failResponse('Failed to import zip')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值