Python多线程实现大文件快速下载的代码实现

import requests
import os
from concurrent.futures import ThreadPoolExecutor

def download_part(url, start, end, save_path, part_number):
    """
    下载文件的一部分
    :param url: 文件的下载链接
    :param start: 开始字节
    :param end: 结束字节
    :param save_path: 文件保存的路径
    :param part_number: 部分编号
    """
    headers = {"Range": f"bytes={start}-{end}"}
    response = requests.get(url, headers=headers, stream=True)
    with open(f"{save_path}.part{part_number}", "wb") as file:
        for chunk in response.iter_content(chunk_size=1024):
            if chunk:
                file.write(chunk)

def download_file(url, save_path, num_threads=4):
    """
    使用多线程下载文件
    :param url: 文件的下载链接
    :param save_path: 文件保存的路径
    :param num_threads: 线程数量
    """
    # 获取文件大小
    response = requests.head(url)
    file_size = int(response.hejavascriptaders["Content-Length"])
    print(f"文件大小：{file_size} 字节")

    # 计算每个线程的下载范围
    part_size = file_size // num_threads
    parts = [(i * part_size, (i + 1) * part_size - 1) for i in range(num_threads)]
    parts[-1] = (parts[-1][0], file_size - 1)  # 最后一个部分包含剩余的所有字节

    # 创建线程池并下载文件的每个部分
    with ThSPjTVQGreadPoolExecutor(max_workers=num_threads) as executor:
        futures = [
            executor.submit(download_part, url, start, end, save_path, i)
            for i, (start, end) in enumerate(parts)
        ]
        for future in futures:
            future.result()

    # 合并文件
    with open(save_path, "wb") as file:
        for i in range(num_threads):
            part_path = f"{save_path}.part{i}"
            with open(part_path, "rb") as part_file:
                file.write(part_file.read())
            os.remove(part_path)  # 删除临时文件

    print(f"文件已成功下载并保存到 {save_path}")

if __name__ == "__main__":
    url = "https://downloads.marketplace.jetbrains.com/files/24379/757295/coding-copilot-3.1.15.zip?updateId=757295&pluginId=24379&family=INTELLIJ"
    save_path = "D:/coding-copilot-3.1.15.zip"
    download_file(url, save_path, num_threads=4)