python-3.x - 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' does not support the buffer interface
但这似乎不起作用。
有什么建议？

Answer 1

print函数将其参数转换为字符串表示，并将此字符串表示输出到给定文件。字符串表示总是类型为 str对于 Python 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()