如何从 zip 中提取 csv 文件并在 python 中将其保存到磁盘？ [复制]

Posted 2023-03-06

【中文标题】如何从 zip 中提取 csv 文件并在 python 中将其保存到磁盘？ [复制]

【英文标题】：How to extract csv file from a zip and save it to disk in python? [duplicate]

【发布时间】：2020-03-16 10:09:09

【问题描述】：

我有一个从用户那里收到的zip 文件。 zip 文件包含一个 csv 文件，我需要将其保存到服务器的目录中。我在后端使用django，这就是我所做的

from zipfile import ZipFile
from django.views.decorators.csrf import csrf_exempt
from django.http import HttpResponse

@csrf_exempt
def store_file(request):
    if request.method == "POST":
            user_id = request.POST.get("user_id")
            client_file = request.FILES['attachment']

            file_path = "/home/sr/uploads/"
            try:
                with ZipFile(client_file, 'r') as zip_ref:
                    csv_file = zip_ref.infolist()[0]

                with open(file_path + '%s' % csv_file, 'wb+') as dest:
                    for chunk in csv_file.chunks():
                        dest.write(chunk)
                # return success message
                return HttpResponse(0)
            except Exception as e:
                # return error message
                return HttpResponse(1)

但我收到此错误消息

Traceback (most recent call last):
  File "/home/sr/scripts/interact.py", line 2364, in store_file
    for chunk in csv_file.chunks():
AttributeError: 'ZipInfo' object has no attribute 'chunks'

现在这就是我通常将csv 文件存储为 csv 文件的方式。但是现在在这种情况下，csv 文件位于zip 中，因此即使从zip 中提取csv 文件后，我也无法将其保存在我的服务器中。

我做错了什么？

【参考方案1】：

infolist() 是一种返回有关存档文件的“元”信息的方法。 而不是这个，你可以使用类似的东西：

csv_file = zip_ref.namelist()[0]
data = zip_ref.read(csv_file)
with open(file_path + '%s' % csv_file, 'wb+') as dest:
    dest.write(date)

【讨论】：

为什么不直接使用 zip_ref.extractall(file_path)