share socket object with multiprocessing in python2.7

py27_share_socket.py

import os
import time
from multiprocessing import Process, Manager
import socket
import copy_reg
from multiprocessing.reduction import rebuild_socket, reduce_socket


copy_reg.pickle(socket.socket, reduce_socket, rebuild_socket)

manager = Manager()
sockets = manager.dict()

def func():
    pid = os.getpid()
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect(('127.0.0.1', 6152))
    sockets[pid] = s
    while 1:
        print 'in child', pid, sockets
        time.sleep(1)

ps = []
for i in range(2):
    proc = Process(target=func)
    proc.start()
    ps.append(proc)

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 6152))
sockets[0] = s

for p in ps:
    p.join()