如何在Python中管道Python进程的输出?

时间:2015-02-03 11:04:51

标签: python pipe piping youtube-dl

我正在编写一个使用youtube-dl从YouTube下载视频的程序。

我以前用subprocess打电话给youtube-dl:

import subprocess

p = subprocess.Popen([command], \
    stdout=subprocess.PIPE, \
    stderr=subprocess.STDOUT, \
    universal_newlines = True)

然后,我会阅读这个过程'通过调用输出:

for line in iter(p.stdout.readline, ""):
    hide_some_stuff_using_regex()
    show_some_stuff_using_regex()

但是,我更喜欢使用youtube-dl作为Python类。所以我现在这样做了:

from youtube_dl import YoutubeDL as youtube_dl

options = {"restrictfilenames": True, \
           "progress_with_newline": True}

ydl = youtube_dl(options)
ydl.download([url])

代码有效,但我很难找到,如何管道youtube-dl的输出。请注意,我想使用youtube-dl输出的部分内容进行实时打印,因此将sys.stdout重定向到自定义输出流将无法正常工作,因为我仍需要sys.stdout进行打印。 / p>

你能帮助我吗?

2 个答案:

答案 0 :(得分:3)

特别是对于youtube-dl,您可以设置记录器对象,如文档中的advanced example

from youtube_dl import YoutubeDL


class MyLogger(object):
    def debug(self, msg):
        print('debug information: %r' % msg)

    def warning(self, msg):
        print('warning: %r' % msg)

    def error(self, msg):
        print('error: %r' % msg)


options = {
    "restrictfilenames": True,
    "progress_with_newline": True,
    "logger": MyLogger(),
}

url = 'http://www.youtube.com/watch?v=BaW_jenozKc'
with YoutubeDL(options) as ydl:
    ydl.download([url])

答案 1 :(得分:1)

您可以尝试将sys.stdout重定向到您自己的输出流
见:https://stackoverflow.com/a/1218951/2134702


引用链接的答案:

from cStringIO import StringIO
import sys

old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()

# blah blah lots of code ...

sys.stdout = old_stdout

# examine mystdout.getvalue()

如果你想输出到stdout 在重定向期间,而不是打印使用old_stdout.write()