python多线程下载视频

import requests
import threading

# 定义下载链接和保存路径
url = 'https://www.pidancode.com/video.mp4'
path = './video.mp4'

# 定义线程数量
num_threads = 4

# 获取文件大小
response = requests.head(url)
file_size = int(response.headers.get('Content-Length', 0))

# 定义线程函数
def download_thread(start, end, file):
    headers = {'Range': 'bytes={}-{}'.format(start, end)}
    response = requests.get(url, headers=headers, stream=True)
    for chunk in response.iter_content(chunk_size=1024):
        if chunk:
            file.write(chunk)

# 创建文件对象
with open(path, 'wb') as file:
    # 计算每个线程下载的文件块大小
    block_size = int(file_size / num_threads) + 1

    # 创建线程列表并启动
    threads = []
    for i in range(num_threads):
        start = block_size * i
        end = min(start + block_size, file_size)
        thread = threading.Thread(target=download_thread, args=(start, end, file))
        thread.start()
        threads.append(thread)

    # 等待线程执行完成
    for thread in threads:
        thread.join()

print('下载完成')