Python 3 пишет в трубу - PullRequest
Новая
       41

Python 3 пишет в трубу

7 голосов
/ 11 мая 2011

Я пытаюсь написать код для помещения данных в канал, и я бы хотел, чтобы решение было совместимо с 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' не поддерживает интерфейс буфера Но это не похоже на работу. Есть предложения?

Ответы [ 2 ]

8 голосов
/ 11 мая 2011

Функция 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()
0 голосов
/ 22 марта 2015

но python2 будет жаловаться на 

bytes("something", "ascii")

, если вы используете изменяемый байт-массив, он будет работать как в python2, так и в python3 без изменений 

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