在Python中，在YouTube上传代码中添加一个进度条

Question

我在编写一个脚本，它能帮助我自动化从本地驱动器上传VOD到YouTube的过程。目前，当上传开始、结束以及上传失败时，我能收到反馈信息。不过我还想添加一个进度条，以便了解上传进度，包括已上传兆字节(Mo)和剩余待上传兆字节等信息。

下面是现有的代码，我已经尝试过多种方法添加进度条，但都未能成功。我正在通过批处理文件运行这个代码。虽然这段代码并不整洁，而且在处理分块上传的部分借助了GPT的一些帮助，目前仍在改进中，但它已经能够很好地工作！

现在我想要向你展示这段代码。接下来我会尽量完善它，以便在上传过程中加入进度条。

import os
import webbrowser
import re
import google_auth_oauthlib.flow
import googleapiclient.discovery
import googleapiclient.errors
import pickle
import shutil
import time
import requests

WEBHOOK_URL = "xxxx"

def authenticate_youtube():
    # 设置环境变量并认证YouTube API
    ...

def save_token(credentials, token_file):
    # 保存访问令牌至文件
    ...

def load_token(token_file):
    # 加载访问令牌
    ...

def format_title(filename):
    # 格式化视频标题
    ...

def send_discord_webhook(message):
    # 发送Discord Webhook消息
    ...

def upload_video(youtube, file_path):
    # 上传视频并报告进度
    print(f"检测到文件：{file_path}")
    ...
    
    # 设置视频标题、描述和标签
    ...

    # 初始化可恢复上传请求
    request = youtube.videos().insert(
        ...
    )

    # 循环上传直至完成
    response = None
    while response is None:
        try:
            status, response = request.next_chunk()
            if response is not None and 'id' in response:
                video_url = f"https://www.youtube.com/watch?v={response['id']}"
                print(f"上传完成：{file_path}，视频ID为{response['id']}，地址为{video_url}")
                send_discord_webhook(f"上传完成：{title} | {video_url}")
                webbrowser.open(video_url, new=2)
            else:
                print(f"上传失败：{file_path}，未返回视频ID")
                send_discord_webhook(f"上传失败：{title}")
        except googleapiclient.errors.HttpError as e:
            print(f"上传{file_path}时发生HTTP错误{e.resp.status}：{e.content}")
            break

def main(directory_path):
    # 主函数，认证、查找并上传文件
    ...

if __name__ == "__main__":
    try:
        main(r"M:\VOD TO UPLOAD")
    except googleapiclient.errors.HttpError:
        print("上传配额已用尽！")
    except Exception as e:
        print(f"发生了错误：{e}")
        send_discord_webhook(f"发生了错误：{e}")

NPE · Answer

如果你只是想在终端中显示进度条，可以通过打印由ascii字符组成的进度条（例如 [######............]）并在每次更新时使用 \r 字符跳回行首来实现。

首先，你需要决定进度条的长度 bar_length，然后按照如下方式进行操作：

done_length = (progress_done / progress_total) * bar_length
print("\r[" + "#" * done_length + "." * (bar_length - done_length) + "]")

然后，你需要获取 progress_done 和 progress_total 的值。例如，如果视频大小为500 MB，已上传120 MB，则将 progress_total 设为500，将 progress_done 设为120。

如果你的问题是如何获取这些数值，那在这方面我可能无法直接帮你解决。

看起来你的代码中使用了一个while循环来等待上传完成，这种方式并不能提供视频已上传部分的信息，它只是等待响应而已。

你应该检查一下Google API是否能提供这样的信息，或者你所使用的库中是否有哪个能用来检查文件已经发送了多少内容。

或许你应该提供更多关于代码中如何发送请求的具体信息，以及各个库分别用于什么目的。同时，查阅相关文档也是值得考虑的做法。

Mathias Bynens · Answer

查看文档及源码可知，在你的代码中，status 变量应该是一个 MediaUploadProgress 对象。status.resumable_progress 表示已发送的字节数，而 status.total_size 则表示总的文件大小。

你可以利用这两个属性结合任何支持进度条显示的Python库来创建一个进度条。我建议使用Enlighten库，因为它原生支持在不干扰进度条的情况下进行打印输出，这样就不必像在其他一些库中那样对你的打印语句做额外改动。