本文探讨了使用Python从URL下载文件时，尤其是面对压缩文件（如ZIP）时，如何避免文件损坏并正确提取内容的最佳实践。通过requests库进行流式下载，并结合zipfile库在内存中解压，确保了数据的完整性和准确性，解决了直接下载压缩包内容导致文件损坏的问题。
1. 理解文件下载中的常见陷阱
在使用python的requests库从url下载文件时，一个常见的误区是直接假定url指向的是最终所需的文件格式（例如csv），并尝试以该格式保存。然而，服务器端提供的资源可能并非直接的csv文件，而是一个包含csv文件的压缩包（如zip文件）。如果直接将zip文件的二进制内容保存为.csv文件，那么该文件自然会因为格式不匹配而损坏，无法被相应的解析器（如pandas.read_csv）正确读取。
例如，当一个URL实际指向一个ZIP文件时，即使我们预期得到CSV文件，直接使用以下方式下载并尝试解析也会失败：
import requestsimport pandas as pddef download_and_read_csv_incorrect(url, save_path): try: response = requests.get(url, stream=True) response.raise_for_status() with open(save_path, "wb") as file: for chunk in response.iter_content(chunk_size=8192): if chunk: file.write(chunk) print(f"Downloaded {url}, saved to {save_path}") # 尝试读取，但如果save_path是ZIP文件内容，这里会报错 csv_data = pd.read_csv(save_path) print("CSV loaded successfully.") except Exception as e: print(f"Error occurred: {e}")# 假设URL实际指向一个ZIP文件# download_and_read_csv_incorrect("https://example.com/data.zip", "corrupted.csv")
上述代码的问题在于，它没有识别出URL背后是一个ZIP压缩文件，而是直接将其内容作为CSV保存，导致文件损坏。
2. 正确处理URL指向的ZIP压缩文件
要正确下载并提取包含在ZIP文件中的数据，我们需要采取以下步骤：
使用requests库以流（stream）模式下载整个ZIP文件的二进制内容。将下载的二进制内容暂存起来（例如，在内存中的临时文件）。使用Python内置的zipfile库打开并解压这个临时的ZIP文件。
以下是实现这一过程的专业代码示例：
立即学习“Python免费学习笔记（深入）”；
import requestsimport zipfileimport tempfileimport os# 示例URL，指向一个ZIP文件URL = "https://prod-dcd-datasets-cache-zipfiles.s3.eu-west-1.amazonaws.com/mpjzbtfgfr-1.zip"# 定义下载时的分块大小，有助于处理大文件CHUNK_SIZE = 32 * 1024 # 32 KBdef download_and_extract_zip(url, target_dir="."): """ 从指定URL下载ZIP文件，并在指定目录解压其内容。 Args: url (str): ZIP文件的下载URL。 target_dir (str): 解压文件存放的目标目录。 """ try: # 确保目标目录存在 os.makedirs(target_dir, exist_ok=True) print(f"开始从 {url} 下载ZIP文件...") # 以流模式发送HTTP GET请求 with requests.get(url, stream=True) as response: # 检查请求是否成功，如果状态码是4xx或5xx则抛出异常 response.raise_for_status() # 使用临时文件存储下载的ZIP内容 # tempfile.TemporaryFile() 创建一个在关闭时自动删除的临时文件， # 且在Windows上默认以二进制模式打开。 with tempfile.TemporaryFile() as temp_zip_file: # 迭代响应内容块，并写入临时文件 for chunk in response.iter_content(chunk_size=CHUNK_SIZE): if chunk: # 过滤掉保持连接的空块 temp_zip_file.write(chunk) # 刷新缓冲区，确保所有数据都已写入临时文件 temp_zip_file.flush() # 将文件指针移到文件开头，以便zipfile可以从头读取 temp_zip_file.seek(0) # 使用zipfile库打开临时文件进行解压 with zipfile.ZipFile(temp_zip_file) as zip_ref: print(f"ZIP文件内容: {zip_ref.namelist()}") # 解压所有文件到目标目录 zip_ref.extractall(path=target_dir) print(f"文件已成功解压到 {os.path.abspath(target_dir)}") except requests.exceptions.RequestException as e: print(f"网络请求错误: {e}") except zipfile.BadZipFile: print(f"错误: 下载的文件不是一个有效的ZIP文件。请检查URL或文件内容。") except Exception as e: print(f"发生未知错误: {e}")# 调用函数进行下载和解压if __name__ == "__main__": download_and_extract_zip(URL) # 示例：如果ZIP中包含CRC_clusters_neighborhoods_markers.csv， # 它将出现在当前工作目录。 # 可以进一步处理这个文件，例如使用pandas读取 # import pandas as pd # try: # df = pd.read_csv("CRC_clusters_neighborhoods_markers.csv") # print("CSV文件已成功加载到DataFrame。") # print(df.head()) # except FileNotFoundError: # print("CSV文件未找到，请检查解压是否成功及文件名。") # except Exception as e: # print(f"读取CSV文件时发生错误: {e}")
3. 代码解析与注意事项
requests.get(url, stream=True): stream=True参数至关重要。它告诉requests不要立即下载整个响应内容到内存，而是允许我们以流的方式分块读取，这对于下载大文件非常有效，可以避免内存溢出。response.raise_for_status(): 这是一个良好的实践，用于检查HTTP请求是否成功。如果响应状态码是4xx（客户端错误）或5xx（服务器错误），它将抛出一个requests.exceptions.HTTPError，方便错误处理。tempfile.TemporaryFile(): 这个函数创建了一个匿名的临时文件，通常在内存中（取决于操作系统和文件大小），并在文件对象关闭时自动删除。这避免了在磁盘上创建和管理一个中间的ZIP文件，简化了流程并减少了磁盘I/O。response.iter_content(chunk_size=CHUNK_SIZE): iter_content方法用于迭代响应内容。chunk_size参数指定了每次迭代时读取的字节数。将下载内容分块写入临时文件，进一步优化了内存使用。temp_zip_file.flush() 和 temp_zip_file.seek(0): flush()确保所有写入的数据都已提交到文件系统（或内存文件）。seek(0)将文件指针重置到文件开头，因为zipfile.ZipFile需要从文件开头读取ZIP文件的元数据。zipfile.ZipFile(temp_zip_file): zipfile模块是Python标准库的一部分，用于处理ZIP档案。这里它打开了我们下载到临时文件中的ZIP内容。zip_ref.namelist(): 返回ZIP文件中包含的所有文件和目录的名称列表，有助于了解解压后会得到什么。zip_ref.extractall(path=target_dir): 这是解压的核心方法，它将ZIP文件中的所有内容解压到指定的target_dir目录。如果target_dir不存在，os.makedirs(target_dir, exist_ok=True)会创建它。错误处理: 代码中包含了对requests.exceptions.RequestException和zipfile.BadZipFile的捕获，分别处理网络错误和文件格式错误，提高了程序的健壮性。
4. 总结
在从URL下载文件时，务必首先确认URL指向的实际资源类型。当资源是一个压缩文件（如ZIP）时，直接保存为目标文件格式会导致文件损坏。正确的做法是：
使用requests的stream=True进行流式下载。将下载的二进制内容存储到临时文件或内存中。利用zipfile（或tarfile等对应库）对压缩文件进行解压，获取最终的目标文件。
遵循这些步骤，可以确保文件下载的完整性和准确性，避免因文件格式不匹配而导致的损坏问题。
以上就是Python从URL下载文件：正确处理ZIP压缩文件以避免损坏的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1373999.html
微信扫一扫 
支付宝扫一扫 
