Python 2.6 отправляет объект подключения через Queue/Pipe/etc

Учитывая эту ошибку (Python Issue 4892), которая вызывает следующую ошибку:

>>> import multiprocessing
>>> multiprocessing.allow_connection_pickling()
>>> q = multiprocessing.Queue()
>>> p = multiprocessing.Pipe()
>>> q.put(p)
>>> q.get()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File     "/.../python2.6/multiprocessing/queues.py", line 91, in get
    res = self._recv()
TypeError: Required argument 'handle' (pos 1) not found

Кто-нибудь знает об обходном пути для передачи объекта Connection в очереди?

Спасибо.

Ответы

Ответ 1

(Я считаю, что это лучший метод, после некоторых игр вокруг (у меня была такая же проблема. Хотелось пройти трубку через трубу). Прежде чем открыть этот пост:

>>> from multiprocessing import Pipe, reduction
>>> i, o = Pipe()
>>> reduced = reduction.reduce_connection(i)
>>> newi = reduced[0](*reduced[1])
>>> newi.send("hi")
>>> o.recv()
'hi'

Я не совсем уверен, почему это построено таким образом (кому-то нужно было бы понять, для чего важна часть сокращения многопроцессорности), но он определенно работает и не требует импорта pickle. Помимо этого, он довольно близок к тому, что он делает, но проще. Я также бросил это в отчет об ошибке python, чтобы другие знали об обходном пути.

Ответ 2

Здесь примерно то, что я сделал:

# Producer
from multiprocessing.reduction import reduce_connection
from multiprocessing import Pipe

   # Producer and Consumer share the Queue we call queue
def handle(queue):
   reader, writer = Pipe()
   pickled_writer = pickle.dumps(reduce_connection(writer))
   queue.put(pickled_writer)

и

# Consumer
from multiprocessing.reduction import rebuild_connection

def wait_for_request():
    pickled_write = queue.get(block=True) # block=True isn't necessary, of course
    upw = pickle.loads(pickled_writer) # unpickled writer
    writer = upw[0](upw[1][0],upw[1][1],upw[1][2])

Последняя строка является загадочной, исходя из следующего:

>>> upw
(<function rebuild_connection at 0x1005df140>,
(('/var/folders/.../pymp-VhT3wX/listener-FKMB0W',
17, False), True, True))

Надеюсь, что это поможет кому-то другому. Это отлично работает для меня.