关闭

python的sys.stdout重定向

标签: python
922人阅读 评论(1) 收藏 举报
分类:
本文环境:Python 2.7 
使用 print obj 而非 print(obj)
 
一些背景
sys.stdout 与 print
当我们在 Python 中打印对象调用 print obj 时候,事实上是调用了 sys.stdout.write(obj+'\n')
print 将你需要的内容打印到了控制台,然后追加了一个换行符
print 会调用 sys.stdout 的 write 方法
以下两行在事实上等价:
sys.stdout.write('hello'+'\n') 
print 'hello'
sys.stdin 与 raw_input
当我们用 raw_input('Input promption: ') 时,事实上是先把提示信息输出,然后捕获输入
以下两组在事实上等价:
hi=raw_input('hello? ') 

print 'hello? ', #comma to stay in the same line 
hi=sys.stdin.readline()[:-1] # -1 to discard the '\n' in input stream
 
从控制台重定向到文件
原始的 sys.stdout 指向控制台
如果把文件的对象的引用赋给 sys.stdout,那么 print 调用的就是文件对象的 write 方法
f_handler=open('out.log', 'w') 
sys.stdout=f_handler 
print 'hello'
# this hello can't be viewed on concole 
# this hello is in file out.log
记住,如果你还想在控制台打印一些东西的话,最好先将原始的控制台对象引用保存下来,向文件中打印之后再恢复 sys.stdout
__console__=sys.stdout 
# redirection start # 
... 
# redirection end 
sys.stdout=__console__

同时重定向到控制台和文件
如果我们希望打印的内容一方面输出到控制台,另一方面输出到文件作为日志保存,那么该怎么办?
将打印的内容保留在内存中,而不是一打印就将 buffer 释放刷新,那么放到一个字符串区域中会怎样?
a='' 
sys.stdout=a 
print 'hello'
OK,上述代码是无法正常运行的
Traceback (most recent call last): File 
".\hello.py", line xx, in print 'hello' 
AttributeError: 'str' 
object has no attribute 'write'
错误很明显,就是上面强调过的,在尝试调用 sys.stdout.write() 的时候,发现没有 write 方法
另外,这里之所以提示 attribute error 而不是找不到函数等等,我猜想是因为 python 将对象/类的函数指针记录作为对象/类的一个属性来对待,只是保留了函数的入口地址
既然这样,那么我们必须给重定向到的对象实现一个 write 方法:
import sys 
class __redirection__: 
  def __init__(self): 
    self.buff='' 
    self.__console__=sys.stdout 

  def write(self, output_stream): 
    self.buff+=output_stream 
  
  def to_console(self): 
    sys.stdout=self.__console__ 
    print self.buff 
  
  def to_file(self, file_path):
    f=open(file_path,'w') 
    sys.stdout=f 
    print self.buff 
    f.close() 

  def flush(self): 
    self.buff='' 
 
  def reset(self): 
    sys.stdout=self.__console__ 

if __name__=="__main__": 
  # redirection 
  r_obj=__redirection__() 
  sys.stdout=r_obj

  # get output stream 
  print 'hello' 
  print 'there' 

  # redirect to console 
  r_obj.to_console() 

  # redirect to file 
  r_obj.to_file('out.log') 

  # flush buffer
  r_obj.flush() 

  # reset 
  r_obj.reset()
同样的,sys.stderr, sys.stdin 也都可以被重定向到多个地址,举一反三的事情就自己动手实践吧

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:82456次
    • 积分:3262
    • 等级:
    • 排名:第10819名
    • 原创:248篇
    • 转载:26篇
    • 译文:1篇
    • 评论:7条
    最新评论