[python]重定向输出

最新推荐文章于 2024-06-02 16:09:13 发布

原创最新推荐文章于 2024-06-02 16:09:13 发布 · 4.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #output #command #subprocess #shell #import

python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了如何使用Python 2.5的subprocess模块来调用控制台程序并获取其标准输出。通过两个函数RunShellWithReturnCode和RunShell，实现了执行命令、获取输出及返回码的功能。

本文作者:riag
本文出处:http://blog.youkuaiyun.com/riag
声明: 本文可以不经作者同意, 任意复制, 转载, 但任何对本文的引用都请保留文章开始前三行的作者, 出处以及声明信息. 谢谢.

调用一个控制台程序，获取它的标准输出，或把它的标准输出重定向到界面上，这里只介绍如何获取它的标准输出，因为原理都一样的。

使用python2.5的subprocess模块来实现。

import sys import subprocess def RunShellWithReturnCode(command, print_output=False, universal_newlines=True): """Executes a command and returns the output from stdout and the return code. Args: command: Command to execute. print_output: If True, the output is printed to stdout. If False, both stdout and stderr are ignored. universal_newlines: Use universal_newlines flag (default: True). Returns: Tuple (output, return code) """ p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=use_shell, universal_newlines=universal_newlines) if print_output: output_array = [] while True: line = p.stdout.readline() if not line: break print line.strip("/n") output_array.append(line) output = "".join(output_array) else: output = p.stdout.read() p.wait() errout = p.stderr.read() #if print_output and errout: # print >>sys.stderr, errout p.stdout.close() p.stderr.close() return output, p.returncode def RunShell(command, silent_ok=False, universal_newlines=True, print_output=False): data, retcode = RunShellWithReturnCode(command, print_output, universal_newlines) if retcode: ErrorExit("Got error status from %s:/n%s" % (command, data)) if not silent_ok and not data: ErrorExit("No output from %s" % command) return data

上面的代码来自于开源的代码审查工具 rietveld 的代码，其中RunShellWithReturnCode函数就是返回控制台的标准输出和返回结果