实时获取Python的print输出流

最新推荐文章于 2024-11-27 10:52:24 发布

最新推荐文章于 2024-11-27 10:52:24 发布 · 1.8w 阅读

文章标签：

#python #stdout #flush

Python 专栏收录该内容

31 篇文章

订阅专栏

在shell中执行Python程序时，需实时获取print输出流。原程序无法实时显示print内容，因print将输出存于缓冲区。解决办法是通过`sys.stdout.flush()`强制刷新缓冲区，实现实时打印。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我的应用场景是：使用shell执行python文件，并且通过调用的返回值获取python的标准输出流。
shell程序如下：

cmd='python '$1' '$2' '$3' '$5' '$4
RESULT=eval $cmd
echo $RESULT

之前我的写的python程序如下：

# coding: utf-8
import time
import json

def execute(_database, _parameter):
    print 'sleep start'
    sleepTime = 30
    print 'sleep ' , sleepTime , 'second.'
    time.sleep(sleepTime)
    print 'sleep done'
    testDic={'doneCode':0,'doneMsg':'Done','logList':'success'}
    return json.dumps(testDic, ensure_ascii=False)

if __name__ == "__main__":
    p = 'param'
    db = 'databsae'
    result = execute(db, p)
    print result

之后遇到的问题是shell不能实时的获取python的print流，也就是说不是获取第一条print语句之后，休眠了30秒之后才获取最后一条print语句。
所有的print流在shell中都是一次性获取的，这种情况对于执行时间比较短的程序脚本没什么影响，但是当python程序需要执行很长时间，而需要通过print流追踪程序，就影响比较大。
通过查阅资料，可知：
当我们在 Python 中打印对象调用 print obj 时候，事实上是调用了 sys.stdout.write(obj+’\n’)
print 将你需要的内容打印到了控制台，然后追加了一个换行符
print 会调用 sys.stdout 的 write 方法
以下两行在事实上等价：

sys.stdout.write('hello'+'\n')
print 'hello'

调用sys.stdout.flush()强制其“缓冲，这意味着它会写的一切在缓冲区到终端，即使通常会在这样做之前等待。
改动后程序如下：

# coding: utf-8
import time
import json
import sys
def execute(_database, _parameter):
    print 'time 1:',time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
    print 'sleep start.'
    for i in range(1,10):
        print 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa:',i
        print 'bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb:',i*i
        print 'ccccccccccccccccccccccccccccccccccccccccccccccccccccccc:',i+i
        sys.stdout.flush()
        time.sleep(10)
    print 'sleep end '
    print 'time 2:',time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
    testDic={'doneCode':0,'doneMsg':'Done','logList':'success'}
    return json.dumps(testDic, ensure_ascii=False)

if __name__ == "__main__":
    p = 'param'
    db = 'database'
    result = execute(db, p)
    print result