| 
									
										
										
										
											2008-06-13 19:20:48 +00:00
										 |  |  | # | 
					
						
							|  |  |  | # Module providing the `SyncManager` class for dealing | 
					
						
							|  |  |  | # with shared objects | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # multiprocessing/managers.py | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Copyright (c) 2006-2008, R Oudkerk --- see COPYING.txt | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | __all__ = [ 'BaseManager', 'SyncManager', 'BaseProxy', 'Token' ] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Imports | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import os | 
					
						
							|  |  |  | import sys | 
					
						
							|  |  |  | import weakref | 
					
						
							|  |  |  | import threading | 
					
						
							|  |  |  | import array | 
					
						
							|  |  |  | import copy_reg | 
					
						
							|  |  |  | import Queue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from traceback import format_exc | 
					
						
							|  |  |  | from multiprocessing import Process, current_process, active_children, Pool, util, connection | 
					
						
							|  |  |  | from multiprocessing.process import AuthenticationString | 
					
						
							|  |  |  | from multiprocessing.forking import exit, Popen, assert_spawning | 
					
						
							|  |  |  | from multiprocessing.util import Finalize, info | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | try: | 
					
						
							|  |  |  |     from cPickle import PicklingError | 
					
						
							|  |  |  | except ImportError: | 
					
						
							|  |  |  |     from pickle import PicklingError | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Register some things for pickling | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def reduce_array(a): | 
					
						
							|  |  |  |     return array.array, (a.typecode, a.tostring()) | 
					
						
							|  |  |  | copy_reg.pickle(array.array, reduce_array) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | view_types = [type(getattr({}, name)()) for name in ('items','keys','values')] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Type for identifying shared objects | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Token(object): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Type to uniquely indentify a shared object | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     __slots__ = ('typeid', 'address', 'id') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, typeid, address, id): | 
					
						
							|  |  |  |         (self.typeid, self.address, self.id) = (typeid, address, id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __getstate__(self): | 
					
						
							|  |  |  |         return (self.typeid, self.address, self.id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __setstate__(self, state): | 
					
						
							|  |  |  |         (self.typeid, self.address, self.id) = state | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return 'Token(typeid=%r, address=%r, id=%r)' % \ | 
					
						
							|  |  |  |                (self.typeid, self.address, self.id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Function for communication with a manager's server process | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def dispatch(c, id, methodname, args=(), kwds={}): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Send a message to manager using connection `c` and return response | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     c.send((id, methodname, args, kwds)) | 
					
						
							|  |  |  |     kind, result = c.recv() | 
					
						
							|  |  |  |     if kind == '#RETURN': | 
					
						
							|  |  |  |         return result | 
					
						
							|  |  |  |     raise convert_to_error(kind, result) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def convert_to_error(kind, result): | 
					
						
							|  |  |  |     if kind == '#ERROR': | 
					
						
							|  |  |  |         return result | 
					
						
							|  |  |  |     elif kind == '#TRACEBACK': | 
					
						
							|  |  |  |         assert type(result) is str | 
					
						
							|  |  |  |         return  RemoteError(result) | 
					
						
							|  |  |  |     elif kind == '#UNSERIALIZABLE': | 
					
						
							|  |  |  |         assert type(result) is str | 
					
						
							|  |  |  |         return RemoteError('Unserializable message: %s\n' % result) | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         return ValueError('Unrecognized message type') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class RemoteError(Exception): | 
					
						
							|  |  |  |     def __str__(self): | 
					
						
							|  |  |  |         return ('\n' + '-'*75 + '\n' + str(self.args[0]) + '-'*75) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Functions for finding the method names of an object | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def all_methods(obj): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Return a list of names of methods of `obj` | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     temp = [] | 
					
						
							|  |  |  |     for name in dir(obj): | 
					
						
							|  |  |  |         func = getattr(obj, name) | 
					
						
							|  |  |  |         if hasattr(func, '__call__'): | 
					
						
							|  |  |  |             temp.append(name) | 
					
						
							|  |  |  |     return temp | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def public_methods(obj): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Return a list of names of methods of `obj` which do not start with '_' | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     return [name for name in all_methods(obj) if name[0] != '_'] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Server which is run in a process controlled by a manager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Server(object): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Server class which runs in a process controlled by a manager object | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     public = ['shutdown', 'create', 'accept_connection', 'get_methods', | 
					
						
							|  |  |  |               'debug_info', 'number_of_objects', 'dummy', 'incref', 'decref'] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, registry, address, authkey, serializer): | 
					
						
							|  |  |  |         assert isinstance(authkey, bytes) | 
					
						
							|  |  |  |         self.registry = registry | 
					
						
							|  |  |  |         self.authkey = AuthenticationString(authkey) | 
					
						
							|  |  |  |         Listener, Client = listener_client[serializer] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # do authentication later | 
					
						
							|  |  |  |         self.listener = Listener(address=address, backlog=5) | 
					
						
							|  |  |  |         self.address = self.listener.address | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.id_to_obj = {0: (None, ())} | 
					
						
							|  |  |  |         self.id_to_refcount = {} | 
					
						
							|  |  |  |         self.mutex = threading.RLock() | 
					
						
							|  |  |  |         self.stop = 0 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def serve_forever(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Run the server forever | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         current_process()._manager_server = self | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 while 1: | 
					
						
							|  |  |  |                     try: | 
					
						
							|  |  |  |                         c = self.listener.accept() | 
					
						
							|  |  |  |                     except (OSError, IOError): | 
					
						
							|  |  |  |                         continue | 
					
						
							|  |  |  |                     t = threading.Thread(target=self.handle_request, args=(c,)) | 
					
						
							|  |  |  |                     t.set_daemon(True) | 
					
						
							|  |  |  |                     t.start() | 
					
						
							|  |  |  |             except (KeyboardInterrupt, SystemExit): | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             self.stop = 999 | 
					
						
							|  |  |  |             self.listener.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def handle_request(self, c): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Handle a new connection | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         funcname = result = request = None | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             connection.deliver_challenge(c, self.authkey) | 
					
						
							|  |  |  |             connection.answer_challenge(c, self.authkey) | 
					
						
							|  |  |  |             request = c.recv() | 
					
						
							|  |  |  |             ignore, funcname, args, kwds = request | 
					
						
							|  |  |  |             assert funcname in self.public, '%r unrecognized' % funcname | 
					
						
							|  |  |  |             func = getattr(self, funcname) | 
					
						
							|  |  |  |         except Exception: | 
					
						
							|  |  |  |             msg = ('#TRACEBACK', format_exc()) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 result = func(c, *args, **kwds) | 
					
						
							|  |  |  |             except Exception: | 
					
						
							|  |  |  |                 msg = ('#TRACEBACK', format_exc()) | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 msg = ('#RETURN', result) | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             c.send(msg) | 
					
						
							|  |  |  |         except Exception, e: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 c.send(('#TRACEBACK', format_exc())) | 
					
						
							|  |  |  |             except Exception: | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  |             util.info('Failure to send message: %r', msg) | 
					
						
							|  |  |  |             util.info(' ... request was %r', request) | 
					
						
							|  |  |  |             util.info(' ... exception was %r', e) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         c.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def serve_client(self, conn): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Handle requests from the proxies in a particular process/thread | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         util.debug('starting server thread to service %r', | 
					
						
							|  |  |  |                    threading.current_thread().get_name()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         recv = conn.recv | 
					
						
							|  |  |  |         send = conn.send | 
					
						
							|  |  |  |         id_to_obj = self.id_to_obj | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         while not self.stop: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 methodname = obj = None | 
					
						
							|  |  |  |                 request = recv() | 
					
						
							|  |  |  |                 ident, methodname, args, kwds = request | 
					
						
							|  |  |  |                 obj, exposed, gettypeid = id_to_obj[ident] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 if methodname not in exposed: | 
					
						
							|  |  |  |                     raise AttributeError( | 
					
						
							|  |  |  |                         'method %r of %r object is not in exposed=%r' % | 
					
						
							|  |  |  |                         (methodname, type(obj), exposed) | 
					
						
							|  |  |  |                         ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 function = getattr(obj, methodname) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 try: | 
					
						
							|  |  |  |                     res = function(*args, **kwds) | 
					
						
							|  |  |  |                 except Exception, e: | 
					
						
							|  |  |  |                     msg = ('#ERROR', e) | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     typeid = gettypeid and gettypeid.get(methodname, None) | 
					
						
							|  |  |  |                     if typeid: | 
					
						
							|  |  |  |                         rident, rexposed = self.create(conn, typeid, res) | 
					
						
							|  |  |  |                         token = Token(typeid, self.address, rident) | 
					
						
							|  |  |  |                         msg = ('#PROXY', (rexposed, token)) | 
					
						
							|  |  |  |                     else: | 
					
						
							|  |  |  |                         msg = ('#RETURN', res) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             except AttributeError: | 
					
						
							|  |  |  |                 if methodname is None: | 
					
						
							|  |  |  |                     msg = ('#TRACEBACK', format_exc()) | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     try: | 
					
						
							|  |  |  |                         fallback_func = self.fallback_mapping[methodname] | 
					
						
							|  |  |  |                         result = fallback_func( | 
					
						
							|  |  |  |                             self, conn, ident, obj, *args, **kwds | 
					
						
							|  |  |  |                             ) | 
					
						
							|  |  |  |                         msg = ('#RETURN', result) | 
					
						
							|  |  |  |                     except Exception: | 
					
						
							|  |  |  |                         msg = ('#TRACEBACK', format_exc()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             except EOFError: | 
					
						
							|  |  |  |                 util.debug('got EOF -- exiting thread serving %r', | 
					
						
							|  |  |  |                            threading.current_thread().get_name()) | 
					
						
							|  |  |  |                 sys.exit(0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             except Exception: | 
					
						
							|  |  |  |                 msg = ('#TRACEBACK', format_exc()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 try: | 
					
						
							|  |  |  |                     send(msg) | 
					
						
							|  |  |  |                 except Exception, e: | 
					
						
							|  |  |  |                     send(('#UNSERIALIZABLE', repr(msg))) | 
					
						
							|  |  |  |             except Exception, e: | 
					
						
							|  |  |  |                 util.info('exception in thread serving %r', | 
					
						
							|  |  |  |                         threading.current_thread().get_name()) | 
					
						
							|  |  |  |                 util.info(' ... message was %r', msg) | 
					
						
							|  |  |  |                 util.info(' ... exception was %r', e) | 
					
						
							|  |  |  |                 conn.close() | 
					
						
							|  |  |  |                 sys.exit(1) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def fallback_getvalue(self, conn, ident, obj): | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def fallback_str(self, conn, ident, obj): | 
					
						
							|  |  |  |         return str(obj) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def fallback_repr(self, conn, ident, obj): | 
					
						
							|  |  |  |         return repr(obj) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     fallback_mapping = { | 
					
						
							|  |  |  |         '__str__':fallback_str, | 
					
						
							|  |  |  |         '__repr__':fallback_repr, | 
					
						
							|  |  |  |         '#GETVALUE':fallback_getvalue | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def dummy(self, c): | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def debug_info(self, c): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return some info --- useful to spot problems with refcounting | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         self.mutex.acquire() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             result = [] | 
					
						
							|  |  |  |             keys = self.id_to_obj.keys() | 
					
						
							|  |  |  |             keys.sort() | 
					
						
							|  |  |  |             for ident in keys: | 
					
						
							|  |  |  |                 if ident != 0: | 
					
						
							|  |  |  |                     result.append('  %s:       refcount=%s\n    %s' % | 
					
						
							|  |  |  |                                   (ident, self.id_to_refcount[ident], | 
					
						
							|  |  |  |                                    str(self.id_to_obj[ident][0])[:75])) | 
					
						
							|  |  |  |             return '\n'.join(result) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             self.mutex.release() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def number_of_objects(self, c): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Number of shared objects | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         return len(self.id_to_obj) - 1      # don't count ident=0 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def shutdown(self, c): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Shutdown this process | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 util.debug('manager received shutdown message') | 
					
						
							|  |  |  |                 c.send(('#RETURN', None)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 if sys.stdout != sys.__stdout__: | 
					
						
							|  |  |  |                     util.debug('resetting stdout, stderr') | 
					
						
							|  |  |  |                     sys.stdout = sys.__stdout__ | 
					
						
							|  |  |  |                     sys.stderr = sys.__stderr__ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 util._run_finalizers(0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 for p in active_children(): | 
					
						
							|  |  |  |                     util.debug('terminating a child process of manager') | 
					
						
							|  |  |  |                     p.terminate() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 for p in active_children(): | 
					
						
							|  |  |  |                     util.debug('terminating a child process of manager') | 
					
						
							|  |  |  |                     p.join() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 util._run_finalizers() | 
					
						
							|  |  |  |                 util.info('manager exiting with exitcode 0') | 
					
						
							|  |  |  |             except: | 
					
						
							|  |  |  |                 import traceback | 
					
						
							|  |  |  |                 traceback.print_exc() | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             exit(0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def create(self, c, typeid, *args, **kwds): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Create a new shared object and return its id | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         self.mutex.acquire() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             callable, exposed, method_to_typeid, proxytype = \ | 
					
						
							|  |  |  |                       self.registry[typeid] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if callable is None: | 
					
						
							|  |  |  |                 assert len(args) == 1 and not kwds | 
					
						
							|  |  |  |                 obj = args[0] | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 obj = callable(*args, **kwds) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if exposed is None: | 
					
						
							|  |  |  |                 exposed = public_methods(obj) | 
					
						
							|  |  |  |             if method_to_typeid is not None: | 
					
						
							|  |  |  |                 assert type(method_to_typeid) is dict | 
					
						
							|  |  |  |                 exposed = list(exposed) + list(method_to_typeid) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             ident = '%x' % id(obj)  # convert to string because xmlrpclib | 
					
						
							|  |  |  |                                     # only has 32 bit signed integers | 
					
						
							|  |  |  |             util.debug('%r callable returned object with id %r', typeid, ident) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             self.id_to_obj[ident] = (obj, set(exposed), method_to_typeid) | 
					
						
							|  |  |  |             if ident not in self.id_to_refcount: | 
					
						
							|  |  |  |                 self.id_to_refcount[ident] = None | 
					
						
							|  |  |  |             return ident, tuple(exposed) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             self.mutex.release() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def get_methods(self, c, token): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return the methods of the shared object indicated by token | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         return tuple(self.id_to_obj[token.id][1]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def accept_connection(self, c, name): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Spawn a new thread to serve this connection | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         threading.current_thread().set_name(name) | 
					
						
							|  |  |  |         c.send(('#RETURN', None)) | 
					
						
							|  |  |  |         self.serve_client(c) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def incref(self, c, ident): | 
					
						
							|  |  |  |         self.mutex.acquire() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 self.id_to_refcount[ident] += 1 | 
					
						
							|  |  |  |             except TypeError: | 
					
						
							|  |  |  |                 assert self.id_to_refcount[ident] is None | 
					
						
							|  |  |  |                 self.id_to_refcount[ident] = 1 | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             self.mutex.release() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def decref(self, c, ident): | 
					
						
							|  |  |  |         self.mutex.acquire() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             assert self.id_to_refcount[ident] >= 1 | 
					
						
							|  |  |  |             self.id_to_refcount[ident] -= 1 | 
					
						
							|  |  |  |             if self.id_to_refcount[ident] == 0: | 
					
						
							|  |  |  |                 del self.id_to_obj[ident], self.id_to_refcount[ident] | 
					
						
							|  |  |  |                 util.debug('disposing of obj with id %d', ident) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             self.mutex.release() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Class to represent state of a manager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class State(object): | 
					
						
							|  |  |  |     __slots__ = ['value'] | 
					
						
							|  |  |  |     INITIAL = 0 | 
					
						
							|  |  |  |     STARTED = 1 | 
					
						
							|  |  |  |     SHUTDOWN = 2 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Mapping from serializer name to Listener and Client types | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | listener_client = { | 
					
						
							|  |  |  |     'pickle' : (connection.Listener, connection.Client), | 
					
						
							|  |  |  |     'xmlrpclib' : (connection.XmlListener, connection.XmlClient) | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Definition of BaseManager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class BaseManager(object): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Base class for managers | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     _registry = {} | 
					
						
							|  |  |  |     _Server = Server | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, address=None, authkey=None, serializer='pickle'): | 
					
						
							|  |  |  |         if authkey is None: | 
					
						
							|  |  |  |             authkey = current_process().get_authkey() | 
					
						
							|  |  |  |         self._address = address     # XXX not final address if eg ('', 0) | 
					
						
							|  |  |  |         self._authkey = AuthenticationString(authkey) | 
					
						
							|  |  |  |         self._state = State() | 
					
						
							|  |  |  |         self._state.value = State.INITIAL | 
					
						
							|  |  |  |         self._serializer = serializer | 
					
						
							|  |  |  |         self._Listener, self._Client = listener_client[serializer] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __reduce__(self): | 
					
						
							|  |  |  |         return type(self).from_address, \ | 
					
						
							|  |  |  |                (self._address, self._authkey, self._serializer) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def get_server(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return server object with serve_forever() method and address attribute | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         assert self._state.value == State.INITIAL | 
					
						
							|  |  |  |         return Server(self._registry, self._address, | 
					
						
							|  |  |  |                       self._authkey, self._serializer) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def connect(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Connect manager object to the server process | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Listener, Client = listener_client[self._serializer] | 
					
						
							|  |  |  |         conn = Client(self._address, authkey=self._authkey) | 
					
						
							|  |  |  |         dispatch(conn, None, 'dummy') | 
					
						
							|  |  |  |         self._state.value = State.STARTED | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def start(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Spawn a server process for this manager object | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         assert self._state.value == State.INITIAL | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # pipe over which we will retrieve address of server | 
					
						
							|  |  |  |         reader, writer = connection.Pipe(duplex=False) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # spawn process which runs a server | 
					
						
							|  |  |  |         self._process = Process( | 
					
						
							|  |  |  |             target=type(self)._run_server, | 
					
						
							|  |  |  |             args=(self._registry, self._address, self._authkey, | 
					
						
							|  |  |  |                   self._serializer, writer), | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  |         ident = ':'.join(str(i) for i in self._process._identity) | 
					
						
							|  |  |  |         self._process.set_name(type(self).__name__  + '-' + ident) | 
					
						
							|  |  |  |         self._process.start() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # get address of server | 
					
						
							|  |  |  |         writer.close() | 
					
						
							|  |  |  |         self._address = reader.recv() | 
					
						
							|  |  |  |         reader.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # register a finalizer | 
					
						
							|  |  |  |         self._state.value = State.STARTED | 
					
						
							|  |  |  |         self.shutdown = util.Finalize( | 
					
						
							|  |  |  |             self, type(self)._finalize_manager, | 
					
						
							|  |  |  |             args=(self._process, self._address, self._authkey, | 
					
						
							|  |  |  |                   self._state, self._Client), | 
					
						
							|  |  |  |             exitpriority=0 | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @classmethod | 
					
						
							|  |  |  |     def _run_server(cls, registry, address, authkey, serializer, writer): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Create a server, report its address and run it | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         # create server | 
					
						
							|  |  |  |         server = cls._Server(registry, address, authkey, serializer) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # inform parent process of the server's address | 
					
						
							|  |  |  |         writer.send(server.address) | 
					
						
							|  |  |  |         writer.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # run the manager | 
					
						
							|  |  |  |         util.info('manager serving at %r', server.address) | 
					
						
							|  |  |  |         server.serve_forever() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _create(self, typeid, *args, **kwds): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Create a new shared object; return the token and exposed tuple | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         assert self._state.value == State.STARTED, 'server not yet started' | 
					
						
							|  |  |  |         conn = self._Client(self._address, authkey=self._authkey) | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             id, exposed = dispatch(conn, None, 'create', (typeid,)+args, kwds) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             conn.close() | 
					
						
							|  |  |  |         return Token(typeid, self._address, id), exposed | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def join(self, timeout=None): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Join the manager process (if it has been spawned) | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         self._process.join(timeout) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _debug_info(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return some info about the servers shared objects and connections | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         conn = self._Client(self._address, authkey=self._authkey) | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             return dispatch(conn, None, 'debug_info') | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             conn.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _number_of_objects(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return the number of shared objects | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         conn = self._Client(self._address, authkey=self._authkey) | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             return dispatch(conn, None, 'number_of_objects') | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             conn.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __enter__(self): | 
					
						
							|  |  |  |         return self | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __exit__(self, exc_type, exc_val, exc_tb): | 
					
						
							|  |  |  |         self.shutdown() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @staticmethod | 
					
						
							|  |  |  |     def _finalize_manager(process, address, authkey, state, _Client): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Shutdown the manager process; will be registered as a finalizer | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         if process.is_alive(): | 
					
						
							|  |  |  |             util.info('sending shutdown message to manager') | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 conn = _Client(address, authkey=authkey) | 
					
						
							|  |  |  |                 try: | 
					
						
							|  |  |  |                     dispatch(conn, None, 'shutdown') | 
					
						
							|  |  |  |                 finally: | 
					
						
							|  |  |  |                     conn.close() | 
					
						
							|  |  |  |             except Exception: | 
					
						
							|  |  |  |                 pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             process.join(timeout=0.2) | 
					
						
							|  |  |  |             if process.is_alive(): | 
					
						
							|  |  |  |                 util.info('manager still alive') | 
					
						
							|  |  |  |                 if hasattr(process, 'terminate'): | 
					
						
							|  |  |  |                     util.info('trying to `terminate()` manager process') | 
					
						
							|  |  |  |                     process.terminate() | 
					
						
							|  |  |  |                     process.join(timeout=0.1) | 
					
						
							|  |  |  |                     if process.is_alive(): | 
					
						
							|  |  |  |                         util.info('manager still alive after terminate') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         state.value = State.SHUTDOWN | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             del BaseProxy._address_to_local[address] | 
					
						
							|  |  |  |         except KeyError: | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     address = property(lambda self: self._address) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @classmethod | 
					
						
							|  |  |  |     def register(cls, typeid, callable=None, proxytype=None, exposed=None, | 
					
						
							|  |  |  |                  method_to_typeid=None, create_method=True): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Register a typeid with the manager type | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         if '_registry' not in cls.__dict__: | 
					
						
							|  |  |  |             cls._registry = cls._registry.copy() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if proxytype is None: | 
					
						
							|  |  |  |             proxytype = AutoProxy | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         exposed = exposed or getattr(proxytype, '_exposed_', None) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         method_to_typeid = method_to_typeid or \ | 
					
						
							|  |  |  |                            getattr(proxytype, '_method_to_typeid_', None) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if method_to_typeid: | 
					
						
							|  |  |  |             for key, value in method_to_typeid.items(): | 
					
						
							|  |  |  |                 assert type(key) is str, '%r is not a string' % key | 
					
						
							|  |  |  |                 assert type(value) is str, '%r is not a string' % value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         cls._registry[typeid] = ( | 
					
						
							|  |  |  |             callable, exposed, method_to_typeid, proxytype | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if create_method: | 
					
						
							|  |  |  |             def temp(self, *args, **kwds): | 
					
						
							|  |  |  |                 util.debug('requesting creation of a shared %r object', typeid) | 
					
						
							|  |  |  |                 token, exp = self._create(typeid, *args, **kwds) | 
					
						
							|  |  |  |                 proxy = proxytype( | 
					
						
							|  |  |  |                     token, self._serializer, manager=self, | 
					
						
							|  |  |  |                     authkey=self._authkey, exposed=exp | 
					
						
							|  |  |  |                     ) | 
					
						
							|  |  |  |                 return proxy | 
					
						
							|  |  |  |             temp.__name__ = typeid | 
					
						
							|  |  |  |             setattr(cls, typeid, temp) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Subclass of set which get cleared after a fork | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class ProcessLocalSet(set): | 
					
						
							|  |  |  |     def __init__(self): | 
					
						
							|  |  |  |         util.register_after_fork(self, lambda obj: obj.clear()) | 
					
						
							|  |  |  |     def __reduce__(self): | 
					
						
							|  |  |  |         return type(self), () | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Definition of BaseProxy | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class BaseProxy(object): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     A base for proxies of shared objects | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     _address_to_local = {} | 
					
						
							|  |  |  |     _mutex = util.ForkAwareThreadLock() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, token, serializer, manager=None, | 
					
						
							|  |  |  |                  authkey=None, exposed=None, incref=True): | 
					
						
							|  |  |  |         BaseProxy._mutex.acquire() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             tls_idset = BaseProxy._address_to_local.get(token.address, None) | 
					
						
							|  |  |  |             if tls_idset is None: | 
					
						
							|  |  |  |                 tls_idset = util.ForkAwareLocal(), ProcessLocalSet() | 
					
						
							|  |  |  |                 BaseProxy._address_to_local[token.address] = tls_idset | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             BaseProxy._mutex.release() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # self._tls is used to record the connection used by this | 
					
						
							|  |  |  |         # thread to communicate with the manager at token.address | 
					
						
							|  |  |  |         self._tls = tls_idset[0] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # self._idset is used to record the identities of all shared | 
					
						
							|  |  |  |         # objects for which the current process owns references and | 
					
						
							|  |  |  |         # which are in the manager at token.address | 
					
						
							|  |  |  |         self._idset = tls_idset[1] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self._token = token | 
					
						
							|  |  |  |         self._id = self._token.id | 
					
						
							|  |  |  |         self._manager = manager | 
					
						
							|  |  |  |         self._serializer = serializer | 
					
						
							|  |  |  |         self._Client = listener_client[serializer][1] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if authkey is not None: | 
					
						
							|  |  |  |             self._authkey = AuthenticationString(authkey) | 
					
						
							|  |  |  |         elif self._manager is not None: | 
					
						
							|  |  |  |             self._authkey = self._manager._authkey | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             self._authkey = current_process().get_authkey() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if incref: | 
					
						
							|  |  |  |             self._incref() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         util.register_after_fork(self, BaseProxy._after_fork) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _connect(self): | 
					
						
							|  |  |  |         util.debug('making connection to manager') | 
					
						
							|  |  |  |         name = current_process().get_name() | 
					
						
							|  |  |  |         if threading.current_thread().get_name() != 'MainThread': | 
					
						
							|  |  |  |             name += '|' + threading.current_thread().get_name() | 
					
						
							|  |  |  |         conn = self._Client(self._token.address, authkey=self._authkey) | 
					
						
							|  |  |  |         dispatch(conn, None, 'accept_connection', (name,)) | 
					
						
							|  |  |  |         self._tls.connection = conn | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _callmethod(self, methodname, args=(), kwds={}): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Try to call a method of the referrent and return a copy of the result | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             conn = self._tls.connection | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             util.debug('thread %r does not own a connection', | 
					
						
							|  |  |  |                        threading.current_thread().get_name()) | 
					
						
							|  |  |  |             self._connect() | 
					
						
							|  |  |  |             conn = self._tls.connection | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         conn.send((self._id, methodname, args, kwds)) | 
					
						
							|  |  |  |         kind, result = conn.recv() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if kind == '#RETURN': | 
					
						
							|  |  |  |             return result | 
					
						
							|  |  |  |         elif kind == '#PROXY': | 
					
						
							|  |  |  |             exposed, token = result | 
					
						
							|  |  |  |             proxytype = self._manager._registry[token.typeid][-1] | 
					
						
							|  |  |  |             return proxytype( | 
					
						
							|  |  |  |                 token, self._serializer, manager=self._manager, | 
					
						
							|  |  |  |                 authkey=self._authkey, exposed=exposed | 
					
						
							|  |  |  |                 ) | 
					
						
							|  |  |  |         raise convert_to_error(kind, result) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _getvalue(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Get a copy of the value of the referent | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         return self._callmethod('#GETVALUE') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _incref(self): | 
					
						
							|  |  |  |         conn = self._Client(self._token.address, authkey=self._authkey) | 
					
						
							|  |  |  |         dispatch(conn, None, 'incref', (self._id,)) | 
					
						
							|  |  |  |         util.debug('INCREF %r', self._token.id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self._idset.add(self._id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         state = self._manager and self._manager._state | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self._close = util.Finalize( | 
					
						
							|  |  |  |             self, BaseProxy._decref, | 
					
						
							|  |  |  |             args=(self._token, self._authkey, state, | 
					
						
							|  |  |  |                   self._tls, self._idset, self._Client), | 
					
						
							|  |  |  |             exitpriority=10 | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @staticmethod | 
					
						
							|  |  |  |     def _decref(token, authkey, state, tls, idset, _Client): | 
					
						
							|  |  |  |         idset.discard(token.id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # check whether manager is still alive | 
					
						
							|  |  |  |         if state is None or state.value == State.STARTED: | 
					
						
							|  |  |  |             # tell manager this process no longer cares about referent | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 util.debug('DECREF %r', token.id) | 
					
						
							|  |  |  |                 conn = _Client(token.address, authkey=authkey) | 
					
						
							|  |  |  |                 dispatch(conn, None, 'decref', (token.id,)) | 
					
						
							|  |  |  |             except Exception, e: | 
					
						
							|  |  |  |                 util.debug('... decref failed %s', e) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             util.debug('DECREF %r -- manager already shutdown', token.id) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # check whether we can close this thread's connection because | 
					
						
							|  |  |  |         # the process owns no more references to objects for this manager | 
					
						
							|  |  |  |         if not idset and hasattr(tls, 'connection'): | 
					
						
							|  |  |  |             util.debug('thread %r has no more proxies so closing conn', | 
					
						
							|  |  |  |                        threading.current_thread().get_name()) | 
					
						
							|  |  |  |             tls.connection.close() | 
					
						
							|  |  |  |             del tls.connection | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _after_fork(self): | 
					
						
							|  |  |  |         self._manager = None | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             self._incref() | 
					
						
							|  |  |  |         except Exception, e: | 
					
						
							|  |  |  |             # the proxy may just be for a manager which has shutdown | 
					
						
							|  |  |  |             util.info('incref failed: %s' % e) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __reduce__(self): | 
					
						
							|  |  |  |         kwds = {} | 
					
						
							|  |  |  |         if Popen.thread_is_spawning(): | 
					
						
							|  |  |  |             kwds['authkey'] = self._authkey | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if getattr(self, '_isauto', False): | 
					
						
							|  |  |  |             kwds['exposed'] = self._exposed_ | 
					
						
							|  |  |  |             return (RebuildProxy, | 
					
						
							|  |  |  |                     (AutoProxy, self._token, self._serializer, kwds)) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             return (RebuildProxy, | 
					
						
							|  |  |  |                     (type(self), self._token, self._serializer, kwds)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __deepcopy__(self, memo): | 
					
						
							|  |  |  |         return self._getvalue() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return '<%s object, typeid %r at %s>' % \ | 
					
						
							|  |  |  |                (type(self).__name__, self._token.typeid, '0x%x' % id(self)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __str__(self): | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         Return representation of the referent (or a fall-back if that fails) | 
					
						
							|  |  |  |         '''
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             return self._callmethod('__repr__') | 
					
						
							|  |  |  |         except Exception: | 
					
						
							|  |  |  |             return repr(self)[:-1] + "; '__str__()' failed>" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Function used for unpickling | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def RebuildProxy(func, token, serializer, kwds): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Function used for unpickling proxy objects. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If possible the shared object is returned, or otherwise a proxy for it. | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     server = getattr(current_process(), '_manager_server', None) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if server and server.address == token.address: | 
					
						
							|  |  |  |         return server.id_to_obj[token.id][0] | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         incref = ( | 
					
						
							|  |  |  |             kwds.pop('incref', True) and | 
					
						
							|  |  |  |             not getattr(current_process(), '_inheriting', False) | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  |         return func(token, serializer, incref=incref, **kwds) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Functions to create proxies and proxy types | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def MakeProxyType(name, exposed, _cache={}): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Return an proxy type whose methods are given by `exposed` | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     exposed = tuple(exposed) | 
					
						
							|  |  |  |     try: | 
					
						
							|  |  |  |         return _cache[(name, exposed)] | 
					
						
							|  |  |  |     except KeyError: | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     dic = {} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for meth in exposed: | 
					
						
							|  |  |  |         exec '''def %s(self, *args, **kwds):
 | 
					
						
							|  |  |  |         return self._callmethod(%r, args, kwds)''' % (meth, meth) in dic
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     ProxyType = type(name, (BaseProxy,), dic) | 
					
						
							|  |  |  |     ProxyType._exposed_ = exposed | 
					
						
							|  |  |  |     _cache[(name, exposed)] = ProxyType | 
					
						
							|  |  |  |     return ProxyType | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def AutoProxy(token, serializer, manager=None, authkey=None, | 
					
						
							|  |  |  |               exposed=None, incref=True): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Return an auto-proxy for `token` | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     _Client = listener_client[serializer][1] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if exposed is None: | 
					
						
							|  |  |  |         conn = _Client(token.address, authkey=authkey) | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             exposed = dispatch(conn, None, 'get_methods', (token,)) | 
					
						
							|  |  |  |         finally: | 
					
						
							|  |  |  |             conn.close() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if authkey is None and manager is not None: | 
					
						
							|  |  |  |         authkey = manager._authkey | 
					
						
							|  |  |  |     if authkey is None: | 
					
						
							|  |  |  |         authkey = current_process().get_authkey() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     ProxyType = MakeProxyType('AutoProxy[%s]' % token.typeid, exposed) | 
					
						
							|  |  |  |     proxy = ProxyType(token, serializer, manager=manager, authkey=authkey, | 
					
						
							|  |  |  |                       incref=incref) | 
					
						
							|  |  |  |     proxy._isauto = True | 
					
						
							|  |  |  |     return proxy | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Types/callables which we will register with SyncManager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Namespace(object): | 
					
						
							|  |  |  |     def __init__(self, **kwds): | 
					
						
							|  |  |  |         self.__dict__.update(kwds) | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         items = self.__dict__.items() | 
					
						
							|  |  |  |         temp = [] | 
					
						
							|  |  |  |         for name, value in items: | 
					
						
							|  |  |  |             if not name.startswith('_'): | 
					
						
							|  |  |  |                 temp.append('%s=%r' % (name, value)) | 
					
						
							|  |  |  |         temp.sort() | 
					
						
							|  |  |  |         return 'Namespace(%s)' % str.join(', ', temp) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Value(object): | 
					
						
							|  |  |  |     def __init__(self, typecode, value, lock=True): | 
					
						
							|  |  |  |         self._typecode = typecode | 
					
						
							|  |  |  |         self._value = value | 
					
						
							|  |  |  |     def get(self): | 
					
						
							|  |  |  |         return self._value | 
					
						
							|  |  |  |     def set(self, value): | 
					
						
							|  |  |  |         self._value = value | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return '%s(%r, %r)'%(type(self).__name__, self._typecode, self._value) | 
					
						
							|  |  |  |     value = property(get, set) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def Array(typecode, sequence, lock=True): | 
					
						
							|  |  |  |     return array.array(typecode, sequence) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Proxy types used by SyncManager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class IteratorProxy(BaseProxy): | 
					
						
							|  |  |  |     # XXX remove methods for Py3.0 and Py2.6 | 
					
						
							|  |  |  |     _exposed_ = ('__next__', 'next', 'send', 'throw', 'close') | 
					
						
							|  |  |  |     def __iter__(self): | 
					
						
							|  |  |  |         return self | 
					
						
							|  |  |  |     def __next__(self, *args): | 
					
						
							|  |  |  |         return self._callmethod('__next__', args) | 
					
						
							|  |  |  |     def next(self, *args): | 
					
						
							|  |  |  |         return self._callmethod('next', args) | 
					
						
							|  |  |  |     def send(self, *args): | 
					
						
							|  |  |  |         return self._callmethod('send', args) | 
					
						
							|  |  |  |     def throw(self, *args): | 
					
						
							|  |  |  |         return self._callmethod('throw', args) | 
					
						
							|  |  |  |     def close(self, *args): | 
					
						
							|  |  |  |         return self._callmethod('close', args) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class AcquirerProxy(BaseProxy): | 
					
						
							|  |  |  |     _exposed_ = ('acquire', 'release') | 
					
						
							|  |  |  |     def acquire(self, blocking=True): | 
					
						
							|  |  |  |         return self._callmethod('acquire', (blocking,)) | 
					
						
							|  |  |  |     def release(self): | 
					
						
							|  |  |  |         return self._callmethod('release') | 
					
						
							|  |  |  |     def __enter__(self): | 
					
						
							|  |  |  |         return self._callmethod('acquire') | 
					
						
							|  |  |  |     def __exit__(self, exc_type, exc_val, exc_tb): | 
					
						
							|  |  |  |         return self._callmethod('release') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class ConditionProxy(AcquirerProxy): | 
					
						
							|  |  |  |     # XXX will Condition.notfyAll() name be available in Py3.0? | 
					
						
							|  |  |  |     _exposed_ = ('acquire', 'release', 'wait', 'notify', 'notify_all') | 
					
						
							|  |  |  |     def wait(self, timeout=None): | 
					
						
							|  |  |  |         return self._callmethod('wait', (timeout,)) | 
					
						
							|  |  |  |     def notify(self): | 
					
						
							|  |  |  |         return self._callmethod('notify') | 
					
						
							|  |  |  |     def notify_all(self): | 
					
						
							|  |  |  |         return self._callmethod('notify_all') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class EventProxy(BaseProxy): | 
					
						
							|  |  |  |     # XXX will Event.isSet name be available in Py3.0? | 
					
						
							|  |  |  |     _exposed_ = ('isSet', 'set', 'clear', 'wait') | 
					
						
							|  |  |  |     def is_set(self): | 
					
						
							| 
									
										
										
										
											2008-06-26 21:24:35 +00:00
										 |  |  |         return self._callmethod('is_set') | 
					
						
							| 
									
										
										
										
											2008-06-13 19:20:48 +00:00
										 |  |  |     def set(self): | 
					
						
							|  |  |  |         return self._callmethod('set') | 
					
						
							|  |  |  |     def clear(self): | 
					
						
							|  |  |  |         return self._callmethod('clear') | 
					
						
							|  |  |  |     def wait(self, timeout=None): | 
					
						
							|  |  |  |         return self._callmethod('wait', (timeout,)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class NamespaceProxy(BaseProxy): | 
					
						
							|  |  |  |     _exposed_ = ('__getattribute__', '__setattr__', '__delattr__') | 
					
						
							|  |  |  |     def __getattr__(self, key): | 
					
						
							|  |  |  |         if key[0] == '_': | 
					
						
							|  |  |  |             return object.__getattribute__(self, key) | 
					
						
							|  |  |  |         callmethod = object.__getattribute__(self, '_callmethod') | 
					
						
							|  |  |  |         return callmethod('__getattribute__', (key,)) | 
					
						
							|  |  |  |     def __setattr__(self, key, value): | 
					
						
							|  |  |  |         if key[0] == '_': | 
					
						
							|  |  |  |             return object.__setattr__(self, key, value) | 
					
						
							|  |  |  |         callmethod = object.__getattribute__(self, '_callmethod') | 
					
						
							|  |  |  |         return callmethod('__setattr__', (key, value)) | 
					
						
							|  |  |  |     def __delattr__(self, key): | 
					
						
							|  |  |  |         if key[0] == '_': | 
					
						
							|  |  |  |             return object.__delattr__(self, key) | 
					
						
							|  |  |  |         callmethod = object.__getattribute__(self, '_callmethod') | 
					
						
							|  |  |  |         return callmethod('__delattr__', (key,)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class ValueProxy(BaseProxy): | 
					
						
							|  |  |  |     _exposed_ = ('get', 'set') | 
					
						
							|  |  |  |     def get(self): | 
					
						
							|  |  |  |         return self._callmethod('get') | 
					
						
							|  |  |  |     def set(self, value): | 
					
						
							|  |  |  |         return self._callmethod('set', (value,)) | 
					
						
							|  |  |  |     value = property(get, set) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | BaseListProxy = MakeProxyType('BaseListProxy', ( | 
					
						
							|  |  |  |     '__add__', '__contains__', '__delitem__', '__delslice__', | 
					
						
							|  |  |  |     '__getitem__', '__getslice__', '__len__', '__mul__', | 
					
						
							|  |  |  |     '__reversed__', '__rmul__', '__setitem__', '__setslice__', | 
					
						
							|  |  |  |     'append', 'count', 'extend', 'index', 'insert', 'pop', 'remove', | 
					
						
							|  |  |  |     'reverse', 'sort', '__imul__' | 
					
						
							|  |  |  |     ))                  # XXX __getslice__ and __setslice__ unneeded in Py3.0 | 
					
						
							|  |  |  | class ListProxy(BaseListProxy): | 
					
						
							|  |  |  |     def __iadd__(self, value): | 
					
						
							|  |  |  |         self._callmethod('extend', (value,)) | 
					
						
							|  |  |  |         return self | 
					
						
							|  |  |  |     def __imul__(self, value): | 
					
						
							|  |  |  |         self._callmethod('__imul__', (value,)) | 
					
						
							|  |  |  |         return self | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | DictProxy = MakeProxyType('DictProxy', ( | 
					
						
							|  |  |  |     '__contains__', '__delitem__', '__getitem__', '__len__', | 
					
						
							|  |  |  |     '__setitem__', 'clear', 'copy', 'get', 'has_key', 'items', | 
					
						
							|  |  |  |     'keys', 'pop', 'popitem', 'setdefault', 'update', 'values' | 
					
						
							|  |  |  |     )) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | ArrayProxy = MakeProxyType('ArrayProxy', ( | 
					
						
							|  |  |  |     '__len__', '__getitem__', '__setitem__', '__getslice__', '__setslice__' | 
					
						
							|  |  |  |     ))                  # XXX __getslice__ and __setslice__ unneeded in Py3.0 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PoolProxy = MakeProxyType('PoolProxy', ( | 
					
						
							|  |  |  |     'apply', 'apply_async', 'close', 'imap', 'imap_unordered', 'join', | 
					
						
							|  |  |  |     'map', 'map_async', 'terminate' | 
					
						
							|  |  |  |     )) | 
					
						
							|  |  |  | PoolProxy._method_to_typeid_ = { | 
					
						
							|  |  |  |     'apply_async': 'AsyncResult', | 
					
						
							|  |  |  |     'map_async': 'AsyncResult', | 
					
						
							|  |  |  |     'imap': 'Iterator', | 
					
						
							|  |  |  |     'imap_unordered': 'Iterator' | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | # Definition of SyncManager | 
					
						
							|  |  |  | # | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class SyncManager(BaseManager): | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  |     Subclass of `BaseManager` which supports a number of shared object types. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The types registered are those intended for the synchronization | 
					
						
							|  |  |  |     of threads, plus `dict`, `list` and `Namespace`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The `multiprocessing.Manager()` function creates started instances of | 
					
						
							|  |  |  |     this class. | 
					
						
							|  |  |  |     '''
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | SyncManager.register('Queue', Queue.Queue) | 
					
						
							|  |  |  | SyncManager.register('JoinableQueue', Queue.Queue) | 
					
						
							|  |  |  | SyncManager.register('Event', threading.Event, EventProxy) | 
					
						
							|  |  |  | SyncManager.register('Lock', threading.Lock, AcquirerProxy) | 
					
						
							|  |  |  | SyncManager.register('RLock', threading.RLock, AcquirerProxy) | 
					
						
							|  |  |  | SyncManager.register('Semaphore', threading.Semaphore, AcquirerProxy) | 
					
						
							|  |  |  | SyncManager.register('BoundedSemaphore', threading.BoundedSemaphore, | 
					
						
							|  |  |  |                      AcquirerProxy) | 
					
						
							|  |  |  | SyncManager.register('Condition', threading.Condition, ConditionProxy) | 
					
						
							|  |  |  | SyncManager.register('Pool', Pool, PoolProxy) | 
					
						
							|  |  |  | SyncManager.register('list', list, ListProxy) | 
					
						
							|  |  |  | SyncManager.register('dict', dict, DictProxy) | 
					
						
							|  |  |  | SyncManager.register('Value', Value, ValueProxy) | 
					
						
							|  |  |  | SyncManager.register('Array', Array, ArrayProxy) | 
					
						
							|  |  |  | SyncManager.register('Namespace', Namespace, NamespaceProxy) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # types returned by methods of PoolProxy | 
					
						
							|  |  |  | SyncManager.register('Iterator', proxytype=IteratorProxy, create_method=False) | 
					
						
							|  |  |  | SyncManager.register('AsyncResult', create_method=False) |