Python 3写入管道

Question

我正在尝试编写一些代码将数据放入管道,我希望解决方案是python 2.6+和3.x兼容.例:

from __future__ import print_function

import subprocess
import sys

if(sys.version_info > (3,0)):
    print ("using python3")
    def raw_input(*prmpt):
        """in python3, input behaves like raw_input in python2"""
        return input(*prmpt)

class pipe(object):
    def __init__(self,openstr):
        self.gnuProcess=subprocess.Popen(openstr.split(),
                                         stdin=subprocess.PIPE)

    def putInPipe(self,mystr):
        print(mystr, file=self.gnuProcess.stdin)

if(__name__=="__main__"):
    print("This simple program just echoes what you say (control-d to exit)")
    p=pipe("cat -")
    while(True):
        try:
            inpt=raw_input()
        except EOFError:
            break
        print('putting in pipe:%s'%inpt)
        p.putInPipe(inpt)

上面的代码适用于python 2.6但在python 3.2中失败(请注意,上面的代码主要是用2to3生成的 - 我只是稍微搞砸了它,使它与python 2.6兼容.)

Traceback (most recent call last):
  File "test.py", line 30, in <module>
   p.putInPipe(inpt)
  File "test.py", line 18, in putInPipe
   print(mystr, file=self.gnuProcess.stdin)
TypeError: 'str' does not support the buffer interface

我已经尝试了这里建议的字节函数(例如print(bytes(mystr,'ascii')), TypeError:'str'不支持缓冲区接口 但是这似乎不起作用.有什么建议吗？

Answer 1

该print函数将其参数转换为字符串表示形式,并将此字符串表示形式输出到给定文件.字符串表示始终是strPython 2.x和Python 3.x 的类型.在Python 3.x中,管道只接受bytes或缓冲对象,因此这不起作用.(即使您将bytes对象传递给print它,它也会被转换为str.)

一种解决方案是使用该write()方法(并在写入后刷新):

self.gnuProcess.stdin.write(bytes(mystr + "\n", "ascii"))
self.gnuProcess.stdin.flush()