8

我正在尝试编写一些代码来将数据放入管道中，并且我希望该解决方案与 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 但这似乎不起作用。有什么建议吗？

4

2 回答 2

9

该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()
于 2011-05-11T14:49:34.540 回答
0

但python2会抱怨

bytes("something", "ascii")

如果您使用可变字节数组，它将在 python2 和 python3 中正常工作

self.gnuProcess.stdin.write(bytearray(mystr + "\n", "ascii"))
self.gnuProcess.stdin.flush()
于 2015-03-22T15:04:37.440 回答