## Module providing manager classes for dealing# with shared objects## multiprocessing/managers.py## Copyright (c) 2006-2008, R Oudkerk# Licensed to PSF under a Contributor Agreement.#__all__ = [ 'BaseManager', 'SyncManager', 'BaseProxy', 'Token','SharedMemoryManager' ]## Imports#import sysimport threadingimport signalimport arrayimport queueimport timeimport osfrom os import getpidfrom traceback import format_excfrom . import connectionfrom .context import reduction, get_spawning_popen, ProcessErrorfrom . import poolfrom . import processfrom . import utilfrom . import get_contexttry:from . import shared_memoryHAS_SHMEM = Trueexcept ImportError:HAS_SHMEM = False## Register some things for pickling#def reduce_array(a):return array.array, (a.typecode, a.tobytes())reduction.register(array.array, reduce_array)view_types = [type(getattr({}, name)()) for name in ('items','keys','values')]if view_types[0] is not list: # only needed in Py3.0def rebuild_as_list(obj):return list, (list(obj),)for view_type in view_types:reduction.register(view_type, rebuild_as_list)## 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) = statedef __repr__(self):return '%s(typeid=%r, address=%r, id=%r)' % \(self.__class__.__name__, 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 resultraise convert_to_error(kind, result)def convert_to_error(kind, result):if kind == '#ERROR':return resultelif kind in ('#TRACEBACK', '#UNSERIALIZABLE'):if not isinstance(result, str):raise TypeError("Result {0!r} (kind '{1}') type is {2}, not str".format(result, kind, type(result)))if kind == '#UNSERIALIZABLE':return RemoteError('Unserializable message: %s\n' % result)else:return RemoteError(result)else:return ValueError('Unrecognized message type {!r}'.format(kind))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 callable(func):temp.append(name)return tempdef 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):if not isinstance(authkey, bytes):raise TypeError("Authkey {0!r} is type {1!s}, not bytes".format(authkey, type(authkey)))self.registry = registryself.authkey = process.AuthenticationString(authkey)Listener, Client = listener_client[serializer]# do authentication laterself.listener = Listener(address=address, backlog=16)self.address = self.listener.addressself.id_to_obj = {'0': (None, ())}self.id_to_refcount = {}self.id_to_local_proxy_obj = {}self.mutex = threading.Lock()def serve_forever(self):'''Run the server forever'''self.stop_event = threading.Event()process.current_process()._manager_server = selftry:accepter = threading.Thread(target=self.accepter)accepter.daemon = Trueaccepter.start()try:while not self.stop_event.is_set():self.stop_event.wait(1)except (KeyboardInterrupt, SystemExit):passfinally:if sys.stdout != sys.__stdout__: # what about stderr?util.debug('resetting stdout, stderr')sys.stdout = sys.__stdout__sys.stderr = sys.__stderr__sys.exit(0)def accepter(self):while True:try:c = self.listener.accept()except OSError:continuet = threading.Thread(target=self.handle_request, args=(c,))t.daemon = Truet.start()def handle_request(self, c):'''Handle a new connection'''funcname = result = request = Nonetry:connection.deliver_challenge(c, self.authkey)connection.answer_challenge(c, self.authkey)request = c.recv()ignore, funcname, args, kwds = requestassert funcname in self.public, '%r unrecognized' % funcnamefunc = 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 as e:try:c.send(('#TRACEBACK', format_exc()))except Exception:passutil.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().name)recv = conn.recvsend = conn.sendid_to_obj = self.id_to_objwhile not self.stop_event.is_set():try:methodname = obj = Nonerequest = recv()ident, methodname, args, kwds = requesttry:obj, exposed, gettypeid = id_to_obj[ident]except KeyError as ke:try:obj, exposed, gettypeid = \self.id_to_local_proxy_obj[ident]except KeyError as second_ke:raise keif 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 as 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().name)sys.exit(0)except Exception:msg = ('#TRACEBACK', format_exc())try:try:send(msg)except Exception as e:send(('#UNSERIALIZABLE', format_exc()))except Exception as e:util.info('exception in thread serving %r',threading.current_thread().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 objdef 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):passdef debug_info(self, c):'''Return some info --- useful to spot problems with refcounting'''# Perhaps include debug info about 'c'?with self.mutex:result = []keys = list(self.id_to_refcount.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)def number_of_objects(self, c):'''Number of shared objects'''# Doesn't use (len(self.id_to_obj) - 1) as we shouldn't count ident='0'return len(self.id_to_refcount)def shutdown(self, c):'''Shutdown this process'''try:util.debug('manager received shutdown message')c.send(('#RETURN', None))except:import tracebacktraceback.print_exc()finally:self.stop_event.set()def create(*args, **kwds):'''Create a new shared object and return its id'''if len(args) >= 3:self, c, typeid, *args = argselif not args:raise TypeError("descriptor 'create' of 'Server' object ""needs an argument")else:if 'typeid' not in kwds:raise TypeError('create expected at least 2 positional ''arguments, got %d' % (len(args)-1))typeid = kwds.pop('typeid')if len(args) >= 2:self, c, *args = argsimport warningswarnings.warn("Passing 'typeid' as keyword argument is deprecated",DeprecationWarning, stacklevel=2)else:if 'c' not in kwds:raise TypeError('create expected at least 2 positional ''arguments, got %d' % (len(args)-1))c = kwds.pop('c')self, *args = argsimport warningswarnings.warn("Passing 'c' as keyword argument is deprecated",DeprecationWarning, stacklevel=2)args = tuple(args)with self.mutex:callable, exposed, method_to_typeid, proxytype = \self.registry[typeid]if callable is None:if kwds or (len(args) != 1):raise ValueError("Without callable, must have one non-keyword argument")obj = args[0]else:obj = callable(*args, **kwds)if exposed is None:exposed = public_methods(obj)if method_to_typeid is not None:if not isinstance(method_to_typeid, dict):raise TypeError("Method_to_typeid {0!r}: type {1!s}, not dict".format(method_to_typeid, type(method_to_typeid)))exposed = list(exposed) + list(method_to_typeid)ident = '%x' % id(obj) # convert to string because xmlrpclib# only has 32 bit signed integersutil.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] = 0self.incref(c, ident)return ident, tuple(exposed)create.__text_signature__ = '($self, c, typeid, /, *args, **kwds)'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().name = namec.send(('#RETURN', None))self.serve_client(c)def incref(self, c, ident):with self.mutex:try:self.id_to_refcount[ident] += 1except KeyError as ke:# If no external references exist but an internal (to the# manager) still does and a new external reference is created# from it, restore the manager's tracking of it from the# previously stashed internal ref.if ident in self.id_to_local_proxy_obj:self.id_to_refcount[ident] = 1self.id_to_obj[ident] = \self.id_to_local_proxy_obj[ident]obj, exposed, gettypeid = self.id_to_obj[ident]util.debug('Server re-enabled tracking & INCREF %r', ident)else:raise kedef decref(self, c, ident):if ident not in self.id_to_refcount and \ident in self.id_to_local_proxy_obj:util.debug('Server DECREF skipping %r', ident)returnwith self.mutex:if self.id_to_refcount[ident] <= 0:raise AssertionError("Id {0!s} ({1!r}) has refcount {2:n}, not 1+".format(ident, self.id_to_obj[ident],self.id_to_refcount[ident]))self.id_to_refcount[ident] -= 1if self.id_to_refcount[ident] == 0:del self.id_to_refcount[ident]if ident not in self.id_to_refcount:# Two-step process in case the object turns out to contain other# proxy objects (e.g. a managed list of managed lists).# Otherwise, deleting self.id_to_obj[ident] would trigger the# deleting of the stored value (another managed object) which would# in turn attempt to acquire the mutex that is already held here.self.id_to_obj[ident] = (None, (), None) # thread-safeutil.debug('disposing of obj with id %r', ident)with self.mutex:del self.id_to_obj[ident]## Class to represent state of a manager#class State(object):__slots__ = ['value']INITIAL = 0STARTED = 1SHUTDOWN = 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 = Serverdef __init__(self, address=None, authkey=None, serializer='pickle',ctx=None):if authkey is None:authkey = process.current_process().authkeyself._address = address # XXX not final address if eg ('', 0)self._authkey = process.AuthenticationString(authkey)self._state = State()self._state.value = State.INITIALself._serializer = serializerself._Listener, self._Client = listener_client[serializer]self._ctx = ctx or get_context()def get_server(self):'''Return server object with serve_forever() method and address attribute'''if self._state.value != State.INITIAL:if self._state.value == State.STARTED:raise ProcessError("Already started server")elif self._state.value == State.SHUTDOWN:raise ProcessError("Manager has shut down")else:raise ProcessError("Unknown state {!r}".format(self._state.value))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.STARTEDdef start(self, initializer=None, initargs=()):'''Spawn a server process for this manager object'''if self._state.value != State.INITIAL:if self._state.value == State.STARTED:raise ProcessError("Already started server")elif self._state.value == State.SHUTDOWN:raise ProcessError("Manager has shut down")else:raise ProcessError("Unknown state {!r}".format(self._state.value))if initializer is not None and not callable(initializer):raise TypeError('initializer must be a callable')# pipe over which we will retrieve address of serverreader, writer = connection.Pipe(duplex=False)# spawn process which runs a serverself._process = self._ctx.Process(target=type(self)._run_server,args=(self._registry, self._address, self._authkey,self._serializer, writer, initializer, initargs),)ident = ':'.join(str(i) for i in self._process._identity)self._process.name = type(self).__name__ + '-' + identself._process.start()# get address of serverwriter.close()self._address = reader.recv()reader.close()# register a finalizerself._state.value = State.STARTEDself.shutdown = util.Finalize(self, type(self)._finalize_manager,args=(self._process, self._address, self._authkey,self._state, self._Client),exitpriority=0)@classmethoddef _run_server(cls, registry, address, authkey, serializer, writer,initializer=None, initargs=()):'''Create a server, report its address and run it'''# bpo-36368: protect server process from KeyboardInterrupt signalssignal.signal(signal.SIGINT, signal.SIG_IGN)if initializer is not None:initializer(*initargs)# create serverserver = cls._Server(registry, address, authkey, serializer)# inform parent process of the server's addresswriter.send(server.address)writer.close()# run the managerutil.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), exposeddef join(self, timeout=None):'''Join the manager process (if it has been spawned)'''if self._process is not None:self._process.join(timeout)if not self._process.is_alive():self._process = Nonedef _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):if self._state.value == State.INITIAL:self.start()if self._state.value != State.STARTED:if self._state.value == State.INITIAL:raise ProcessError("Unable to start server")elif self._state.value == State.SHUTDOWN:raise ProcessError("Manager has shut down")else:raise ProcessError("Unknown state {!r}".format(self._state.value))return selfdef __exit__(self, exc_type, exc_val, exc_tb):self.shutdown()@staticmethoddef _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:passprocess.join(timeout=1.0)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.SHUTDOWNtry:del BaseProxy._address_to_local[address]except KeyError:pass@propertydef address(self):return self._address@classmethoddef 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 = AutoProxyexposed = 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 list(method_to_typeid.items()): # isinstance?assert type(key) is str, '%r is not a string' % keyassert type(value) is str, '%r is not a string' % valuecls._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)conn = self._Client(token.address, authkey=self._authkey)dispatch(conn, None, 'decref', (token.id,))return proxytemp.__name__ = typeidsetattr(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, manager_owned=False):with BaseProxy._mutex: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# self._tls is used to record the connection used by this# thread to communicate with the manager at token.addressself._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.addressself._idset = tls_idset[1]self._token = tokenself._id = self._token.idself._manager = managerself._serializer = serializerself._Client = listener_client[serializer][1]# Should be set to True only when a proxy object is being created# on the manager server; primary use case: nested proxy objects.# RebuildProxy detects when a proxy is being created on the manager# and sets this value appropriately.self._owned_by_manager = manager_ownedif authkey is not None:self._authkey = process.AuthenticationString(authkey)elif self._manager is not None:self._authkey = self._manager._authkeyelse:self._authkey = process.current_process().authkeyif incref:self._incref()util.register_after_fork(self, BaseProxy._after_fork)def _connect(self):util.debug('making connection to manager')name = process.current_process().nameif threading.current_thread().name != 'MainThread':name += '|' + threading.current_thread().nameconn = self._Client(self._token.address, authkey=self._authkey)dispatch(conn, None, 'accept_connection', (name,))self._tls.connection = conndef _callmethod(self, methodname, args=(), kwds={}):'''Try to call a method of the referrent and return a copy of the result'''try:conn = self._tls.connectionexcept AttributeError:util.debug('thread %r does not own a connection',threading.current_thread().name)self._connect()conn = self._tls.connectionconn.send((self._id, methodname, args, kwds))kind, result = conn.recv()if kind == '#RETURN':return resultelif kind == '#PROXY':exposed, token = resultproxytype = self._manager._registry[token.typeid][-1]token.address = self._token.addressproxy = proxytype(token, self._serializer, manager=self._manager,authkey=self._authkey, exposed=exposed)conn = self._Client(token.address, authkey=self._authkey)dispatch(conn, None, 'decref', (token.id,))return proxyraise convert_to_error(kind, result)def _getvalue(self):'''Get a copy of the value of the referent'''return self._callmethod('#GETVALUE')def _incref(self):if self._owned_by_manager:util.debug('owned_by_manager skipped INCREF of %r', self._token.id)returnconn = 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._stateself._close = util.Finalize(self, BaseProxy._decref,args=(self._token, self._authkey, state,self._tls, self._idset, self._Client),exitpriority=10)@staticmethoddef _decref(token, authkey, state, tls, idset, _Client):idset.discard(token.id)# check whether manager is still aliveif state is None or state.value == State.STARTED:# tell manager this process no longer cares about referenttry:util.debug('DECREF %r', token.id)conn = _Client(token.address, authkey=authkey)dispatch(conn, None, 'decref', (token.id,))except Exception as 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 managerif not idset and hasattr(tls, 'connection'):util.debug('thread %r has no more proxies so closing conn',threading.current_thread().name)tls.connection.close()del tls.connectiondef _after_fork(self):self._manager = Nonetry:self._incref()except Exception as e:# the proxy may just be for a manager which has shutdownutil.info('incref failed: %s' % e)def __reduce__(self):kwds = {}if get_spawning_popen() is not None:kwds['authkey'] = self._authkeyif 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 %#x>' % \(type(self).__name__, self._token.typeid, 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.'''server = getattr(process.current_process(), '_manager_server', None)if server and server.address == token.address:util.debug('Rebuild a proxy owned by manager, token=%r', token)kwds['manager_owned'] = Trueif token.id not in server.id_to_local_proxy_obj:server.id_to_local_proxy_obj[token.id] = \server.id_to_obj[token.id]incref = (kwds.pop('incref', True) andnot getattr(process.current_process(), '_inheriting', False))return func(token, serializer, incref=incref, **kwds)## Functions to create proxies and proxy types#def MakeProxyType(name, exposed, _cache={}):'''Return a proxy type whose methods are given by `exposed`'''exposed = tuple(exposed)try:return _cache[(name, exposed)]except KeyError:passdic = {}for meth in exposed:exec('''def %s(self, /, *args, **kwds):return self._callmethod(%r, args, kwds)''' % (meth, meth), dic)ProxyType = type(name, (BaseProxy,), dic)ProxyType._exposed_ = exposed_cache[(name, exposed)] = ProxyTypereturn ProxyTypedef 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._authkeyif authkey is None:authkey = process.current_process().authkeyProxyType = MakeProxyType('AutoProxy[%s]' % token.typeid, exposed)proxy = ProxyType(token, serializer, manager=manager, authkey=authkey,incref=incref)proxy._isauto = Truereturn proxy## Types/callables which we will register with SyncManager#class Namespace(object):def __init__(self, /, **kwds):self.__dict__.update(kwds)def __repr__(self):items = list(self.__dict__.items())temp = []for name, value in items:if not name.startswith('_'):temp.append('%s=%r' % (name, value))temp.sort()return '%s(%s)' % (self.__class__.__name__, ', '.join(temp))class Value(object):def __init__(self, typecode, value, lock=True):self._typecode = typecodeself._value = valuedef get(self):return self._valuedef set(self, value):self._value = valuedef __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):_exposed_ = ('__next__', 'send', 'throw', 'close')def __iter__(self):return selfdef __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, timeout=None):args = (blocking,) if timeout is None else (blocking, timeout)return self._callmethod('acquire', args)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):_exposed_ = ('acquire', 'release', 'wait', 'notify', 'notify_all')def wait(self, timeout=None):return self._callmethod('wait', (timeout,))def notify(self, n=1):return self._callmethod('notify', (n,))def notify_all(self):return self._callmethod('notify_all')def wait_for(self, predicate, timeout=None):result = predicate()if result:return resultif timeout is not None:endtime = time.monotonic() + timeoutelse:endtime = Nonewaittime = Nonewhile not result:if endtime is not None:waittime = endtime - time.monotonic()if waittime <= 0:breakself.wait(waittime)result = predicate()return resultclass EventProxy(BaseProxy):_exposed_ = ('is_set', 'set', 'clear', 'wait')def is_set(self):return self._callmethod('is_set')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 BarrierProxy(BaseProxy):_exposed_ = ('__getattribute__', 'wait', 'abort', 'reset')def wait(self, timeout=None):return self._callmethod('wait', (timeout,))def abort(self):return self._callmethod('abort')def reset(self):return self._callmethod('reset')@propertydef parties(self):return self._callmethod('__getattribute__', ('parties',))@propertydef n_waiting(self):return self._callmethod('__getattribute__', ('n_waiting',))@propertydef broken(self):return self._callmethod('__getattribute__', ('broken',))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__', '__getitem__', '__len__','__mul__', '__reversed__', '__rmul__', '__setitem__','append', 'count', 'extend', 'index', 'insert', 'pop', 'remove','reverse', 'sort', '__imul__'))class ListProxy(BaseListProxy):def __iadd__(self, value):self._callmethod('extend', (value,))return selfdef __imul__(self, value):self._callmethod('__imul__', (value,))return selfDictProxy = MakeProxyType('DictProxy', ('__contains__', '__delitem__', '__getitem__', '__iter__', '__len__','__setitem__', 'clear', 'copy', 'get', 'items','keys', 'pop', 'popitem', 'setdefault', 'update', 'values'))DictProxy._method_to_typeid_ = {'__iter__': 'Iterator',}ArrayProxy = MakeProxyType('ArrayProxy', ('__len__', '__getitem__', '__setitem__'))BasePoolProxy = MakeProxyType('PoolProxy', ('apply', 'apply_async', 'close', 'imap', 'imap_unordered', 'join','map', 'map_async', 'starmap', 'starmap_async', 'terminate',))BasePoolProxy._method_to_typeid_ = {'apply_async': 'AsyncResult','map_async': 'AsyncResult','starmap_async': 'AsyncResult','imap': 'Iterator','imap_unordered': 'Iterator'}class PoolProxy(BasePoolProxy):def __enter__(self):return selfdef __exit__(self, exc_type, exc_val, exc_tb):self.terminate()## 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 synchronizationof threads, plus `dict`, `list` and `Namespace`.The `multiprocessing.Manager()` function creates started instances ofthis 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('Barrier', threading.Barrier, BarrierProxy)SyncManager.register('Pool', 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 PoolProxySyncManager.register('Iterator', proxytype=IteratorProxy, create_method=False)SyncManager.register('AsyncResult', create_method=False)## Definition of SharedMemoryManager and SharedMemoryServer#if HAS_SHMEM:class _SharedMemoryTracker:"Manages one or more shared memory segments."def __init__(self, name, segment_names=[]):self.shared_memory_context_name = nameself.segment_names = segment_namesdef register_segment(self, segment_name):"Adds the supplied shared memory block name to tracker."util.debug(f"Register segment {segment_name!r} in pid {getpid()}")self.segment_names.append(segment_name)def destroy_segment(self, segment_name):"""Calls unlink() on the shared memory block with the supplied nameand removes it from the list of blocks being tracked."""util.debug(f"Destroy segment {segment_name!r} in pid {getpid()}")self.segment_names.remove(segment_name)segment = shared_memory.SharedMemory(segment_name)segment.close()segment.unlink()def unlink(self):"Calls destroy_segment() on all tracked shared memory blocks."for segment_name in self.segment_names[:]:self.destroy_segment(segment_name)def __del__(self):util.debug(f"Call {self.__class__.__name__}.__del__ in {getpid()}")self.unlink()def __getstate__(self):return (self.shared_memory_context_name, self.segment_names)def __setstate__(self, state):self.__init__(*state)class SharedMemoryServer(Server):public = Server.public + \['track_segment', 'release_segment', 'list_segments']def __init__(self, *args, **kwargs):Server.__init__(self, *args, **kwargs)self.shared_memory_context = \_SharedMemoryTracker(f"shmm_{self.address}_{getpid()}")util.debug(f"SharedMemoryServer started by pid {getpid()}")def create(*args, **kwargs):"""Create a new distributed-shared object (not backed by a sharedmemory block) and return its id to be used in a Proxy Object."""# Unless set up as a shared proxy, don't make shared_memory_context# a standard part of kwargs. This makes things easier for supplying# simple functions.if len(args) >= 3:typeod = args[2]elif 'typeid' in kwargs:typeid = kwargs['typeid']elif not args:raise TypeError("descriptor 'create' of 'SharedMemoryServer' ""object needs an argument")else:raise TypeError('create expected at least 2 positional ''arguments, got %d' % (len(args)-1))if hasattr(self.registry[typeid][-1], "_shared_memory_proxy"):kwargs['shared_memory_context'] = self.shared_memory_contextreturn Server.create(*args, **kwargs)create.__text_signature__ = '($self, c, typeid, /, *args, **kwargs)'def shutdown(self, c):"Call unlink() on all tracked shared memory, terminate the Server."self.shared_memory_context.unlink()return Server.shutdown(self, c)def track_segment(self, c, segment_name):"Adds the supplied shared memory block name to Server's tracker."self.shared_memory_context.register_segment(segment_name)def release_segment(self, c, segment_name):"""Calls unlink() on the shared memory block with the supplied nameand removes it from the tracker instance inside the Server."""self.shared_memory_context.destroy_segment(segment_name)def list_segments(self, c):"""Returns a list of names of shared memory blocks that the Serveris currently tracking."""return self.shared_memory_context.segment_namesclass SharedMemoryManager(BaseManager):"""Like SyncManager but uses SharedMemoryServer instead of Server.It provides methods for creating and returning SharedMemory instancesand for creating a list-like object (ShareableList) backed by sharedmemory. It also provides methods that create and return Proxy Objectsthat support synchronization across processes (i.e. multi-process-safelocks and semaphores)."""_Server = SharedMemoryServerdef __init__(self, *args, **kwargs):if os.name == "posix":# bpo-36867: Ensure the resource_tracker is running before# launching the manager process, so that concurrent# shared_memory manipulation both in the manager and in the# current process does not create two resource_tracker# processes.from . import resource_trackerresource_tracker.ensure_running()BaseManager.__init__(self, *args, **kwargs)util.debug(f"{self.__class__.__name__} created by pid {getpid()}")def __del__(self):util.debug(f"{self.__class__.__name__}.__del__ by pid {getpid()}")passdef get_server(self):'Better than monkeypatching for now; merge into Server ultimately'if self._state.value != State.INITIAL:if self._state.value == State.STARTED:raise ProcessError("Already started SharedMemoryServer")elif self._state.value == State.SHUTDOWN:raise ProcessError("SharedMemoryManager has shut down")else:raise ProcessError("Unknown state {!r}".format(self._state.value))return self._Server(self._registry, self._address,self._authkey, self._serializer)def SharedMemory(self, size):"""Returns a new SharedMemory instance with the specified size inbytes, to be tracked by the manager."""with self._Client(self._address, authkey=self._authkey) as conn:sms = shared_memory.SharedMemory(None, create=True, size=size)try:dispatch(conn, None, 'track_segment', (sms.name,))except BaseException as e:sms.unlink()raise ereturn smsdef ShareableList(self, sequence):"""Returns a new ShareableList instance populated with the valuesfrom the input sequence, to be tracked by the manager."""with self._Client(self._address, authkey=self._authkey) as conn:sl = shared_memory.ShareableList(sequence)try:dispatch(conn, None, 'track_segment', (sl.shm.name,))except BaseException as e:sl.shm.unlink()raise ereturn sl
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。