在pytube中添加进度条

from pytube import YouTube
from tqdm import tqdm

def download_video(url):
    video = YouTube(url)
    stream = video.streams.get_highest_resolution()
    file_size = stream.filesize

    # 使用tqdm创建一个进度条，并指定总大小
    progress_bar = tqdm(total=file_size, unit='bytes', unit_scale=True)

    # 定义回调函数，在下载过程中更新进度条
    def progress_callback(stream, chunk, bytes_remaining):
        # 计算已下载的大小
        bytes_downloaded = file_size - bytes_remaining
        progress_bar.update(bytes_downloaded - progress_bar.n)

    # 下载视频，并指定回调函数
    stream.download(filename='video.mp4', on_progress_callback=progress_callback)

    # 关闭进度条
    progress_bar.close()

url = 'https://www.youtube.com/watch?v=xxxxxxxx'
download_video(url)