如何捕获yt-dlp下载过程的控制台输出并存储到变量中

嘿，我来帮你搞定这个问题～ yt-dlp其实自带了很方便的日志捕获机制，不需要绕远路去搞输出重定向，这里给你两种实用的实现方式：

方法一：用yt-dlp内置的Logger捕获（推荐） #

这种方法最贴合yt-dlp的设计，能精准区分不同级别的日志（调试信息、警告、错误），而且不会干扰程序其他部分的输出。

你只需要定义一个简单的日志捕获类，然后把它传给yt-dlp的配置选项就行：

import yt_dlp

# 自定义日志捕获类，实现yt-dlp要求的三个方法
class LogCollector:
    def __init__(self):
        self.all_logs = []
    
    def debug(self, msg):
        # 捕获普通的调试输出（对应控制台的常规信息）
        self.all_logs.append(msg)
    
    def warning(self, msg):
        # 捕获警告信息，加上标识方便区分
        self.all_logs.append(f"WARNING: {msg}")
    
    def error(self, msg):
        # 捕获错误信息
        self.all_logs.append(f"ERROR: {msg}")

# 初始化日志收集器
log_collector = LogCollector()

URL = "https://www.youtube.com/watch?v=YV2tUZIZ3Bs"

ydl_opts = {
    "format": f"bestvideo[height<=144]",
    "merge_output_format": "mp4",
    "outtmpl": "/home/lankergd/PycharmProjects/LankerTools/films_test/yez.mp4",
    "quiet": False,  # 保持False不影响，日志会被我们的collector捕获
    "logger": log_collector  # 关键：把自定义日志器传入配置
}

with yt_dlp.YoutubeDL(ydl_opts) as ydl:
    ydl.download(URL)

# 现在所有日志都存在all_logs列表里，转成字符串存到变量
captured_logs = "\n".join(log_collector.all_logs)
# 可以打印验证或者直接使用这个变量
print("捕获到的完整日志：\n", captured_logs)

方法二：重定向标准输出/错误流 #

如果你想要快速捕获所有控制台输出（包括其他代码的print），可以用Python的StringIO来重定向stdout和stderr：

import yt_dlp
import sys
from io import StringIO

# 先保存原来的输出流，用完要恢复
original_stdout = sys.stdout
original_stderr = sys.stderr

# 创建一个字符串缓冲区来存输出
output_buffer = StringIO()
sys.stdout = output_buffer
sys.stderr = output_buffer

URL = "https://www.youtube.com/watch?v=YV2tUZIZ3Bs"

ydl_opts = {
    "format": f"bestvideo[height<=144]",
    "merge_output_format": "mp4",
    "outtmpl": "/home/lankergd/PycharmProjects/LankerTools/films_test/yez.mp4",
    "quiet": False
}

try:
    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        ydl.download(URL)
finally:
    # 一定要恢复原来的输出流，不然之后的print会失效
    sys.stdout = original_stdout
    sys.stderr = original_stderr

# 从缓冲区获取所有捕获到的内容
captured_output = output_buffer.getvalue()
output_buffer.close()

# 查看结果
print("捕获到的所有控制台输出：\n", captured_output)

两种方法各有优劣：第一种更可控，能筛选不同类型的日志；第二种更简单粗暴，适合快速需求。你可以根据自己的场景选择～

备注：内容来源于stack exchange，提问作者LankerBG