在Colab中从Google Drive直接下载并解压Zip文件的正确方法

import requests

file_id = '1fdFu5NGXe4rTLYKD5wOqk9dl-eJOefXo' # 替换为您的Google Drive文件ID
download_url = f'https://drive.google.com/uc?export=download&id={file_id}'

try:
    response = requests.get(download_url, stream=True) # 使用stream=True以处理大文件
    content_type = response.headers.get("Content-Type")
    print(f"Content-Type: {content_type}")

    if "application/zip" in content_type:
        print("链接指向的是一个Zip文件。")
    elif "text/html" in content_type:
        print("警告：链接返回的是HTML内容，而非Zip文件。请检查文件ID和共享设置。")
    else:
        print(f"链接返回的是未知内容类型: {content_type}")
    response.close() # 及时关闭连接
except requests.exceptions.RequestException as e:
    print(f"请求失败: {e}")

import os

file_id = '1fdFu5NGXe4rTLYKD5wOqk9dl-eJOefXo' # 替换为您的Google Drive文件ID
download_url = f'https://drive.google.com/uc?export=download&id={file_id}'
output_path = '/content/dataset.zip' # 指定下载文件的保存路径

# 确保目标目录存在
os.makedirs(os.path.dirname(output_path), exist_ok=True)

# 使用wget下载文件
# --no-check-certificate: 某些情况下可能需要，但请谨慎使用
# -O: 指定输出文件名
!wget --no-check-certificate -O '{output_path}' '{download_url}'

# 检查文件是否下载成功
if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
    print(f"文件 '{output_path}' 下载成功。")
else:
    print(f"文件 '{output_path}' 下载失败或为空。")

import requests
import os

file_id = '1fdFu5NGXe4rTLYKD5wOqk9dl-eJOefXo' # 替换为您的Google Drive文件ID
download_url = f'https://drive.google.com/uc?export=download&id={file_id}'
output_path = '/content/dataset_requests.zip' # 指定下载文件的保存路径

os.makedirs(os.path.dirname(output_path), exist_ok=True)

try:
    with requests.get(download_url, stream=True) as r:
        r.raise_for_status() # 检查HTTP请求是否成功
        content_type = r.headers.get("Content-Type")
        if "application/zip" not in content_type:
            print(f"错误：下载链接返回的是 '{content_type}' 而非 Zip 文件。")
        else:
            with open(output_path, 'wb') as f:
                for chunk in r.iter_content(chunk_size=8192):
                    f.write(chunk)
            print(f"文件 '{output_path}' 下载成功。")
except requests.exceptions.RequestException as e:
    print(f"下载失败: {e}")

import zipfile
import os

zip_file_path = '/content/dataset.zip' # 替换为实际下载的Zip文件路径
extract_path = '/content/extracted_data/' # 指定解压目标路径

if os.path.exists(zip_file_path):
    try:
        with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
            zip_ref.extractall(extract_path)
        print(f"文件 '{zip_file_path}' 已成功解压到 '{extract_path}'。")
        # 列出解压后的文件，验证是否成功
        print("解压后的文件/目录：")
        for item in os.listdir(extract_path):
            print(f"- {item}")
    except zipfile.BadZipFile:
        print(f"错误：'{zip_file_path}' 不是一个有效的Zip文件。")
    except Exception as e:
        print(f"解压过程中发生错误: {e}")
else:
    print(f"错误：Zip文件 '{zip_file_path}' 不存在。请确认下载是否成功。")