Question

我正在使用Django 2.0

我必须从压缩文件中压缩的csv格式的多个模型下载数据。

我正在使用django-import-export插件为模型数据生成csv文件

我正在做的是

def download_all_data(request):
    # user data
    user_resource = UserResource()
    queryset = User.objects.filter(username=request.user)
    data_set = user_resource.export(queryset)

    # favourite arbitrase
    favourite_arbitrase_resource = FavouriteArbitraseResource()
    favourite_arbitrase_queryset = FavouriteArbitrase.objects.filter(user=request.user)
    data_set_favourite_arbitrase = favourite_arbitrase_resource.export(favourite_arbitrase_queryset)

    response_user_data = HttpResponse(data_set.csv, content_type='text/csv')
    response_user_data['Content-Disposition'] = 'attachment; filename="user_data.csv"'

    response_favourite_arbitrase = HttpResponse(data_set_favourite_arbitrase.csv, content_type='text/csv')
    response_favourite_arbitrase['Content-Disposition'] = 'attachment; filename="favourite_arbitrase_data.csv"'

    zipped_file = io.BytesIO()
    with zipfile.ZipFile(zipped_file, 'w') as zip_file:
        zip_file.writestr('text/csv', response_user_data)
        zip_file.writestr('text/csv', response_favourite_arbitrase)

    zip_response = HttpResponse(zipped_file, content_type='application/octet-stream')
    zip_response['Content-Disposition'] = 'attachment; filename=my_file.zip'
    return zip_response

这是错误

object of type 'HttpResponse' has no len()

Answer 1

您需要将实际数据放入zip中，而不是捆绑的HttpResponse。

根据ZipFile.writestr docs，第一个参数是用于数据的名称。

你可以这样：

zipped_file = io.BytesIO()
with zipfile.ZipFile(zipped_file, 'w') as zip_file:
    zip_file.writestr('user.csv', data_set.csv)
    zip_file.writestr('favourite.csv', data_set_favourite_arbitrase.csv)

下载包含csv格式的多个模型数据的zip文件

1 个答案: