在测试的时候,偶尔需要捕获函数的输出,如果函数运行到最后才输出的话,直接把print换成return就好了
但是函数在运行中,也会输出很多日志,我们需要把它都捕获下来,这时就需要用到IO重定向。
bash
#!/bin/bash
func (){
echo "func output."
}
exec 6>&1 # 将fd6与stdout链接起来, 保存stdout.
exec >func_output # stdout的输出重定向到func_output中去了.
func # 终端没有输出, 因为输出到文件中去了.
exec 1>&6 6>&- # 恢复stdout, 关闭fd6.
python
import sys
import StringIO
def func():
print "func output."
buf = StringIO.StringIO() # 创建IO
tmp = sys.stdout # 保存stdout
sys.stdout = buf # 将stdout替换成buf
hello() # 函数的输出到buf, 终端没有输出
sys.stdout = tmp # 恢复stdout
print "buf output=", buf.getvalue()