def get_client_manager(addr, authkey): QueueManager.register('get_queue') manager = QueueManager(address=addr, authkey=to_bytes(authkey)) manager.connect() return manager
import sys import os from multiprocessing.managers import SyncManager from multiprocessing import Queue from testflo.test import Test from testflo.util import get_addr_auth_from_args, to_bytes class QueueManager(SyncManager): pass def get_client_manager(addr, authkey): QueueManager.register('get_queue') manager = QueueManager(address=addr, authkey=to_bytes(authkey)) manager.connect() return manager if __name__ == '__main__': queue = Queue() QueueManager.register('get_queue', callable=lambda:queue) address, authkey = get_addr_auth_from_args(sys.argv[1:]) manager = QueueManager(address=address, authkey=to_bytes(authkey)) server = manager.get_server() server.serve_forever()