实时获取Python的print输出流方法

Z时代
2024-01-10
分类：综合

我的应用场景是：使用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