blob: 320e49298beee97da84e8760e0cbf389b349f37f [file] [log] [blame]
Antoine Pitrou64a467d2010-12-12 20:34:49 +00001:mod:`multiprocessing` --- Process-based parallelism
2====================================================
Benjamin Petersone711caf2008-06-11 16:44:04 +00003
4.. module:: multiprocessing
Antoine Pitrou64a467d2010-12-12 20:34:49 +00005 :synopsis: Process-based parallelism.
Benjamin Petersone711caf2008-06-11 16:44:04 +00006
Benjamin Petersone711caf2008-06-11 16:44:04 +00007
8Introduction
Georg Brandl49702152008-09-29 06:43:45 +00009------------
Benjamin Petersone711caf2008-06-11 16:44:04 +000010
Benjamin Peterson5289b2b2008-06-28 00:40:54 +000011:mod:`multiprocessing` is a package that supports spawning processes using an
12API similar to the :mod:`threading` module. The :mod:`multiprocessing` package
13offers both local and remote concurrency, effectively side-stepping the
14:term:`Global Interpreter Lock` by using subprocesses instead of threads. Due
15to this, the :mod:`multiprocessing` module allows the programmer to fully
16leverage multiple processors on a given machine. It runs on both Unix and
17Windows.
Benjamin Petersone711caf2008-06-11 16:44:04 +000018
Raymond Hettingerfd151912010-11-04 03:02:56 +000019.. note::
Benjamin Petersone5384b02008-10-04 22:00:42 +000020
21 Some of this package's functionality requires a functioning shared semaphore
Georg Brandl48310cd2009-01-03 21:18:54 +000022 implementation on the host operating system. Without one, the
23 :mod:`multiprocessing.synchronize` module will be disabled, and attempts to
24 import it will result in an :exc:`ImportError`. See
Benjamin Petersone5384b02008-10-04 22:00:42 +000025 :issue:`3770` for additional information.
Benjamin Petersone711caf2008-06-11 16:44:04 +000026
Jesse Noller45239682008-11-28 18:46:19 +000027.. note::
28
Ezio Melotti2ee88352011-04-29 07:10:24 +030029 Functionality within this package requires that the ``__main__`` module be
Jesse Noller45239682008-11-28 18:46:19 +000030 importable by the children. This is covered in :ref:`multiprocessing-programming`
31 however it is worth pointing out here. This means that some examples, such
32 as the :class:`multiprocessing.Pool` examples will not work in the
33 interactive interpreter. For example::
34
35 >>> from multiprocessing import Pool
36 >>> p = Pool(5)
37 >>> def f(x):
Georg Brandla1c6a1c2009-01-03 21:26:05 +000038 ... return x*x
Georg Brandl48310cd2009-01-03 21:18:54 +000039 ...
Jesse Noller45239682008-11-28 18:46:19 +000040 >>> p.map(f, [1,2,3])
41 Process PoolWorker-1:
42 Process PoolWorker-2:
R. David Murray8e8099c2009-04-28 18:02:00 +000043 Process PoolWorker-3:
44 Traceback (most recent call last):
Jesse Noller45239682008-11-28 18:46:19 +000045 Traceback (most recent call last):
46 Traceback (most recent call last):
47 AttributeError: 'module' object has no attribute 'f'
48 AttributeError: 'module' object has no attribute 'f'
49 AttributeError: 'module' object has no attribute 'f'
50
R. David Murray8e8099c2009-04-28 18:02:00 +000051 (If you try this it will actually output three full tracebacks
52 interleaved in a semi-random fashion, and then you may have to
53 stop the master process somehow.)
54
Jesse Noller45239682008-11-28 18:46:19 +000055
Benjamin Petersone711caf2008-06-11 16:44:04 +000056The :class:`Process` class
57~~~~~~~~~~~~~~~~~~~~~~~~~~
58
59In :mod:`multiprocessing`, processes are spawned by creating a :class:`Process`
Benjamin Peterson5289b2b2008-06-28 00:40:54 +000060object and then calling its :meth:`~Process.start` method. :class:`Process`
Benjamin Petersone711caf2008-06-11 16:44:04 +000061follows the API of :class:`threading.Thread`. A trivial example of a
62multiprocess program is ::
63
Georg Brandlb3959bd2010-04-08 06:33:16 +000064 from multiprocessing import Process
Benjamin Petersone711caf2008-06-11 16:44:04 +000065
66 def f(name):
Georg Brandl49702152008-09-29 06:43:45 +000067 print('hello', name)
Benjamin Petersone711caf2008-06-11 16:44:04 +000068
Georg Brandlb3959bd2010-04-08 06:33:16 +000069 if __name__ == '__main__':
70 p = Process(target=f, args=('bob',))
71 p.start()
72 p.join()
Benjamin Petersone711caf2008-06-11 16:44:04 +000073
Jesse Noller45239682008-11-28 18:46:19 +000074To show the individual process IDs involved, here is an expanded example::
75
76 from multiprocessing import Process
77 import os
78
79 def info(title):
Ezio Melotti985e24d2009-09-13 07:54:02 +000080 print(title)
81 print('module name:', __name__)
82 print('parent process:', os.getppid())
83 print('process id:', os.getpid())
Georg Brandl48310cd2009-01-03 21:18:54 +000084
Jesse Noller45239682008-11-28 18:46:19 +000085 def f(name):
86 info('function f')
Ezio Melotti985e24d2009-09-13 07:54:02 +000087 print('hello', name)
Georg Brandl48310cd2009-01-03 21:18:54 +000088
Jesse Noller45239682008-11-28 18:46:19 +000089 if __name__ == '__main__':
90 info('main line')
91 p = Process(target=f, args=('bob',))
92 p.start()
93 p.join()
Benjamin Petersone711caf2008-06-11 16:44:04 +000094
95For an explanation of why (on Windows) the ``if __name__ == '__main__'`` part is
96necessary, see :ref:`multiprocessing-programming`.
97
98
99
100Exchanging objects between processes
101~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
102
103:mod:`multiprocessing` supports two types of communication channel between
104processes:
105
106**Queues**
107
Benjamin Peterson257060a2008-06-28 01:42:41 +0000108 The :class:`Queue` class is a near clone of :class:`queue.Queue`. For
Benjamin Petersone711caf2008-06-11 16:44:04 +0000109 example::
110
111 from multiprocessing import Process, Queue
112
113 def f(q):
114 q.put([42, None, 'hello'])
115
Georg Brandl1f01deb2009-01-03 22:47:39 +0000116 if __name__ == '__main__':
117 q = Queue()
118 p = Process(target=f, args=(q,))
119 p.start()
120 print(q.get()) # prints "[42, None, 'hello']"
121 p.join()
Benjamin Petersone711caf2008-06-11 16:44:04 +0000122
Ask Solem518eaa82010-11-09 21:46:03 +0000123 Queues are thread and process safe, but note that they must never
124 be instantiated as a side effect of importing a module: this can lead
125 to a deadlock! (see :ref:`threaded-imports`)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000126
127**Pipes**
128
129 The :func:`Pipe` function returns a pair of connection objects connected by a
130 pipe which by default is duplex (two-way). For example::
131
132 from multiprocessing import Process, Pipe
133
134 def f(conn):
135 conn.send([42, None, 'hello'])
136 conn.close()
137
138 if __name__ == '__main__':
139 parent_conn, child_conn = Pipe()
140 p = Process(target=f, args=(child_conn,))
141 p.start()
Georg Brandl49702152008-09-29 06:43:45 +0000142 print(parent_conn.recv()) # prints "[42, None, 'hello']"
Benjamin Petersone711caf2008-06-11 16:44:04 +0000143 p.join()
144
145 The two connection objects returned by :func:`Pipe` represent the two ends of
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000146 the pipe. Each connection object has :meth:`~Connection.send` and
147 :meth:`~Connection.recv` methods (among others). Note that data in a pipe
148 may become corrupted if two processes (or threads) try to read from or write
149 to the *same* end of the pipe at the same time. Of course there is no risk
150 of corruption from processes using different ends of the pipe at the same
151 time.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000152
153
154Synchronization between processes
155~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
156
157:mod:`multiprocessing` contains equivalents of all the synchronization
158primitives from :mod:`threading`. For instance one can use a lock to ensure
159that only one process prints to standard output at a time::
160
161 from multiprocessing import Process, Lock
162
163 def f(l, i):
164 l.acquire()
Georg Brandl49702152008-09-29 06:43:45 +0000165 print('hello world', i)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000166 l.release()
167
168 if __name__ == '__main__':
169 lock = Lock()
170
171 for num in range(10):
172 Process(target=f, args=(lock, num)).start()
173
174Without using the lock output from the different processes is liable to get all
175mixed up.
176
177
178Sharing state between processes
179~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
180
181As mentioned above, when doing concurrent programming it is usually best to
182avoid using shared state as far as possible. This is particularly true when
183using multiple processes.
184
185However, if you really do need to use some shared data then
186:mod:`multiprocessing` provides a couple of ways of doing so.
187
188**Shared memory**
189
190 Data can be stored in a shared memory map using :class:`Value` or
191 :class:`Array`. For example, the following code ::
192
193 from multiprocessing import Process, Value, Array
194
195 def f(n, a):
196 n.value = 3.1415927
197 for i in range(len(a)):
198 a[i] = -a[i]
199
200 if __name__ == '__main__':
201 num = Value('d', 0.0)
202 arr = Array('i', range(10))
203
204 p = Process(target=f, args=(num, arr))
205 p.start()
206 p.join()
207
Georg Brandl49702152008-09-29 06:43:45 +0000208 print(num.value)
209 print(arr[:])
Benjamin Petersone711caf2008-06-11 16:44:04 +0000210
211 will print ::
212
213 3.1415927
214 [0, -1, -2, -3, -4, -5, -6, -7, -8, -9]
215
216 The ``'d'`` and ``'i'`` arguments used when creating ``num`` and ``arr`` are
217 typecodes of the kind used by the :mod:`array` module: ``'d'`` indicates a
Georg Brandl2ee470f2008-07-16 12:55:28 +0000218 double precision float and ``'i'`` indicates a signed integer. These shared
Georg Brandlf285bcc2010-10-19 21:07:16 +0000219 objects will be process and thread-safe.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000220
221 For more flexibility in using shared memory one can use the
222 :mod:`multiprocessing.sharedctypes` module which supports the creation of
223 arbitrary ctypes objects allocated from shared memory.
224
225**Server process**
226
227 A manager object returned by :func:`Manager` controls a server process which
Georg Brandl2ee470f2008-07-16 12:55:28 +0000228 holds Python objects and allows other processes to manipulate them using
Benjamin Petersone711caf2008-06-11 16:44:04 +0000229 proxies.
230
231 A manager returned by :func:`Manager` will support types :class:`list`,
232 :class:`dict`, :class:`Namespace`, :class:`Lock`, :class:`RLock`,
233 :class:`Semaphore`, :class:`BoundedSemaphore`, :class:`Condition`,
234 :class:`Event`, :class:`Queue`, :class:`Value` and :class:`Array`. For
235 example, ::
236
237 from multiprocessing import Process, Manager
238
239 def f(d, l):
240 d[1] = '1'
241 d['2'] = 2
242 d[0.25] = None
243 l.reverse()
244
245 if __name__ == '__main__':
246 manager = Manager()
247
248 d = manager.dict()
249 l = manager.list(range(10))
250
251 p = Process(target=f, args=(d, l))
252 p.start()
253 p.join()
254
Georg Brandl49702152008-09-29 06:43:45 +0000255 print(d)
256 print(l)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000257
258 will print ::
259
260 {0.25: None, 1: '1', '2': 2}
261 [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
262
263 Server process managers are more flexible than using shared memory objects
264 because they can be made to support arbitrary object types. Also, a single
265 manager can be shared by processes on different computers over a network.
266 They are, however, slower than using shared memory.
267
268
269Using a pool of workers
270~~~~~~~~~~~~~~~~~~~~~~~
271
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000272The :class:`~multiprocessing.pool.Pool` class represents a pool of worker
Benjamin Petersone711caf2008-06-11 16:44:04 +0000273processes. It has methods which allows tasks to be offloaded to the worker
274processes in a few different ways.
275
276For example::
277
278 from multiprocessing import Pool
279
280 def f(x):
281 return x*x
282
283 if __name__ == '__main__':
Ezio Melotti985e24d2009-09-13 07:54:02 +0000284 pool = Pool(processes=4) # start 4 worker processes
Jesse Noller45239682008-11-28 18:46:19 +0000285 result = pool.apply_async(f, [10]) # evaluate "f(10)" asynchronously
Ezio Melotti985e24d2009-09-13 07:54:02 +0000286 print(result.get(timeout=1)) # prints "100" unless your computer is *very* slow
287 print(pool.map(f, range(10))) # prints "[0, 1, 4,..., 81]"
Benjamin Petersone711caf2008-06-11 16:44:04 +0000288
289
290Reference
291---------
292
293The :mod:`multiprocessing` package mostly replicates the API of the
294:mod:`threading` module.
295
296
297:class:`Process` and exceptions
298~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
299
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +0000300.. class:: Process([group[, target[, name[, args[, kwargs]]]]], *, daemon=None)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000301
302 Process objects represent activity that is run in a separate process. The
303 :class:`Process` class has equivalents of all the methods of
304 :class:`threading.Thread`.
305
306 The constructor should always be called with keyword arguments. *group*
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000307 should always be ``None``; it exists solely for compatibility with
Benjamin Petersona786b022008-08-25 21:05:21 +0000308 :class:`threading.Thread`. *target* is the callable object to be invoked by
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000309 the :meth:`run()` method. It defaults to ``None``, meaning nothing is
Benjamin Petersone711caf2008-06-11 16:44:04 +0000310 called. *name* is the process name. By default, a unique name is constructed
311 of the form 'Process-N\ :sub:`1`:N\ :sub:`2`:...:N\ :sub:`k`' where N\
312 :sub:`1`,N\ :sub:`2`,...,N\ :sub:`k` is a sequence of integers whose length
313 is determined by the *generation* of the process. *args* is the argument
314 tuple for the target invocation. *kwargs* is a dictionary of keyword
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +0000315 arguments for the target invocation. If provided, the keyword-only *daemon* argument
316 sets the process :attr:`daemon` flag to ``True`` or ``False``. If ``None``
317 (the default), this flag will be inherited from the creating process.
318
319 By default, no arguments are passed to *target*.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000320
321 If a subclass overrides the constructor, it must make sure it invokes the
322 base class constructor (:meth:`Process.__init__`) before doing anything else
323 to the process.
324
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +0000325 .. versionchanged:: 3.3
326 Added the *daemon* argument.
327
Benjamin Petersone711caf2008-06-11 16:44:04 +0000328 .. method:: run()
329
330 Method representing the process's activity.
331
332 You may override this method in a subclass. The standard :meth:`run`
333 method invokes the callable object passed to the object's constructor as
334 the target argument, if any, with sequential and keyword arguments taken
335 from the *args* and *kwargs* arguments, respectively.
336
337 .. method:: start()
338
339 Start the process's activity.
340
341 This must be called at most once per process object. It arranges for the
342 object's :meth:`run` method to be invoked in a separate process.
343
344 .. method:: join([timeout])
345
Charles-François Nataliacd9f7c2011-07-25 18:35:49 +0200346 If the optional argument *timeout* is ``None`` (the default), the method
347 blocks until the process whose :meth:`join` method is called terminates.
348 If *timeout* is a positive number, it blocks at most *timeout* seconds.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000349
350 A process can be joined many times.
351
352 A process cannot join itself because this would cause a deadlock. It is
353 an error to attempt to join a process before it has been started.
354
Benjamin Petersona786b022008-08-25 21:05:21 +0000355 .. attribute:: name
Benjamin Petersone711caf2008-06-11 16:44:04 +0000356
Benjamin Petersona786b022008-08-25 21:05:21 +0000357 The process's name.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000358
359 The name is a string used for identification purposes only. It has no
360 semantics. Multiple processes may be given the same name. The initial
361 name is set by the constructor.
362
Jesse Noller45239682008-11-28 18:46:19 +0000363 .. method:: is_alive
Benjamin Petersone711caf2008-06-11 16:44:04 +0000364
365 Return whether the process is alive.
366
367 Roughly, a process object is alive from the moment the :meth:`start`
368 method returns until the child process terminates.
369
Benjamin Petersona786b022008-08-25 21:05:21 +0000370 .. attribute:: daemon
Benjamin Petersone711caf2008-06-11 16:44:04 +0000371
Benjamin Petersonda10d3b2009-01-01 00:23:30 +0000372 The process's daemon flag, a Boolean value. This must be set before
Benjamin Petersona786b022008-08-25 21:05:21 +0000373 :meth:`start` is called.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000374
375 The initial value is inherited from the creating process.
376
377 When a process exits, it attempts to terminate all of its daemonic child
378 processes.
379
380 Note that a daemonic process is not allowed to create child processes.
381 Otherwise a daemonic process would leave its children orphaned if it gets
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000382 terminated when its parent process exits. Additionally, these are **not**
383 Unix daemons or services, they are normal processes that will be
Georg Brandl6faee4e2010-09-21 14:48:28 +0000384 terminated (and not joined) if non-daemonic processes have exited.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000385
Benjamin Petersona786b022008-08-25 21:05:21 +0000386 In addition to the :class:`Threading.Thread` API, :class:`Process` objects
387 also support the following attributes and methods:
Benjamin Petersone711caf2008-06-11 16:44:04 +0000388
Benjamin Petersona786b022008-08-25 21:05:21 +0000389 .. attribute:: pid
Benjamin Petersone711caf2008-06-11 16:44:04 +0000390
391 Return the process ID. Before the process is spawned, this will be
392 ``None``.
393
Benjamin Petersona786b022008-08-25 21:05:21 +0000394 .. attribute:: exitcode
Benjamin Petersone711caf2008-06-11 16:44:04 +0000395
Benjamin Petersona786b022008-08-25 21:05:21 +0000396 The child's exit code. This will be ``None`` if the process has not yet
397 terminated. A negative value *-N* indicates that the child was terminated
398 by signal *N*.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000399
Benjamin Petersona786b022008-08-25 21:05:21 +0000400 .. attribute:: authkey
Benjamin Petersone711caf2008-06-11 16:44:04 +0000401
Benjamin Petersona786b022008-08-25 21:05:21 +0000402 The process's authentication key (a byte string).
Benjamin Petersone711caf2008-06-11 16:44:04 +0000403
404 When :mod:`multiprocessing` is initialized the main process is assigned a
405 random string using :func:`os.random`.
406
407 When a :class:`Process` object is created, it will inherit the
Benjamin Petersona786b022008-08-25 21:05:21 +0000408 authentication key of its parent process, although this may be changed by
409 setting :attr:`authkey` to another byte string.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000410
411 See :ref:`multiprocessing-auth-keys`.
412
Antoine Pitrou176f07d2011-06-06 19:35:31 +0200413 .. attribute:: sentinel
414
415 A numeric handle of a system object which will become "ready" when
416 the process ends.
417
Antoine Pitroubdb1cf12012-03-05 19:28:37 +0100418 You can use this value if you want to wait on several events at
419 once using :func:`multiprocessing.connection.wait`. Otherwise
420 calling :meth:`join()` is simpler.
421
Antoine Pitrou176f07d2011-06-06 19:35:31 +0200422 On Windows, this is an OS handle usable with the ``WaitForSingleObject``
423 and ``WaitForMultipleObjects`` family of API calls. On Unix, this is
424 a file descriptor usable with primitives from the :mod:`select` module.
425
Antoine Pitrou176f07d2011-06-06 19:35:31 +0200426 .. versionadded:: 3.3
427
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000428 .. method:: terminate()
Benjamin Petersone711caf2008-06-11 16:44:04 +0000429
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000430 Terminate the process. On Unix this is done using the ``SIGTERM`` signal;
Georg Brandl60203b42010-10-06 10:11:56 +0000431 on Windows :c:func:`TerminateProcess` is used. Note that exit handlers and
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000432 finally clauses, etc., will not be executed.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000433
434 Note that descendant processes of the process will *not* be terminated --
435 they will simply become orphaned.
436
437 .. warning::
438
439 If this method is used when the associated process is using a pipe or
440 queue then the pipe or queue is liable to become corrupted and may
441 become unusable by other process. Similarly, if the process has
442 acquired a lock or semaphore etc. then terminating it is liable to
443 cause other processes to deadlock.
444
Ask Solemff7ffdd2010-11-09 21:52:33 +0000445 Note that the :meth:`start`, :meth:`join`, :meth:`is_alive`,
446 :meth:`terminate` and :attr:`exit_code` methods should only be called by
447 the process that created the process object.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000448
R. David Murray8e8099c2009-04-28 18:02:00 +0000449 Example usage of some of the methods of :class:`Process`:
450
451 .. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +0000452
Benjamin Peterson206e3072008-10-19 14:07:49 +0000453 >>> import multiprocessing, time, signal
454 >>> p = multiprocessing.Process(target=time.sleep, args=(1000,))
Georg Brandl49702152008-09-29 06:43:45 +0000455 >>> print(p, p.is_alive())
Benjamin Petersone711caf2008-06-11 16:44:04 +0000456 <Process(Process-1, initial)> False
457 >>> p.start()
Georg Brandl49702152008-09-29 06:43:45 +0000458 >>> print(p, p.is_alive())
Benjamin Petersone711caf2008-06-11 16:44:04 +0000459 <Process(Process-1, started)> True
460 >>> p.terminate()
R. David Murray8e8099c2009-04-28 18:02:00 +0000461 >>> time.sleep(0.1)
Georg Brandl49702152008-09-29 06:43:45 +0000462 >>> print(p, p.is_alive())
Benjamin Petersone711caf2008-06-11 16:44:04 +0000463 <Process(Process-1, stopped[SIGTERM])> False
Benjamin Petersona786b022008-08-25 21:05:21 +0000464 >>> p.exitcode == -signal.SIGTERM
Benjamin Petersone711caf2008-06-11 16:44:04 +0000465 True
466
467
468.. exception:: BufferTooShort
469
470 Exception raised by :meth:`Connection.recv_bytes_into()` when the supplied
471 buffer object is too small for the message read.
472
473 If ``e`` is an instance of :exc:`BufferTooShort` then ``e.args[0]`` will give
474 the message as a byte string.
475
476
477Pipes and Queues
478~~~~~~~~~~~~~~~~
479
480When using multiple processes, one generally uses message passing for
481communication between processes and avoids having to use any synchronization
482primitives like locks.
483
484For passing messages one can use :func:`Pipe` (for a connection between two
485processes) or a queue (which allows multiple producers and consumers).
486
Sandro Tosicd778152012-02-15 23:27:00 +0100487The :class:`Queue`, :class:`SimpleQueue` and :class:`JoinableQueue` types are multi-producer,
Benjamin Peterson257060a2008-06-28 01:42:41 +0000488multi-consumer FIFO queues modelled on the :class:`queue.Queue` class in the
Benjamin Petersone711caf2008-06-11 16:44:04 +0000489standard library. They differ in that :class:`Queue` lacks the
Benjamin Peterson257060a2008-06-28 01:42:41 +0000490:meth:`~queue.Queue.task_done` and :meth:`~queue.Queue.join` methods introduced
491into Python 2.5's :class:`queue.Queue` class.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000492
493If you use :class:`JoinableQueue` then you **must** call
494:meth:`JoinableQueue.task_done` for each task removed from the queue or else the
Eli Benderskyd08effe2011-12-31 07:20:26 +0200495semaphore used to count the number of unfinished tasks may eventually overflow,
Benjamin Petersone711caf2008-06-11 16:44:04 +0000496raising an exception.
497
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000498Note that one can also create a shared queue by using a manager object -- see
499:ref:`multiprocessing-managers`.
500
Benjamin Petersone711caf2008-06-11 16:44:04 +0000501.. note::
502
Benjamin Peterson257060a2008-06-28 01:42:41 +0000503 :mod:`multiprocessing` uses the usual :exc:`queue.Empty` and
504 :exc:`queue.Full` exceptions to signal a timeout. They are not available in
Benjamin Petersone711caf2008-06-11 16:44:04 +0000505 the :mod:`multiprocessing` namespace so you need to import them from
Benjamin Peterson257060a2008-06-28 01:42:41 +0000506 :mod:`queue`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000507
508
509.. warning::
510
511 If a process is killed using :meth:`Process.terminate` or :func:`os.kill`
512 while it is trying to use a :class:`Queue`, then the data in the queue is
Eli Benderskyd08effe2011-12-31 07:20:26 +0200513 likely to become corrupted. This may cause any other process to get an
Benjamin Petersone711caf2008-06-11 16:44:04 +0000514 exception when it tries to use the queue later on.
515
516.. warning::
517
518 As mentioned above, if a child process has put items on a queue (and it has
519 not used :meth:`JoinableQueue.cancel_join_thread`), then that process will
520 not terminate until all buffered items have been flushed to the pipe.
521
522 This means that if you try joining that process you may get a deadlock unless
523 you are sure that all items which have been put on the queue have been
524 consumed. Similarly, if the child process is non-daemonic then the parent
Georg Brandl2ee470f2008-07-16 12:55:28 +0000525 process may hang on exit when it tries to join all its non-daemonic children.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000526
527 Note that a queue created using a manager does not have this issue. See
528 :ref:`multiprocessing-programming`.
529
Benjamin Petersone711caf2008-06-11 16:44:04 +0000530For an example of the usage of queues for interprocess communication see
531:ref:`multiprocessing-examples`.
532
533
534.. function:: Pipe([duplex])
535
536 Returns a pair ``(conn1, conn2)`` of :class:`Connection` objects representing
537 the ends of a pipe.
538
539 If *duplex* is ``True`` (the default) then the pipe is bidirectional. If
540 *duplex* is ``False`` then the pipe is unidirectional: ``conn1`` can only be
541 used for receiving messages and ``conn2`` can only be used for sending
542 messages.
543
544
545.. class:: Queue([maxsize])
546
547 Returns a process shared queue implemented using a pipe and a few
548 locks/semaphores. When a process first puts an item on the queue a feeder
549 thread is started which transfers objects from a buffer into the pipe.
550
Benjamin Peterson257060a2008-06-28 01:42:41 +0000551 The usual :exc:`queue.Empty` and :exc:`queue.Full` exceptions from the
Benjamin Petersone711caf2008-06-11 16:44:04 +0000552 standard library's :mod:`Queue` module are raised to signal timeouts.
553
Benjamin Peterson257060a2008-06-28 01:42:41 +0000554 :class:`Queue` implements all the methods of :class:`queue.Queue` except for
555 :meth:`~queue.Queue.task_done` and :meth:`~queue.Queue.join`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000556
557 .. method:: qsize()
558
559 Return the approximate size of the queue. Because of
560 multithreading/multiprocessing semantics, this number is not reliable.
561
562 Note that this may raise :exc:`NotImplementedError` on Unix platforms like
Georg Brandlc575c902008-09-13 17:46:05 +0000563 Mac OS X where ``sem_getvalue()`` is not implemented.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000564
565 .. method:: empty()
566
567 Return ``True`` if the queue is empty, ``False`` otherwise. Because of
568 multithreading/multiprocessing semantics, this is not reliable.
569
570 .. method:: full()
571
572 Return ``True`` if the queue is full, ``False`` otherwise. Because of
573 multithreading/multiprocessing semantics, this is not reliable.
574
Senthil Kumarane969a212011-09-06 00:21:30 +0800575 .. method:: put(obj[, block[, timeout]])
Benjamin Petersone711caf2008-06-11 16:44:04 +0000576
Senthil Kumarane969a212011-09-06 00:21:30 +0800577 Put obj into the queue. If the optional argument *block* is ``True``
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000578 (the default) and *timeout* is ``None`` (the default), block if necessary until
Benjamin Petersone711caf2008-06-11 16:44:04 +0000579 a free slot is available. If *timeout* is a positive number, it blocks at
Benjamin Peterson257060a2008-06-28 01:42:41 +0000580 most *timeout* seconds and raises the :exc:`queue.Full` exception if no
Benjamin Petersone711caf2008-06-11 16:44:04 +0000581 free slot was available within that time. Otherwise (*block* is
582 ``False``), put an item on the queue if a free slot is immediately
Benjamin Peterson257060a2008-06-28 01:42:41 +0000583 available, else raise the :exc:`queue.Full` exception (*timeout* is
Benjamin Petersone711caf2008-06-11 16:44:04 +0000584 ignored in that case).
585
Senthil Kumarane969a212011-09-06 00:21:30 +0800586 .. method:: put_nowait(obj)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000587
Senthil Kumarane969a212011-09-06 00:21:30 +0800588 Equivalent to ``put(obj, False)``.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000589
590 .. method:: get([block[, timeout]])
591
592 Remove and return an item from the queue. If optional args *block* is
593 ``True`` (the default) and *timeout* is ``None`` (the default), block if
594 necessary until an item is available. If *timeout* is a positive number,
Benjamin Peterson257060a2008-06-28 01:42:41 +0000595 it blocks at most *timeout* seconds and raises the :exc:`queue.Empty`
Benjamin Petersone711caf2008-06-11 16:44:04 +0000596 exception if no item was available within that time. Otherwise (block is
597 ``False``), return an item if one is immediately available, else raise the
Benjamin Peterson257060a2008-06-28 01:42:41 +0000598 :exc:`queue.Empty` exception (*timeout* is ignored in that case).
Benjamin Petersone711caf2008-06-11 16:44:04 +0000599
600 .. method:: get_nowait()
601 get_no_wait()
602
603 Equivalent to ``get(False)``.
604
605 :class:`multiprocessing.Queue` has a few additional methods not found in
Georg Brandl2ee470f2008-07-16 12:55:28 +0000606 :class:`queue.Queue`. These methods are usually unnecessary for most
607 code:
Benjamin Petersone711caf2008-06-11 16:44:04 +0000608
609 .. method:: close()
610
611 Indicate that no more data will be put on this queue by the current
612 process. The background thread will quit once it has flushed all buffered
613 data to the pipe. This is called automatically when the queue is garbage
614 collected.
615
616 .. method:: join_thread()
617
618 Join the background thread. This can only be used after :meth:`close` has
619 been called. It blocks until the background thread exits, ensuring that
620 all data in the buffer has been flushed to the pipe.
621
622 By default if a process is not the creator of the queue then on exit it
623 will attempt to join the queue's background thread. The process can call
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000624 :meth:`cancel_join_thread` to make :meth:`join_thread` do nothing.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000625
626 .. method:: cancel_join_thread()
627
628 Prevent :meth:`join_thread` from blocking. In particular, this prevents
629 the background thread from being joined automatically when the process
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000630 exits -- see :meth:`join_thread`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000631
632
Sandro Tosicd778152012-02-15 23:27:00 +0100633.. class:: SimpleQueue()
Sandro Tosi5cb522c2012-02-15 23:14:21 +0100634
635 It is a simplified :class:`Queue` type, very close to a locked :class:`Pipe`.
636
637 .. method:: empty()
638
639 Return ``True`` if the queue is empty, ``False`` otherwise.
640
641 .. method:: get()
642
643 Remove and return an item from the queue.
644
645 .. method:: put(item)
646
647 Put *item* into the queue.
648
649
Benjamin Petersone711caf2008-06-11 16:44:04 +0000650.. class:: JoinableQueue([maxsize])
651
652 :class:`JoinableQueue`, a :class:`Queue` subclass, is a queue which
653 additionally has :meth:`task_done` and :meth:`join` methods.
654
655 .. method:: task_done()
656
657 Indicate that a formerly enqueued task is complete. Used by queue consumer
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000658 threads. For each :meth:`~Queue.get` used to fetch a task, a subsequent
659 call to :meth:`task_done` tells the queue that the processing on the task
660 is complete.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000661
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000662 If a :meth:`~Queue.join` is currently blocking, it will resume when all
663 items have been processed (meaning that a :meth:`task_done` call was
664 received for every item that had been :meth:`~Queue.put` into the queue).
Benjamin Petersone711caf2008-06-11 16:44:04 +0000665
666 Raises a :exc:`ValueError` if called more times than there were items
667 placed in the queue.
668
669
670 .. method:: join()
671
672 Block until all items in the queue have been gotten and processed.
673
674 The count of unfinished tasks goes up whenever an item is added to the
675 queue. The count goes down whenever a consumer thread calls
676 :meth:`task_done` to indicate that the item was retrieved and all work on
677 it is complete. When the count of unfinished tasks drops to zero,
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000678 :meth:`~Queue.join` unblocks.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000679
680
681Miscellaneous
682~~~~~~~~~~~~~
683
684.. function:: active_children()
685
686 Return list of all live children of the current process.
687
688 Calling this has the side affect of "joining" any processes which have
689 already finished.
690
691.. function:: cpu_count()
692
693 Return the number of CPUs in the system. May raise
694 :exc:`NotImplementedError`.
695
696.. function:: current_process()
697
698 Return the :class:`Process` object corresponding to the current process.
699
700 An analogue of :func:`threading.current_thread`.
701
702.. function:: freeze_support()
703
704 Add support for when a program which uses :mod:`multiprocessing` has been
705 frozen to produce a Windows executable. (Has been tested with **py2exe**,
706 **PyInstaller** and **cx_Freeze**.)
707
708 One needs to call this function straight after the ``if __name__ ==
709 '__main__'`` line of the main module. For example::
710
711 from multiprocessing import Process, freeze_support
712
713 def f():
Georg Brandl49702152008-09-29 06:43:45 +0000714 print('hello world!')
Benjamin Petersone711caf2008-06-11 16:44:04 +0000715
716 if __name__ == '__main__':
717 freeze_support()
718 Process(target=f).start()
719
R. David Murray8e8099c2009-04-28 18:02:00 +0000720 If the ``freeze_support()`` line is omitted then trying to run the frozen
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000721 executable will raise :exc:`RuntimeError`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000722
723 If the module is being run normally by the Python interpreter then
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000724 :func:`freeze_support` has no effect.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000725
726.. function:: set_executable()
727
Ezio Melotti0639d5a2009-12-19 23:26:38 +0000728 Sets the path of the Python interpreter to use when starting a child process.
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000729 (By default :data:`sys.executable` is used). Embedders will probably need to
730 do some thing like ::
Benjamin Petersone711caf2008-06-11 16:44:04 +0000731
Eli Benderskyd08effe2011-12-31 07:20:26 +0200732 set_executable(os.path.join(sys.exec_prefix, 'pythonw.exe'))
Benjamin Petersone711caf2008-06-11 16:44:04 +0000733
R. David Murray8e8099c2009-04-28 18:02:00 +0000734 before they can create child processes. (Windows only)
Benjamin Petersone711caf2008-06-11 16:44:04 +0000735
736
737.. note::
738
739 :mod:`multiprocessing` contains no analogues of
740 :func:`threading.active_count`, :func:`threading.enumerate`,
741 :func:`threading.settrace`, :func:`threading.setprofile`,
742 :class:`threading.Timer`, or :class:`threading.local`.
743
744
745Connection Objects
746~~~~~~~~~~~~~~~~~~
747
748Connection objects allow the sending and receiving of picklable objects or
749strings. They can be thought of as message oriented connected sockets.
750
Eli Benderskyd08effe2011-12-31 07:20:26 +0200751Connection objects are usually created using :func:`Pipe` -- see also
Benjamin Petersone711caf2008-06-11 16:44:04 +0000752:ref:`multiprocessing-listeners-clients`.
753
754.. class:: Connection
755
756 .. method:: send(obj)
757
758 Send an object to the other end of the connection which should be read
759 using :meth:`recv`.
760
Benjamin Peterson965ce872009-04-05 21:24:58 +0000761 The object must be picklable. Very large pickles (approximately 32 MB+,
762 though it depends on the OS) may raise a ValueError exception.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000763
764 .. method:: recv()
765
766 Return an object sent from the other end of the connection using
Sandro Tosib52e7a92012-01-07 17:56:58 +0100767 :meth:`send`. Blocks until there its something to receive. Raises
768 :exc:`EOFError` if there is nothing left to receive
Benjamin Petersone711caf2008-06-11 16:44:04 +0000769 and the other end was closed.
770
771 .. method:: fileno()
772
Eli Benderskyd08effe2011-12-31 07:20:26 +0200773 Return the file descriptor or handle used by the connection.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000774
775 .. method:: close()
776
777 Close the connection.
778
779 This is called automatically when the connection is garbage collected.
780
781 .. method:: poll([timeout])
782
783 Return whether there is any data available to be read.
784
785 If *timeout* is not specified then it will return immediately. If
786 *timeout* is a number then this specifies the maximum time in seconds to
787 block. If *timeout* is ``None`` then an infinite timeout is used.
788
Antoine Pitroubdb1cf12012-03-05 19:28:37 +0100789 Note that multiple connection objects may be polled at once by
790 using :func:`multiprocessing.connection.wait`.
791
Benjamin Petersone711caf2008-06-11 16:44:04 +0000792 .. method:: send_bytes(buffer[, offset[, size]])
793
794 Send byte data from an object supporting the buffer interface as a
795 complete message.
796
797 If *offset* is given then data is read from that position in *buffer*. If
Benjamin Peterson965ce872009-04-05 21:24:58 +0000798 *size* is given then that many bytes will be read from buffer. Very large
799 buffers (approximately 32 MB+, though it depends on the OS) may raise a
Eli Benderskyd08effe2011-12-31 07:20:26 +0200800 :exc:`ValueError` exception
Benjamin Petersone711caf2008-06-11 16:44:04 +0000801
802 .. method:: recv_bytes([maxlength])
803
804 Return a complete message of byte data sent from the other end of the
Sandro Tosib52e7a92012-01-07 17:56:58 +0100805 connection as a string. Blocks until there is something to receive.
806 Raises :exc:`EOFError` if there is nothing left
Benjamin Petersone711caf2008-06-11 16:44:04 +0000807 to receive and the other end has closed.
808
809 If *maxlength* is specified and the message is longer than *maxlength*
Antoine Pitrou62ab10a02011-10-12 20:10:51 +0200810 then :exc:`OSError` is raised and the connection will no longer be
Benjamin Petersone711caf2008-06-11 16:44:04 +0000811 readable.
812
Antoine Pitrou62ab10a02011-10-12 20:10:51 +0200813 .. versionchanged:: 3.3
814 This function used to raise a :exc:`IOError`, which is now an
815 alias of :exc:`OSError`.
816
817
Benjamin Petersone711caf2008-06-11 16:44:04 +0000818 .. method:: recv_bytes_into(buffer[, offset])
819
820 Read into *buffer* a complete message of byte data sent from the other end
Sandro Tosib52e7a92012-01-07 17:56:58 +0100821 of the connection and return the number of bytes in the message. Blocks
822 until there is something to receive. Raises
Benjamin Petersone711caf2008-06-11 16:44:04 +0000823 :exc:`EOFError` if there is nothing left to receive and the other end was
824 closed.
825
826 *buffer* must be an object satisfying the writable buffer interface. If
827 *offset* is given then the message will be written into the buffer from
R. David Murray8e8099c2009-04-28 18:02:00 +0000828 that position. Offset must be a non-negative integer less than the
829 length of *buffer* (in bytes).
Benjamin Petersone711caf2008-06-11 16:44:04 +0000830
831 If the buffer is too short then a :exc:`BufferTooShort` exception is
832 raised and the complete message is available as ``e.args[0]`` where ``e``
833 is the exception instance.
834
Antoine Pitrou5438ed12012-04-24 22:56:57 +0200835 .. versionchanged:: 3.3
836 Connection objects themselves can now be transferred between processes
837 using :meth:`Connection.send` and :meth:`Connection.recv`.
838
Benjamin Petersone711caf2008-06-11 16:44:04 +0000839
840For example:
841
R. David Murray8e8099c2009-04-28 18:02:00 +0000842.. doctest::
843
Benjamin Petersone711caf2008-06-11 16:44:04 +0000844 >>> from multiprocessing import Pipe
845 >>> a, b = Pipe()
846 >>> a.send([1, 'hello', None])
847 >>> b.recv()
848 [1, 'hello', None]
Georg Brandl30176892010-10-29 05:22:17 +0000849 >>> b.send_bytes(b'thank you')
Benjamin Petersone711caf2008-06-11 16:44:04 +0000850 >>> a.recv_bytes()
Georg Brandl30176892010-10-29 05:22:17 +0000851 b'thank you'
Benjamin Petersone711caf2008-06-11 16:44:04 +0000852 >>> import array
853 >>> arr1 = array.array('i', range(5))
854 >>> arr2 = array.array('i', [0] * 10)
855 >>> a.send_bytes(arr1)
856 >>> count = b.recv_bytes_into(arr2)
857 >>> assert count == len(arr1) * arr1.itemsize
858 >>> arr2
859 array('i', [0, 1, 2, 3, 4, 0, 0, 0, 0, 0])
860
861
862.. warning::
863
864 The :meth:`Connection.recv` method automatically unpickles the data it
865 receives, which can be a security risk unless you can trust the process
866 which sent the message.
867
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000868 Therefore, unless the connection object was produced using :func:`Pipe` you
869 should only use the :meth:`~Connection.recv` and :meth:`~Connection.send`
870 methods after performing some sort of authentication. See
871 :ref:`multiprocessing-auth-keys`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000872
873.. warning::
874
875 If a process is killed while it is trying to read or write to a pipe then
876 the data in the pipe is likely to become corrupted, because it may become
877 impossible to be sure where the message boundaries lie.
878
879
880Synchronization primitives
881~~~~~~~~~~~~~~~~~~~~~~~~~~
882
883Generally synchronization primitives are not as necessary in a multiprocess
Georg Brandl2ee470f2008-07-16 12:55:28 +0000884program as they are in a multithreaded program. See the documentation for
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000885:mod:`threading` module.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000886
887Note that one can also create synchronization primitives by using a manager
888object -- see :ref:`multiprocessing-managers`.
889
890.. class:: BoundedSemaphore([value])
891
892 A bounded semaphore object: a clone of :class:`threading.BoundedSemaphore`.
893
Georg Brandl592296e2010-05-21 21:48:27 +0000894 (On Mac OS X, this is indistinguishable from :class:`Semaphore` because
Benjamin Petersone711caf2008-06-11 16:44:04 +0000895 ``sem_getvalue()`` is not implemented on that platform).
896
897.. class:: Condition([lock])
898
Benjamin Peterson5289b2b2008-06-28 00:40:54 +0000899 A condition variable: a clone of :class:`threading.Condition`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000900
901 If *lock* is specified then it should be a :class:`Lock` or :class:`RLock`
902 object from :mod:`multiprocessing`.
903
Charles-François Natalic8ce7152012-04-17 18:45:57 +0200904 .. versionchanged:: 3.3
905 The :meth:`wait_for` method was added.
906
Benjamin Petersone711caf2008-06-11 16:44:04 +0000907.. class:: Event()
908
909 A clone of :class:`threading.Event`.
Benjamin Peterson965ce872009-04-05 21:24:58 +0000910 This method returns the state of the internal semaphore on exit, so it
911 will always return ``True`` except if a timeout is given and the operation
912 times out.
913
Raymond Hettinger35a88362009-04-09 00:08:24 +0000914 .. versionchanged:: 3.1
Benjamin Peterson965ce872009-04-05 21:24:58 +0000915 Previously, the method always returned ``None``.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000916
917.. class:: Lock()
918
919 A non-recursive lock object: a clone of :class:`threading.Lock`.
920
921.. class:: RLock()
922
923 A recursive lock object: a clone of :class:`threading.RLock`.
924
925.. class:: Semaphore([value])
926
Ross Lagerwall8fea2e62011-03-14 10:40:15 +0200927 A semaphore object: a clone of :class:`threading.Semaphore`.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000928
929.. note::
930
Richard Oudkerk59d54042012-05-10 16:11:12 +0100931 The :meth:`acquire` and :meth:`wait` methods of each of these types
932 treat negative timeouts as zero timeouts. This differs from
933 :mod:`threading` where, since version 3.2, the equivalent
934 :meth:`acquire` methods treat negative timeouts as infinite
935 timeouts.
936
Georg Brandl592296e2010-05-21 21:48:27 +0000937 On Mac OS X, ``sem_timedwait`` is unsupported, so calling ``acquire()`` with
938 a timeout will emulate that function's behavior using a sleeping loop.
Benjamin Petersone711caf2008-06-11 16:44:04 +0000939
940.. note::
941
942 If the SIGINT signal generated by Ctrl-C arrives while the main thread is
943 blocked by a call to :meth:`BoundedSemaphore.acquire`, :meth:`Lock.acquire`,
944 :meth:`RLock.acquire`, :meth:`Semaphore.acquire`, :meth:`Condition.acquire`
945 or :meth:`Condition.wait` then the call will be immediately interrupted and
946 :exc:`KeyboardInterrupt` will be raised.
947
948 This differs from the behaviour of :mod:`threading` where SIGINT will be
949 ignored while the equivalent blocking calls are in progress.
950
951
952Shared :mod:`ctypes` Objects
953~~~~~~~~~~~~~~~~~~~~~~~~~~~~
954
955It is possible to create shared objects using shared memory which can be
956inherited by child processes.
957
Jesse Nollerb0516a62009-01-18 03:11:38 +0000958.. function:: Value(typecode_or_type, *args[, lock])
Benjamin Petersone711caf2008-06-11 16:44:04 +0000959
960 Return a :mod:`ctypes` object allocated from shared memory. By default the
961 return value is actually a synchronized wrapper for the object.
962
963 *typecode_or_type* determines the type of the returned object: it is either a
964 ctypes type or a one character typecode of the kind used by the :mod:`array`
965 module. *\*args* is passed on to the constructor for the type.
966
967 If *lock* is ``True`` (the default) then a new lock object is created to
968 synchronize access to the value. If *lock* is a :class:`Lock` or
969 :class:`RLock` object then that will be used to synchronize access to the
970 value. If *lock* is ``False`` then access to the returned object will not be
971 automatically protected by a lock, so it will not necessarily be
972 "process-safe".
973
974 Note that *lock* is a keyword-only argument.
975
976.. function:: Array(typecode_or_type, size_or_initializer, *, lock=True)
977
978 Return a ctypes array allocated from shared memory. By default the return
979 value is actually a synchronized wrapper for the array.
980
981 *typecode_or_type* determines the type of the elements of the returned array:
982 it is either a ctypes type or a one character typecode of the kind used by
983 the :mod:`array` module. If *size_or_initializer* is an integer, then it
984 determines the length of the array, and the array will be initially zeroed.
985 Otherwise, *size_or_initializer* is a sequence which is used to initialize
986 the array and whose length determines the length of the array.
987
988 If *lock* is ``True`` (the default) then a new lock object is created to
989 synchronize access to the value. If *lock* is a :class:`Lock` or
990 :class:`RLock` object then that will be used to synchronize access to the
991 value. If *lock* is ``False`` then access to the returned object will not be
992 automatically protected by a lock, so it will not necessarily be
993 "process-safe".
994
995 Note that *lock* is a keyword only argument.
996
Amaury Forgeot d'Arcb0c29162008-11-22 22:18:04 +0000997 Note that an array of :data:`ctypes.c_char` has *value* and *raw*
Benjamin Petersone711caf2008-06-11 16:44:04 +0000998 attributes which allow one to use it to store and retrieve strings.
999
1000
1001The :mod:`multiprocessing.sharedctypes` module
1002>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
1003
1004.. module:: multiprocessing.sharedctypes
1005 :synopsis: Allocate ctypes objects from shared memory.
1006
1007The :mod:`multiprocessing.sharedctypes` module provides functions for allocating
1008:mod:`ctypes` objects from shared memory which can be inherited by child
1009processes.
1010
1011.. note::
1012
Georg Brandl2ee470f2008-07-16 12:55:28 +00001013 Although it is possible to store a pointer in shared memory remember that
1014 this will refer to a location in the address space of a specific process.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001015 However, the pointer is quite likely to be invalid in the context of a second
1016 process and trying to dereference the pointer from the second process may
1017 cause a crash.
1018
1019.. function:: RawArray(typecode_or_type, size_or_initializer)
1020
1021 Return a ctypes array allocated from shared memory.
1022
1023 *typecode_or_type* determines the type of the elements of the returned array:
1024 it is either a ctypes type or a one character typecode of the kind used by
1025 the :mod:`array` module. If *size_or_initializer* is an integer then it
1026 determines the length of the array, and the array will be initially zeroed.
1027 Otherwise *size_or_initializer* is a sequence which is used to initialize the
1028 array and whose length determines the length of the array.
1029
1030 Note that setting and getting an element is potentially non-atomic -- use
1031 :func:`Array` instead to make sure that access is automatically synchronized
1032 using a lock.
1033
1034.. function:: RawValue(typecode_or_type, *args)
1035
1036 Return a ctypes object allocated from shared memory.
1037
1038 *typecode_or_type* determines the type of the returned object: it is either a
1039 ctypes type or a one character typecode of the kind used by the :mod:`array`
Jesse Nollerb0516a62009-01-18 03:11:38 +00001040 module. *\*args* is passed on to the constructor for the type.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001041
1042 Note that setting and getting the value is potentially non-atomic -- use
1043 :func:`Value` instead to make sure that access is automatically synchronized
1044 using a lock.
1045
Amaury Forgeot d'Arcb0c29162008-11-22 22:18:04 +00001046 Note that an array of :data:`ctypes.c_char` has ``value`` and ``raw``
Benjamin Petersone711caf2008-06-11 16:44:04 +00001047 attributes which allow one to use it to store and retrieve strings -- see
1048 documentation for :mod:`ctypes`.
1049
Jesse Nollerb0516a62009-01-18 03:11:38 +00001050.. function:: Array(typecode_or_type, size_or_initializer, *args[, lock])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001051
1052 The same as :func:`RawArray` except that depending on the value of *lock* a
1053 process-safe synchronization wrapper may be returned instead of a raw ctypes
1054 array.
1055
1056 If *lock* is ``True`` (the default) then a new lock object is created to
1057 synchronize access to the value. If *lock* is a :class:`Lock` or
1058 :class:`RLock` object then that will be used to synchronize access to the
1059 value. If *lock* is ``False`` then access to the returned object will not be
1060 automatically protected by a lock, so it will not necessarily be
1061 "process-safe".
1062
1063 Note that *lock* is a keyword-only argument.
1064
1065.. function:: Value(typecode_or_type, *args[, lock])
1066
1067 The same as :func:`RawValue` except that depending on the value of *lock* a
1068 process-safe synchronization wrapper may be returned instead of a raw ctypes
1069 object.
1070
1071 If *lock* is ``True`` (the default) then a new lock object is created to
1072 synchronize access to the value. If *lock* is a :class:`Lock` or
1073 :class:`RLock` object then that will be used to synchronize access to the
1074 value. If *lock* is ``False`` then access to the returned object will not be
1075 automatically protected by a lock, so it will not necessarily be
1076 "process-safe".
1077
1078 Note that *lock* is a keyword-only argument.
1079
1080.. function:: copy(obj)
1081
1082 Return a ctypes object allocated from shared memory which is a copy of the
1083 ctypes object *obj*.
1084
1085.. function:: synchronized(obj[, lock])
1086
1087 Return a process-safe wrapper object for a ctypes object which uses *lock* to
1088 synchronize access. If *lock* is ``None`` (the default) then a
1089 :class:`multiprocessing.RLock` object is created automatically.
1090
1091 A synchronized wrapper will have two methods in addition to those of the
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001092 object it wraps: :meth:`get_obj` returns the wrapped object and
1093 :meth:`get_lock` returns the lock object used for synchronization.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001094
1095 Note that accessing the ctypes object through the wrapper can be a lot slower
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001096 than accessing the raw ctypes object.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001097
1098
1099The table below compares the syntax for creating shared ctypes objects from
1100shared memory with the normal ctypes syntax. (In the table ``MyStruct`` is some
1101subclass of :class:`ctypes.Structure`.)
1102
1103==================== ========================== ===========================
1104ctypes sharedctypes using type sharedctypes using typecode
1105==================== ========================== ===========================
1106c_double(2.4) RawValue(c_double, 2.4) RawValue('d', 2.4)
1107MyStruct(4, 6) RawValue(MyStruct, 4, 6)
1108(c_short * 7)() RawArray(c_short, 7) RawArray('h', 7)
1109(c_int * 3)(9, 2, 8) RawArray(c_int, (9, 2, 8)) RawArray('i', (9, 2, 8))
1110==================== ========================== ===========================
1111
1112
1113Below is an example where a number of ctypes objects are modified by a child
1114process::
1115
1116 from multiprocessing import Process, Lock
1117 from multiprocessing.sharedctypes import Value, Array
1118 from ctypes import Structure, c_double
1119
1120 class Point(Structure):
1121 _fields_ = [('x', c_double), ('y', c_double)]
1122
1123 def modify(n, x, s, A):
1124 n.value **= 2
1125 x.value **= 2
1126 s.value = s.value.upper()
1127 for a in A:
1128 a.x **= 2
1129 a.y **= 2
1130
1131 if __name__ == '__main__':
1132 lock = Lock()
1133
1134 n = Value('i', 7)
R. David Murray8e8099c2009-04-28 18:02:00 +00001135 x = Value(c_double, 1.0/3.0, lock=False)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001136 s = Array('c', 'hello world', lock=lock)
1137 A = Array(Point, [(1.875,-6.25), (-5.75,2.0), (2.375,9.5)], lock=lock)
1138
1139 p = Process(target=modify, args=(n, x, s, A))
1140 p.start()
1141 p.join()
1142
Georg Brandl49702152008-09-29 06:43:45 +00001143 print(n.value)
1144 print(x.value)
1145 print(s.value)
1146 print([(a.x, a.y) for a in A])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001147
1148
Georg Brandl49702152008-09-29 06:43:45 +00001149.. highlight:: none
Benjamin Petersone711caf2008-06-11 16:44:04 +00001150
1151The results printed are ::
1152
1153 49
1154 0.1111111111111111
1155 HELLO WORLD
1156 [(3.515625, 39.0625), (33.0625, 4.0), (5.640625, 90.25)]
1157
Ezio Melottif86b28e2012-04-13 20:50:48 -06001158.. highlight:: python3
Benjamin Petersone711caf2008-06-11 16:44:04 +00001159
1160
1161.. _multiprocessing-managers:
1162
1163Managers
1164~~~~~~~~
1165
1166Managers provide a way to create data which can be shared between different
1167processes. A manager object controls a server process which manages *shared
1168objects*. Other processes can access the shared objects by using proxies.
1169
1170.. function:: multiprocessing.Manager()
1171
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001172 Returns a started :class:`~multiprocessing.managers.SyncManager` object which
1173 can be used for sharing objects between processes. The returned manager
1174 object corresponds to a spawned child process and has methods which will
1175 create shared objects and return corresponding proxies.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001176
1177.. module:: multiprocessing.managers
1178 :synopsis: Share data between process with shared objects.
1179
1180Manager processes will be shutdown as soon as they are garbage collected or
1181their parent process exits. The manager classes are defined in the
1182:mod:`multiprocessing.managers` module:
1183
1184.. class:: BaseManager([address[, authkey]])
1185
1186 Create a BaseManager object.
1187
Benjamin Peterson21896a32010-03-21 22:03:03 +00001188 Once created one should call :meth:`start` or ``get_server().serve_forever()`` to ensure
Benjamin Petersone711caf2008-06-11 16:44:04 +00001189 that the manager object refers to a started manager process.
1190
1191 *address* is the address on which the manager process listens for new
1192 connections. If *address* is ``None`` then an arbitrary one is chosen.
1193
1194 *authkey* is the authentication key which will be used to check the validity
1195 of incoming connections to the server process. If *authkey* is ``None`` then
Benjamin Petersona786b022008-08-25 21:05:21 +00001196 ``current_process().authkey``. Otherwise *authkey* is used and it
Benjamin Petersone711caf2008-06-11 16:44:04 +00001197 must be a string.
1198
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001199 .. method:: start([initializer[, initargs]])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001200
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001201 Start a subprocess to start the manager. If *initializer* is not ``None``
1202 then the subprocess will call ``initializer(*initargs)`` when it starts.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001203
Jesse Noller45239682008-11-28 18:46:19 +00001204 .. method:: get_server()
Georg Brandl48310cd2009-01-03 21:18:54 +00001205
Jesse Noller45239682008-11-28 18:46:19 +00001206 Returns a :class:`Server` object which represents the actual server under
Georg Brandl48310cd2009-01-03 21:18:54 +00001207 the control of the Manager. The :class:`Server` object supports the
R. David Murray8e8099c2009-04-28 18:02:00 +00001208 :meth:`serve_forever` method::
Georg Brandl48310cd2009-01-03 21:18:54 +00001209
Georg Brandl1f01deb2009-01-03 22:47:39 +00001210 >>> from multiprocessing.managers import BaseManager
R. David Murray8e8099c2009-04-28 18:02:00 +00001211 >>> manager = BaseManager(address=('', 50000), authkey='abc')
1212 >>> server = manager.get_server()
1213 >>> server.serve_forever()
Georg Brandl48310cd2009-01-03 21:18:54 +00001214
R. David Murray8e8099c2009-04-28 18:02:00 +00001215 :class:`Server` additionally has an :attr:`address` attribute.
Jesse Noller45239682008-11-28 18:46:19 +00001216
1217 .. method:: connect()
Georg Brandl48310cd2009-01-03 21:18:54 +00001218
R. David Murray8e8099c2009-04-28 18:02:00 +00001219 Connect a local manager object to a remote manager process::
Georg Brandl48310cd2009-01-03 21:18:54 +00001220
Jesse Noller45239682008-11-28 18:46:19 +00001221 >>> from multiprocessing.managers import BaseManager
R. David Murray8e8099c2009-04-28 18:02:00 +00001222 >>> m = BaseManager(address=('127.0.0.1', 5000), authkey='abc')
Jesse Noller45239682008-11-28 18:46:19 +00001223 >>> m.connect()
1224
Benjamin Petersone711caf2008-06-11 16:44:04 +00001225 .. method:: shutdown()
1226
1227 Stop the process used by the manager. This is only available if
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001228 :meth:`start` has been used to start the server process.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001229
1230 This can be called multiple times.
1231
1232 .. method:: register(typeid[, callable[, proxytype[, exposed[, method_to_typeid[, create_method]]]]])
1233
1234 A classmethod which can be used for registering a type or callable with
1235 the manager class.
1236
1237 *typeid* is a "type identifier" which is used to identify a particular
1238 type of shared object. This must be a string.
1239
1240 *callable* is a callable used for creating objects for this type
1241 identifier. If a manager instance will be created using the
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001242 :meth:`from_address` classmethod or if the *create_method* argument is
Benjamin Petersone711caf2008-06-11 16:44:04 +00001243 ``False`` then this can be left as ``None``.
1244
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001245 *proxytype* is a subclass of :class:`BaseProxy` which is used to create
1246 proxies for shared objects with this *typeid*. If ``None`` then a proxy
1247 class is created automatically.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001248
1249 *exposed* is used to specify a sequence of method names which proxies for
1250 this typeid should be allowed to access using
1251 :meth:`BaseProxy._callMethod`. (If *exposed* is ``None`` then
1252 :attr:`proxytype._exposed_` is used instead if it exists.) In the case
1253 where no exposed list is specified, all "public methods" of the shared
1254 object will be accessible. (Here a "public method" means any attribute
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001255 which has a :meth:`__call__` method and whose name does not begin with
Benjamin Petersone711caf2008-06-11 16:44:04 +00001256 ``'_'``.)
1257
1258 *method_to_typeid* is a mapping used to specify the return type of those
1259 exposed methods which should return a proxy. It maps method names to
1260 typeid strings. (If *method_to_typeid* is ``None`` then
1261 :attr:`proxytype._method_to_typeid_` is used instead if it exists.) If a
1262 method's name is not a key of this mapping or if the mapping is ``None``
1263 then the object returned by the method will be copied by value.
1264
1265 *create_method* determines whether a method should be created with name
1266 *typeid* which can be used to tell the server process to create a new
1267 shared object and return a proxy for it. By default it is ``True``.
1268
1269 :class:`BaseManager` instances also have one read-only property:
1270
1271 .. attribute:: address
1272
1273 The address used by the manager.
1274
1275
1276.. class:: SyncManager
1277
1278 A subclass of :class:`BaseManager` which can be used for the synchronization
1279 of processes. Objects of this type are returned by
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001280 :func:`multiprocessing.Manager`.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001281
1282 It also supports creation of shared lists and dictionaries.
1283
1284 .. method:: BoundedSemaphore([value])
1285
1286 Create a shared :class:`threading.BoundedSemaphore` object and return a
1287 proxy for it.
1288
1289 .. method:: Condition([lock])
1290
1291 Create a shared :class:`threading.Condition` object and return a proxy for
1292 it.
1293
1294 If *lock* is supplied then it should be a proxy for a
1295 :class:`threading.Lock` or :class:`threading.RLock` object.
1296
Charles-François Natalic8ce7152012-04-17 18:45:57 +02001297 .. versionchanged:: 3.3
1298 The :meth:`wait_for` method was added.
1299
Benjamin Petersone711caf2008-06-11 16:44:04 +00001300 .. method:: Event()
1301
1302 Create a shared :class:`threading.Event` object and return a proxy for it.
1303
1304 .. method:: Lock()
1305
1306 Create a shared :class:`threading.Lock` object and return a proxy for it.
1307
1308 .. method:: Namespace()
1309
1310 Create a shared :class:`Namespace` object and return a proxy for it.
1311
1312 .. method:: Queue([maxsize])
1313
Benjamin Peterson257060a2008-06-28 01:42:41 +00001314 Create a shared :class:`queue.Queue` object and return a proxy for it.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001315
1316 .. method:: RLock()
1317
1318 Create a shared :class:`threading.RLock` object and return a proxy for it.
1319
1320 .. method:: Semaphore([value])
1321
1322 Create a shared :class:`threading.Semaphore` object and return a proxy for
1323 it.
1324
1325 .. method:: Array(typecode, sequence)
1326
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001327 Create an array and return a proxy for it.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001328
1329 .. method:: Value(typecode, value)
1330
1331 Create an object with a writable ``value`` attribute and return a proxy
1332 for it.
1333
1334 .. method:: dict()
1335 dict(mapping)
1336 dict(sequence)
1337
1338 Create a shared ``dict`` object and return a proxy for it.
1339
1340 .. method:: list()
1341 list(sequence)
1342
1343 Create a shared ``list`` object and return a proxy for it.
1344
Georg Brandl3ed41142010-10-15 16:19:43 +00001345 .. note::
1346
1347 Modifications to mutable values or items in dict and list proxies will not
1348 be propagated through the manager, because the proxy has no way of knowing
1349 when its values or items are modified. To modify such an item, you can
1350 re-assign the modified object to the container proxy::
1351
1352 # create a list proxy and append a mutable object (a dictionary)
1353 lproxy = manager.list()
1354 lproxy.append({})
1355 # now mutate the dictionary
1356 d = lproxy[0]
1357 d['a'] = 1
1358 d['b'] = 2
1359 # at this point, the changes to d are not yet synced, but by
1360 # reassigning the dictionary, the proxy is notified of the change
1361 lproxy[0] = d
1362
Benjamin Petersone711caf2008-06-11 16:44:04 +00001363
1364Namespace objects
1365>>>>>>>>>>>>>>>>>
1366
1367A namespace object has no public methods, but does have writable attributes.
1368Its representation shows the values of its attributes.
1369
1370However, when using a proxy for a namespace object, an attribute beginning with
R. David Murray8e8099c2009-04-28 18:02:00 +00001371``'_'`` will be an attribute of the proxy and not an attribute of the referent:
1372
1373.. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001374
1375 >>> manager = multiprocessing.Manager()
1376 >>> Global = manager.Namespace()
1377 >>> Global.x = 10
1378 >>> Global.y = 'hello'
1379 >>> Global._z = 12.3 # this is an attribute of the proxy
Georg Brandl49702152008-09-29 06:43:45 +00001380 >>> print(Global)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001381 Namespace(x=10, y='hello')
1382
1383
1384Customized managers
1385>>>>>>>>>>>>>>>>>>>
1386
1387To create one's own manager, one creates a subclass of :class:`BaseManager` and
Eli Benderskyd08effe2011-12-31 07:20:26 +02001388uses the :meth:`~BaseManager.register` classmethod to register new types or
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001389callables with the manager class. For example::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001390
1391 from multiprocessing.managers import BaseManager
1392
Éric Araujo28053fb2010-11-22 03:09:19 +00001393 class MathsClass:
Benjamin Petersone711caf2008-06-11 16:44:04 +00001394 def add(self, x, y):
1395 return x + y
1396 def mul(self, x, y):
1397 return x * y
1398
1399 class MyManager(BaseManager):
1400 pass
1401
1402 MyManager.register('Maths', MathsClass)
1403
1404 if __name__ == '__main__':
1405 manager = MyManager()
1406 manager.start()
1407 maths = manager.Maths()
Georg Brandl49702152008-09-29 06:43:45 +00001408 print(maths.add(4, 3)) # prints 7
1409 print(maths.mul(7, 8)) # prints 56
Benjamin Petersone711caf2008-06-11 16:44:04 +00001410
1411
1412Using a remote manager
1413>>>>>>>>>>>>>>>>>>>>>>
1414
1415It is possible to run a manager server on one machine and have clients use it
1416from other machines (assuming that the firewalls involved allow it).
1417
1418Running the following commands creates a server for a single shared queue which
1419remote clients can access::
1420
1421 >>> from multiprocessing.managers import BaseManager
Benjamin Peterson257060a2008-06-28 01:42:41 +00001422 >>> import queue
1423 >>> queue = queue.Queue()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001424 >>> class QueueManager(BaseManager): pass
Jesse Noller45239682008-11-28 18:46:19 +00001425 >>> QueueManager.register('get_queue', callable=lambda:queue)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001426 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
Jesse Noller45239682008-11-28 18:46:19 +00001427 >>> s = m.get_server()
R. David Murray8e8099c2009-04-28 18:02:00 +00001428 >>> s.serve_forever()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001429
1430One client can access the server as follows::
1431
1432 >>> from multiprocessing.managers import BaseManager
1433 >>> class QueueManager(BaseManager): pass
Jesse Noller45239682008-11-28 18:46:19 +00001434 >>> QueueManager.register('get_queue')
1435 >>> m = QueueManager(address=('foo.bar.org', 50000), authkey='abracadabra')
1436 >>> m.connect()
1437 >>> queue = m.get_queue()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001438 >>> queue.put('hello')
1439
1440Another client can also use it::
1441
1442 >>> from multiprocessing.managers import BaseManager
1443 >>> class QueueManager(BaseManager): pass
R. David Murray8e8099c2009-04-28 18:02:00 +00001444 >>> QueueManager.register('get_queue')
1445 >>> m = QueueManager(address=('foo.bar.org', 50000), authkey='abracadabra')
1446 >>> m.connect()
1447 >>> queue = m.get_queue()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001448 >>> queue.get()
1449 'hello'
1450
Georg Brandl48310cd2009-01-03 21:18:54 +00001451Local processes can also access that queue, using the code from above on the
Jesse Noller45239682008-11-28 18:46:19 +00001452client to access it remotely::
1453
1454 >>> from multiprocessing import Process, Queue
1455 >>> from multiprocessing.managers import BaseManager
1456 >>> class Worker(Process):
1457 ... def __init__(self, q):
1458 ... self.q = q
1459 ... super(Worker, self).__init__()
1460 ... def run(self):
1461 ... self.q.put('local hello')
Georg Brandl48310cd2009-01-03 21:18:54 +00001462 ...
Jesse Noller45239682008-11-28 18:46:19 +00001463 >>> queue = Queue()
1464 >>> w = Worker(queue)
1465 >>> w.start()
1466 >>> class QueueManager(BaseManager): pass
Georg Brandl48310cd2009-01-03 21:18:54 +00001467 ...
Jesse Noller45239682008-11-28 18:46:19 +00001468 >>> QueueManager.register('get_queue', callable=lambda: queue)
1469 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
1470 >>> s = m.get_server()
1471 >>> s.serve_forever()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001472
1473Proxy Objects
1474~~~~~~~~~~~~~
1475
1476A proxy is an object which *refers* to a shared object which lives (presumably)
1477in a different process. The shared object is said to be the *referent* of the
1478proxy. Multiple proxy objects may have the same referent.
1479
1480A proxy object has methods which invoke corresponding methods of its referent
1481(although not every method of the referent will necessarily be available through
1482the proxy). A proxy can usually be used in most of the same ways that its
R. David Murray8e8099c2009-04-28 18:02:00 +00001483referent can:
1484
1485.. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001486
1487 >>> from multiprocessing import Manager
1488 >>> manager = Manager()
1489 >>> l = manager.list([i*i for i in range(10)])
Georg Brandl49702152008-09-29 06:43:45 +00001490 >>> print(l)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001491 [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
Georg Brandl49702152008-09-29 06:43:45 +00001492 >>> print(repr(l))
R. David Murray8e8099c2009-04-28 18:02:00 +00001493 <ListProxy object, typeid 'list' at 0x...>
Benjamin Petersone711caf2008-06-11 16:44:04 +00001494 >>> l[4]
1495 16
1496 >>> l[2:5]
1497 [4, 9, 16]
1498
1499Notice that applying :func:`str` to a proxy will return the representation of
1500the referent, whereas applying :func:`repr` will return the representation of
1501the proxy.
1502
1503An important feature of proxy objects is that they are picklable so they can be
1504passed between processes. Note, however, that if a proxy is sent to the
1505corresponding manager's process then unpickling it will produce the referent
R. David Murray8e8099c2009-04-28 18:02:00 +00001506itself. This means, for example, that one shared object can contain a second:
1507
1508.. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001509
1510 >>> a = manager.list()
1511 >>> b = manager.list()
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001512 >>> a.append(b) # referent of a now contains referent of b
Georg Brandl49702152008-09-29 06:43:45 +00001513 >>> print(a, b)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001514 [[]] []
1515 >>> b.append('hello')
Georg Brandl49702152008-09-29 06:43:45 +00001516 >>> print(a, b)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001517 [['hello']] ['hello']
1518
1519.. note::
1520
1521 The proxy types in :mod:`multiprocessing` do nothing to support comparisons
R. David Murray8e8099c2009-04-28 18:02:00 +00001522 by value. So, for instance, we have:
Benjamin Petersone711caf2008-06-11 16:44:04 +00001523
R. David Murray8e8099c2009-04-28 18:02:00 +00001524 .. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001525
R. David Murray8e8099c2009-04-28 18:02:00 +00001526 >>> manager.list([1,2,3]) == [1,2,3]
1527 False
1528
1529 One should just use a copy of the referent instead when making comparisons.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001530
1531.. class:: BaseProxy
1532
1533 Proxy objects are instances of subclasses of :class:`BaseProxy`.
1534
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001535 .. method:: _callmethod(methodname[, args[, kwds]])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001536
1537 Call and return the result of a method of the proxy's referent.
1538
1539 If ``proxy`` is a proxy whose referent is ``obj`` then the expression ::
1540
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001541 proxy._callmethod(methodname, args, kwds)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001542
1543 will evaluate the expression ::
1544
1545 getattr(obj, methodname)(*args, **kwds)
1546
1547 in the manager's process.
1548
1549 The returned value will be a copy of the result of the call or a proxy to
1550 a new shared object -- see documentation for the *method_to_typeid*
1551 argument of :meth:`BaseManager.register`.
1552
Ezio Melottie130a522011-10-19 10:58:56 +03001553 If an exception is raised by the call, then is re-raised by
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001554 :meth:`_callmethod`. If some other exception is raised in the manager's
Benjamin Petersone711caf2008-06-11 16:44:04 +00001555 process then this is converted into a :exc:`RemoteError` exception and is
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001556 raised by :meth:`_callmethod`.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001557
1558 Note in particular that an exception will be raised if *methodname* has
1559 not been *exposed*
1560
R. David Murray8e8099c2009-04-28 18:02:00 +00001561 An example of the usage of :meth:`_callmethod`:
1562
1563 .. doctest::
Benjamin Petersone711caf2008-06-11 16:44:04 +00001564
1565 >>> l = manager.list(range(10))
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001566 >>> l._callmethod('__len__')
Benjamin Petersone711caf2008-06-11 16:44:04 +00001567 10
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001568 >>> l._callmethod('__getslice__', (2, 7)) # equiv to `l[2:7]`
Benjamin Petersone711caf2008-06-11 16:44:04 +00001569 [2, 3, 4, 5, 6]
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001570 >>> l._callmethod('__getitem__', (20,)) # equiv to `l[20]`
Benjamin Petersone711caf2008-06-11 16:44:04 +00001571 Traceback (most recent call last):
1572 ...
1573 IndexError: list index out of range
1574
Benjamin Peterson6ebe78f2008-12-21 00:06:59 +00001575 .. method:: _getvalue()
Benjamin Petersone711caf2008-06-11 16:44:04 +00001576
1577 Return a copy of the referent.
1578
1579 If the referent is unpicklable then this will raise an exception.
1580
1581 .. method:: __repr__
1582
1583 Return a representation of the proxy object.
1584
1585 .. method:: __str__
1586
1587 Return the representation of the referent.
1588
1589
1590Cleanup
1591>>>>>>>
1592
1593A proxy object uses a weakref callback so that when it gets garbage collected it
1594deregisters itself from the manager which owns its referent.
1595
1596A shared object gets deleted from the manager process when there are no longer
1597any proxies referring to it.
1598
1599
1600Process Pools
1601~~~~~~~~~~~~~
1602
1603.. module:: multiprocessing.pool
1604 :synopsis: Create pools of processes.
1605
1606One can create a pool of processes which will carry out tasks submitted to it
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001607with the :class:`Pool` class.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001608
Jesse Noller1f0b6582010-01-27 03:36:01 +00001609.. class:: multiprocessing.Pool([processes[, initializer[, initargs[, maxtasksperchild]]]])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001610
1611 A process pool object which controls a pool of worker processes to which jobs
1612 can be submitted. It supports asynchronous results with timeouts and
1613 callbacks and has a parallel map implementation.
1614
1615 *processes* is the number of worker processes to use. If *processes* is
1616 ``None`` then the number returned by :func:`cpu_count` is used. If
1617 *initializer* is not ``None`` then each worker process will call
1618 ``initializer(*initargs)`` when it starts.
1619
Georg Brandl17ef0d52010-10-17 06:21:59 +00001620 .. versionadded:: 3.2
1621 *maxtasksperchild* is the number of tasks a worker process can complete
1622 before it will exit and be replaced with a fresh worker process, to enable
1623 unused resources to be freed. The default *maxtasksperchild* is None, which
1624 means worker processes will live as long as the pool.
Jesse Noller1f0b6582010-01-27 03:36:01 +00001625
1626 .. note::
1627
Georg Brandl17ef0d52010-10-17 06:21:59 +00001628 Worker processes within a :class:`Pool` typically live for the complete
1629 duration of the Pool's work queue. A frequent pattern found in other
1630 systems (such as Apache, mod_wsgi, etc) to free resources held by
1631 workers is to allow a worker within a pool to complete only a set
1632 amount of work before being exiting, being cleaned up and a new
1633 process spawned to replace the old one. The *maxtasksperchild*
1634 argument to the :class:`Pool` exposes this ability to the end user.
Jesse Noller1f0b6582010-01-27 03:36:01 +00001635
Benjamin Petersone711caf2008-06-11 16:44:04 +00001636 .. method:: apply(func[, args[, kwds]])
1637
Benjamin Peterson37d2fe02008-10-24 22:28:58 +00001638 Call *func* with arguments *args* and keyword arguments *kwds*. It blocks
Eli Benderskyd08effe2011-12-31 07:20:26 +02001639 until the result is ready. Given this blocks, :meth:`apply_async` is
1640 better suited for performing work in parallel. Additionally, *func*
1641 is only executed in one of the workers of the pool.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001642
Ask Solem1d3b8932010-11-09 21:36:56 +00001643 .. method:: apply_async(func[, args[, kwds[, callback[, error_callback]]]])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001644
1645 A variant of the :meth:`apply` method which returns a result object.
1646
1647 If *callback* is specified then it should be a callable which accepts a
1648 single argument. When the result becomes ready *callback* is applied to
Ask Solem1d3b8932010-11-09 21:36:56 +00001649 it, that is unless the call failed, in which case the *error_callback*
1650 is applied instead
1651
1652 If *error_callback* is specified then it should be a callable which
1653 accepts a single argument. If the target function fails, then
1654 the *error_callback* is called with the exception instance.
1655
1656 Callbacks should complete immediately since otherwise the thread which
1657 handles the results will get blocked.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001658
1659 .. method:: map(func, iterable[, chunksize])
1660
Georg Brandl22b34312009-07-26 14:54:51 +00001661 A parallel equivalent of the :func:`map` built-in function (it supports only
Eli Benderskyd08effe2011-12-31 07:20:26 +02001662 one *iterable* argument though). It blocks until the result is ready.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001663
1664 This method chops the iterable into a number of chunks which it submits to
1665 the process pool as separate tasks. The (approximate) size of these
1666 chunks can be specified by setting *chunksize* to a positive integer.
1667
Sandro Tosidb79e952011-08-08 16:38:13 +02001668 .. method:: map_async(func, iterable[, chunksize[, callback[, error_callback]]])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001669
Georg Brandl502d9a52009-07-26 15:02:41 +00001670 A variant of the :meth:`.map` method which returns a result object.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001671
1672 If *callback* is specified then it should be a callable which accepts a
1673 single argument. When the result becomes ready *callback* is applied to
Ask Solem1d3b8932010-11-09 21:36:56 +00001674 it, that is unless the call failed, in which case the *error_callback*
1675 is applied instead
1676
1677 If *error_callback* is specified then it should be a callable which
1678 accepts a single argument. If the target function fails, then
1679 the *error_callback* is called with the exception instance.
1680
1681 Callbacks should complete immediately since otherwise the thread which
1682 handles the results will get blocked.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001683
1684 .. method:: imap(func, iterable[, chunksize])
1685
Georg Brandl92905032008-11-22 08:51:39 +00001686 A lazier version of :meth:`map`.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001687
1688 The *chunksize* argument is the same as the one used by the :meth:`.map`
1689 method. For very long iterables using a large value for *chunksize* can
Ezio Melottie130a522011-10-19 10:58:56 +03001690 make the job complete **much** faster than using the default value of
Benjamin Petersone711caf2008-06-11 16:44:04 +00001691 ``1``.
1692
Georg Brandl502d9a52009-07-26 15:02:41 +00001693 Also if *chunksize* is ``1`` then the :meth:`!next` method of the iterator
Benjamin Petersone711caf2008-06-11 16:44:04 +00001694 returned by the :meth:`imap` method has an optional *timeout* parameter:
1695 ``next(timeout)`` will raise :exc:`multiprocessing.TimeoutError` if the
1696 result cannot be returned within *timeout* seconds.
1697
1698 .. method:: imap_unordered(func, iterable[, chunksize])
1699
1700 The same as :meth:`imap` except that the ordering of the results from the
1701 returned iterator should be considered arbitrary. (Only when there is
1702 only one worker process is the order guaranteed to be "correct".)
1703
Antoine Pitroude911b22011-12-21 11:03:24 +01001704 .. method:: starmap(func, iterable[, chunksize])
1705
1706 Like :meth:`map` except that the elements of the `iterable` are expected
1707 to be iterables that are unpacked as arguments.
1708
1709 Hence an `iterable` of `[(1,2), (3, 4)]` results in `[func(1,2),
1710 func(3,4)]`.
1711
1712 .. versionadded:: 3.3
1713
1714 .. method:: starmap_async(func, iterable[, chunksize[, callback[, error_back]]])
1715
1716 A combination of :meth:`starmap` and :meth:`map_async` that iterates over
1717 `iterable` of iterables and calls `func` with the iterables unpacked.
1718 Returns a result object.
1719
1720 .. versionadded:: 3.3
1721
Benjamin Petersone711caf2008-06-11 16:44:04 +00001722 .. method:: close()
1723
1724 Prevents any more tasks from being submitted to the pool. Once all the
1725 tasks have been completed the worker processes will exit.
1726
1727 .. method:: terminate()
1728
1729 Stops the worker processes immediately without completing outstanding
1730 work. When the pool object is garbage collected :meth:`terminate` will be
1731 called immediately.
1732
1733 .. method:: join()
1734
1735 Wait for the worker processes to exit. One must call :meth:`close` or
1736 :meth:`terminate` before using :meth:`join`.
1737
1738
1739.. class:: AsyncResult
1740
1741 The class of the result returned by :meth:`Pool.apply_async` and
1742 :meth:`Pool.map_async`.
1743
Georg Brandle3d70ae2008-11-22 08:54:21 +00001744 .. method:: get([timeout])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001745
1746 Return the result when it arrives. If *timeout* is not ``None`` and the
1747 result does not arrive within *timeout* seconds then
1748 :exc:`multiprocessing.TimeoutError` is raised. If the remote call raised
1749 an exception then that exception will be reraised by :meth:`get`.
1750
1751 .. method:: wait([timeout])
1752
1753 Wait until the result is available or until *timeout* seconds pass.
1754
1755 .. method:: ready()
1756
1757 Return whether the call has completed.
1758
1759 .. method:: successful()
1760
1761 Return whether the call completed without raising an exception. Will
1762 raise :exc:`AssertionError` if the result is not ready.
1763
1764The following example demonstrates the use of a pool::
1765
1766 from multiprocessing import Pool
1767
1768 def f(x):
1769 return x*x
1770
1771 if __name__ == '__main__':
1772 pool = Pool(processes=4) # start 4 worker processes
1773
Georg Brandle3d70ae2008-11-22 08:54:21 +00001774 result = pool.apply_async(f, (10,)) # evaluate "f(10)" asynchronously
Georg Brandl49702152008-09-29 06:43:45 +00001775 print(result.get(timeout=1)) # prints "100" unless your computer is *very* slow
Benjamin Petersone711caf2008-06-11 16:44:04 +00001776
Georg Brandl49702152008-09-29 06:43:45 +00001777 print(pool.map(f, range(10))) # prints "[0, 1, 4,..., 81]"
Benjamin Petersone711caf2008-06-11 16:44:04 +00001778
1779 it = pool.imap(f, range(10))
Georg Brandl49702152008-09-29 06:43:45 +00001780 print(next(it)) # prints "0"
1781 print(next(it)) # prints "1"
1782 print(it.next(timeout=1)) # prints "4" unless your computer is *very* slow
Benjamin Petersone711caf2008-06-11 16:44:04 +00001783
1784 import time
Georg Brandle3d70ae2008-11-22 08:54:21 +00001785 result = pool.apply_async(time.sleep, (10,))
Georg Brandl49702152008-09-29 06:43:45 +00001786 print(result.get(timeout=1)) # raises TimeoutError
Benjamin Petersone711caf2008-06-11 16:44:04 +00001787
1788
1789.. _multiprocessing-listeners-clients:
1790
1791Listeners and Clients
1792~~~~~~~~~~~~~~~~~~~~~
1793
1794.. module:: multiprocessing.connection
1795 :synopsis: API for dealing with sockets.
1796
1797Usually message passing between processes is done using queues or by using
1798:class:`Connection` objects returned by :func:`Pipe`.
1799
1800However, the :mod:`multiprocessing.connection` module allows some extra
1801flexibility. It basically gives a high level message oriented API for dealing
Antoine Pitroubdb1cf12012-03-05 19:28:37 +01001802with sockets or Windows named pipes. It also has support for *digest
1803authentication* using the :mod:`hmac` module, and for polling
1804multiple connections at the same time.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001805
1806
1807.. function:: deliver_challenge(connection, authkey)
1808
1809 Send a randomly generated message to the other end of the connection and wait
1810 for a reply.
1811
1812 If the reply matches the digest of the message using *authkey* as the key
1813 then a welcome message is sent to the other end of the connection. Otherwise
1814 :exc:`AuthenticationError` is raised.
1815
1816.. function:: answerChallenge(connection, authkey)
1817
1818 Receive a message, calculate the digest of the message using *authkey* as the
1819 key, and then send the digest back.
1820
1821 If a welcome message is not received, then :exc:`AuthenticationError` is
1822 raised.
1823
1824.. function:: Client(address[, family[, authenticate[, authkey]]])
1825
1826 Attempt to set up a connection to the listener which is using address
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00001827 *address*, returning a :class:`~multiprocessing.Connection`.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001828
1829 The type of the connection is determined by *family* argument, but this can
1830 generally be omitted since it can usually be inferred from the format of
1831 *address*. (See :ref:`multiprocessing-address-formats`)
1832
Alexandre Vassalottic57a84f2009-07-17 12:07:01 +00001833 If *authenticate* is ``True`` or *authkey* is a string then digest
Benjamin Petersone711caf2008-06-11 16:44:04 +00001834 authentication is used. The key used for authentication will be either
Benjamin Petersona786b022008-08-25 21:05:21 +00001835 *authkey* or ``current_process().authkey)`` if *authkey* is ``None``.
Benjamin Petersone711caf2008-06-11 16:44:04 +00001836 If authentication fails then :exc:`AuthenticationError` is raised. See
1837 :ref:`multiprocessing-auth-keys`.
1838
1839.. class:: Listener([address[, family[, backlog[, authenticate[, authkey]]]]])
1840
1841 A wrapper for a bound socket or Windows named pipe which is 'listening' for
1842 connections.
1843
1844 *address* is the address to be used by the bound socket or named pipe of the
1845 listener object.
1846
Benjamin Petersond23f8222009-04-05 19:13:16 +00001847 .. note::
1848
1849 If an address of '0.0.0.0' is used, the address will not be a connectable
1850 end point on Windows. If you require a connectable end-point,
1851 you should use '127.0.0.1'.
1852
Benjamin Petersone711caf2008-06-11 16:44:04 +00001853 *family* is the type of socket (or named pipe) to use. This can be one of
1854 the strings ``'AF_INET'`` (for a TCP socket), ``'AF_UNIX'`` (for a Unix
1855 domain socket) or ``'AF_PIPE'`` (for a Windows named pipe). Of these only
1856 the first is guaranteed to be available. If *family* is ``None`` then the
1857 family is inferred from the format of *address*. If *address* is also
1858 ``None`` then a default is chosen. This default is the family which is
1859 assumed to be the fastest available. See
1860 :ref:`multiprocessing-address-formats`. Note that if *family* is
1861 ``'AF_UNIX'`` and address is ``None`` then the socket will be created in a
1862 private temporary directory created using :func:`tempfile.mkstemp`.
1863
1864 If the listener object uses a socket then *backlog* (1 by default) is passed
1865 to the :meth:`listen` method of the socket once it has been bound.
1866
1867 If *authenticate* is ``True`` (``False`` by default) or *authkey* is not
1868 ``None`` then digest authentication is used.
1869
1870 If *authkey* is a string then it will be used as the authentication key;
1871 otherwise it must be *None*.
1872
1873 If *authkey* is ``None`` and *authenticate* is ``True`` then
Benjamin Petersona786b022008-08-25 21:05:21 +00001874 ``current_process().authkey`` is used as the authentication key. If
Alexandre Vassalottic57a84f2009-07-17 12:07:01 +00001875 *authkey* is ``None`` and *authenticate* is ``False`` then no
Benjamin Petersone711caf2008-06-11 16:44:04 +00001876 authentication is done. If authentication fails then
1877 :exc:`AuthenticationError` is raised. See :ref:`multiprocessing-auth-keys`.
1878
1879 .. method:: accept()
1880
1881 Accept a connection on the bound socket or named pipe of the listener
1882 object and return a :class:`Connection` object. If authentication is
1883 attempted and fails, then :exc:`AuthenticationError` is raised.
1884
1885 .. method:: close()
1886
1887 Close the bound socket or named pipe of the listener object. This is
1888 called automatically when the listener is garbage collected. However it
1889 is advisable to call it explicitly.
1890
1891 Listener objects have the following read-only properties:
1892
1893 .. attribute:: address
1894
1895 The address which is being used by the Listener object.
1896
1897 .. attribute:: last_accepted
1898
1899 The address from which the last accepted connection came. If this is
1900 unavailable then it is ``None``.
1901
Antoine Pitroubdb1cf12012-03-05 19:28:37 +01001902.. function:: wait(object_list, timeout=None)
1903
1904 Wait till an object in *object_list* is ready. Returns the list of
1905 those objects in *object_list* which are ready. If *timeout* is a
1906 float then the call blocks for at most that many seconds. If
1907 *timeout* is ``None`` then it will block for an unlimited period.
Richard Oudkerk59d54042012-05-10 16:11:12 +01001908 A negative timeout is equivalent to a zero timeout.
Antoine Pitroubdb1cf12012-03-05 19:28:37 +01001909
1910 For both Unix and Windows, an object can appear in *object_list* if
1911 it is
1912
1913 * a readable :class:`~multiprocessing.Connection` object;
1914 * a connected and readable :class:`socket.socket` object; or
1915 * the :attr:`~multiprocessing.Process.sentinel` attribute of a
1916 :class:`~multiprocessing.Process` object.
1917
1918 A connection or socket object is ready when there is data available
1919 to be read from it, or the other end has been closed.
1920
1921 **Unix**: ``wait(object_list, timeout)`` almost equivalent
1922 ``select.select(object_list, [], [], timeout)``. The difference is
1923 that, if :func:`select.select` is interrupted by a signal, it can
1924 raise :exc:`OSError` with an error number of ``EINTR``, whereas
1925 :func:`wait` will not.
1926
1927 **Windows**: An item in *object_list* must either be an integer
1928 handle which is waitable (according to the definition used by the
1929 documentation of the Win32 function ``WaitForMultipleObjects()``)
1930 or it can be an object with a :meth:`fileno` method which returns a
1931 socket handle or pipe handle. (Note that pipe handles and socket
1932 handles are **not** waitable handles.)
1933
1934 .. versionadded:: 3.3
Benjamin Petersone711caf2008-06-11 16:44:04 +00001935
1936The module defines two exceptions:
1937
1938.. exception:: AuthenticationError
1939
1940 Exception raised when there is an authentication error.
1941
Benjamin Petersone711caf2008-06-11 16:44:04 +00001942
1943**Examples**
1944
1945The following server code creates a listener which uses ``'secret password'`` as
1946an authentication key. It then waits for a connection and sends some data to
1947the client::
1948
1949 from multiprocessing.connection import Listener
1950 from array import array
1951
1952 address = ('localhost', 6000) # family is deduced to be 'AF_INET'
Senthil Kumaran79941b52010-10-10 06:13:49 +00001953 listener = Listener(address, authkey=b'secret password')
Benjamin Petersone711caf2008-06-11 16:44:04 +00001954
1955 conn = listener.accept()
Georg Brandl49702152008-09-29 06:43:45 +00001956 print('connection accepted from', listener.last_accepted)
Benjamin Petersone711caf2008-06-11 16:44:04 +00001957
1958 conn.send([2.25, None, 'junk', float])
1959
Senthil Kumaran79941b52010-10-10 06:13:49 +00001960 conn.send_bytes(b'hello')
Benjamin Petersone711caf2008-06-11 16:44:04 +00001961
1962 conn.send_bytes(array('i', [42, 1729]))
1963
1964 conn.close()
1965 listener.close()
1966
1967The following code connects to the server and receives some data from the
1968server::
1969
1970 from multiprocessing.connection import Client
1971 from array import array
1972
1973 address = ('localhost', 6000)
Senthil Kumaran79941b52010-10-10 06:13:49 +00001974 conn = Client(address, authkey=b'secret password')
Benjamin Petersone711caf2008-06-11 16:44:04 +00001975
Georg Brandl49702152008-09-29 06:43:45 +00001976 print(conn.recv()) # => [2.25, None, 'junk', float]
Benjamin Petersone711caf2008-06-11 16:44:04 +00001977
Georg Brandl49702152008-09-29 06:43:45 +00001978 print(conn.recv_bytes()) # => 'hello'
Benjamin Petersone711caf2008-06-11 16:44:04 +00001979
1980 arr = array('i', [0, 0, 0, 0, 0])
Georg Brandl49702152008-09-29 06:43:45 +00001981 print(conn.recv_bytes_into(arr)) # => 8
1982 print(arr) # => array('i', [42, 1729, 0, 0, 0])
Benjamin Petersone711caf2008-06-11 16:44:04 +00001983
1984 conn.close()
1985
Antoine Pitroubdb1cf12012-03-05 19:28:37 +01001986The following code uses :func:`~multiprocessing.connection.wait` to
1987wait for messages from multiple processes at once::
1988
1989 import time, random
1990 from multiprocessing import Process, Pipe, current_process
1991 from multiprocessing.connection import wait
1992
1993 def foo(w):
1994 for i in range(10):
1995 w.send((i, current_process().name))
1996 w.close()
1997
1998 if __name__ == '__main__':
1999 readers = []
2000
2001 for i in range(4):
2002 r, w = Pipe(duplex=False)
2003 readers.append(r)
2004 p = Process(target=foo, args=(w,))
2005 p.start()
2006 # We close the writable end of the pipe now to be sure that
2007 # p is the only process which owns a handle for it. This
2008 # ensures that when p closes its handle for the writable end,
2009 # wait() will promptly report the readable end as being ready.
2010 w.close()
2011
2012 while readers:
2013 for r in wait(readers):
2014 try:
2015 msg = r.recv()
2016 except EOFError:
2017 readers.remove(r)
2018 else:
2019 print(msg)
2020
Benjamin Petersone711caf2008-06-11 16:44:04 +00002021
2022.. _multiprocessing-address-formats:
2023
2024Address Formats
2025>>>>>>>>>>>>>>>
2026
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002027* An ``'AF_INET'`` address is a tuple of the form ``(hostname, port)`` where
Benjamin Petersone711caf2008-06-11 16:44:04 +00002028 *hostname* is a string and *port* is an integer.
2029
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002030* An ``'AF_UNIX'`` address is a string representing a filename on the
Benjamin Petersone711caf2008-06-11 16:44:04 +00002031 filesystem.
2032
2033* An ``'AF_PIPE'`` address is a string of the form
Benjamin Petersonda10d3b2009-01-01 00:23:30 +00002034 :samp:`r'\\\\.\\pipe\\{PipeName}'`. To use :func:`Client` to connect to a named
Georg Brandl1f01deb2009-01-03 22:47:39 +00002035 pipe on a remote computer called *ServerName* one should use an address of the
Benjamin Peterson28d88b42009-01-09 03:03:23 +00002036 form :samp:`r'\\\\{ServerName}\\pipe\\{PipeName}'` instead.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002037
2038Note that any string beginning with two backslashes is assumed by default to be
2039an ``'AF_PIPE'`` address rather than an ``'AF_UNIX'`` address.
2040
2041
2042.. _multiprocessing-auth-keys:
2043
2044Authentication keys
2045~~~~~~~~~~~~~~~~~~~
2046
2047When one uses :meth:`Connection.recv`, the data received is automatically
2048unpickled. Unfortunately unpickling data from an untrusted source is a security
2049risk. Therefore :class:`Listener` and :func:`Client` use the :mod:`hmac` module
2050to provide digest authentication.
2051
2052An authentication key is a string which can be thought of as a password: once a
2053connection is established both ends will demand proof that the other knows the
2054authentication key. (Demonstrating that both ends are using the same key does
2055**not** involve sending the key over the connection.)
2056
2057If authentication is requested but do authentication key is specified then the
Benjamin Petersona786b022008-08-25 21:05:21 +00002058return value of ``current_process().authkey`` is used (see
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002059:class:`~multiprocessing.Process`). This value will automatically inherited by
2060any :class:`~multiprocessing.Process` object that the current process creates.
2061This means that (by default) all processes of a multi-process program will share
2062a single authentication key which can be used when setting up connections
Benjamin Petersond23f8222009-04-05 19:13:16 +00002063between themselves.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002064
2065Suitable authentication keys can also be generated by using :func:`os.urandom`.
2066
2067
2068Logging
2069~~~~~~~
2070
2071Some support for logging is available. Note, however, that the :mod:`logging`
2072package does not use process shared locks so it is possible (depending on the
2073handler type) for messages from different processes to get mixed up.
2074
2075.. currentmodule:: multiprocessing
2076.. function:: get_logger()
2077
2078 Returns the logger used by :mod:`multiprocessing`. If necessary, a new one
2079 will be created.
2080
Jesse Noller41faa542009-01-25 03:45:53 +00002081 When first created the logger has level :data:`logging.NOTSET` and no
2082 default handler. Messages sent to this logger will not by default propagate
2083 to the root logger.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002084
2085 Note that on Windows child processes will only inherit the level of the
2086 parent process's logger -- any other customization of the logger will not be
2087 inherited.
2088
Jesse Noller41faa542009-01-25 03:45:53 +00002089.. currentmodule:: multiprocessing
2090.. function:: log_to_stderr()
2091
2092 This function performs a call to :func:`get_logger` but in addition to
2093 returning the logger created by get_logger, it adds a handler which sends
2094 output to :data:`sys.stderr` using format
2095 ``'[%(levelname)s/%(processName)s] %(message)s'``.
2096
Benjamin Petersone711caf2008-06-11 16:44:04 +00002097Below is an example session with logging turned on::
2098
Benjamin Peterson206e3072008-10-19 14:07:49 +00002099 >>> import multiprocessing, logging
Jesse Noller41faa542009-01-25 03:45:53 +00002100 >>> logger = multiprocessing.log_to_stderr()
Benjamin Petersone711caf2008-06-11 16:44:04 +00002101 >>> logger.setLevel(logging.INFO)
2102 >>> logger.warning('doomed')
2103 [WARNING/MainProcess] doomed
Benjamin Peterson206e3072008-10-19 14:07:49 +00002104 >>> m = multiprocessing.Manager()
R. David Murray8e8099c2009-04-28 18:02:00 +00002105 [INFO/SyncManager-...] child process calling self.run()
2106 [INFO/SyncManager-...] created temp directory /.../pymp-...
2107 [INFO/SyncManager-...] manager serving at '/.../listener-...'
Benjamin Petersone711caf2008-06-11 16:44:04 +00002108 >>> del m
2109 [INFO/MainProcess] sending shutdown message to manager
R. David Murray8e8099c2009-04-28 18:02:00 +00002110 [INFO/SyncManager-...] manager exiting with exitcode 0
Benjamin Petersone711caf2008-06-11 16:44:04 +00002111
Jesse Noller41faa542009-01-25 03:45:53 +00002112In addition to having these two logging functions, the multiprocessing also
2113exposes two additional logging level attributes. These are :const:`SUBWARNING`
2114and :const:`SUBDEBUG`. The table below illustrates where theses fit in the
2115normal level hierarchy.
2116
2117+----------------+----------------+
2118| Level | Numeric value |
2119+================+================+
2120| ``SUBWARNING`` | 25 |
2121+----------------+----------------+
2122| ``SUBDEBUG`` | 5 |
2123+----------------+----------------+
2124
2125For a full table of logging levels, see the :mod:`logging` module.
2126
2127These additional logging levels are used primarily for certain debug messages
2128within the multiprocessing module. Below is the same example as above, except
2129with :const:`SUBDEBUG` enabled::
2130
2131 >>> import multiprocessing, logging
2132 >>> logger = multiprocessing.log_to_stderr()
2133 >>> logger.setLevel(multiprocessing.SUBDEBUG)
2134 >>> logger.warning('doomed')
2135 [WARNING/MainProcess] doomed
2136 >>> m = multiprocessing.Manager()
R. David Murray8e8099c2009-04-28 18:02:00 +00002137 [INFO/SyncManager-...] child process calling self.run()
2138 [INFO/SyncManager-...] created temp directory /.../pymp-...
2139 [INFO/SyncManager-...] manager serving at '/.../pymp-djGBXN/listener-...'
Jesse Noller41faa542009-01-25 03:45:53 +00002140 >>> del m
2141 [SUBDEBUG/MainProcess] finalizer calling ...
2142 [INFO/MainProcess] sending shutdown message to manager
R. David Murray8e8099c2009-04-28 18:02:00 +00002143 [DEBUG/SyncManager-...] manager received shutdown message
2144 [SUBDEBUG/SyncManager-...] calling <Finalize object, callback=unlink, ...
2145 [SUBDEBUG/SyncManager-...] finalizer calling <built-in function unlink> ...
2146 [SUBDEBUG/SyncManager-...] calling <Finalize object, dead>
2147 [SUBDEBUG/SyncManager-...] finalizer calling <function rmtree at 0x5aa730> ...
2148 [INFO/SyncManager-...] manager exiting with exitcode 0
Benjamin Petersone711caf2008-06-11 16:44:04 +00002149
2150The :mod:`multiprocessing.dummy` module
2151~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2152
2153.. module:: multiprocessing.dummy
2154 :synopsis: Dumb wrapper around threading.
2155
2156:mod:`multiprocessing.dummy` replicates the API of :mod:`multiprocessing` but is
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002157no more than a wrapper around the :mod:`threading` module.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002158
2159
2160.. _multiprocessing-programming:
2161
2162Programming guidelines
2163----------------------
2164
2165There are certain guidelines and idioms which should be adhered to when using
2166:mod:`multiprocessing`.
2167
2168
2169All platforms
2170~~~~~~~~~~~~~
2171
2172Avoid shared state
2173
2174 As far as possible one should try to avoid shifting large amounts of data
2175 between processes.
2176
2177 It is probably best to stick to using queues or pipes for communication
2178 between processes rather than using the lower level synchronization
2179 primitives from the :mod:`threading` module.
2180
2181Picklability
2182
2183 Ensure that the arguments to the methods of proxies are picklable.
2184
2185Thread safety of proxies
2186
2187 Do not use a proxy object from more than one thread unless you protect it
2188 with a lock.
2189
2190 (There is never a problem with different processes using the *same* proxy.)
2191
2192Joining zombie processes
2193
2194 On Unix when a process finishes but has not been joined it becomes a zombie.
2195 There should never be very many because each time a new process starts (or
2196 :func:`active_children` is called) all completed processes which have not
2197 yet been joined will be joined. Also calling a finished process's
2198 :meth:`Process.is_alive` will join the process. Even so it is probably good
2199 practice to explicitly join all the processes that you start.
2200
2201Better to inherit than pickle/unpickle
2202
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002203 On Windows many types from :mod:`multiprocessing` need to be picklable so
Benjamin Petersone711caf2008-06-11 16:44:04 +00002204 that child processes can use them. However, one should generally avoid
2205 sending shared objects to other processes using pipes or queues. Instead
Eli Benderskyd08effe2011-12-31 07:20:26 +02002206 you should arrange the program so that a process which needs access to a
Benjamin Petersone711caf2008-06-11 16:44:04 +00002207 shared resource created elsewhere can inherit it from an ancestor process.
2208
2209Avoid terminating processes
2210
2211 Using the :meth:`Process.terminate` method to stop a process is liable to
2212 cause any shared resources (such as locks, semaphores, pipes and queues)
2213 currently being used by the process to become broken or unavailable to other
2214 processes.
2215
2216 Therefore it is probably best to only consider using
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002217 :meth:`Process.terminate` on processes which never use any shared resources.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002218
2219Joining processes that use queues
2220
2221 Bear in mind that a process that has put items in a queue will wait before
2222 terminating until all the buffered items are fed by the "feeder" thread to
2223 the underlying pipe. (The child process can call the
Benjamin Petersonae5360b2008-09-08 23:05:23 +00002224 :meth:`Queue.cancel_join_thread` method of the queue to avoid this behaviour.)
Benjamin Petersone711caf2008-06-11 16:44:04 +00002225
2226 This means that whenever you use a queue you need to make sure that all
2227 items which have been put on the queue will eventually be removed before the
2228 process is joined. Otherwise you cannot be sure that processes which have
2229 put items on the queue will terminate. Remember also that non-daemonic
2230 processes will be automatically be joined.
2231
2232 An example which will deadlock is the following::
2233
2234 from multiprocessing import Process, Queue
2235
2236 def f(q):
2237 q.put('X' * 1000000)
2238
2239 if __name__ == '__main__':
2240 queue = Queue()
2241 p = Process(target=f, args=(queue,))
2242 p.start()
2243 p.join() # this deadlocks
2244 obj = queue.get()
2245
2246 A fix here would be to swap the last two lines round (or simply remove the
2247 ``p.join()`` line).
2248
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002249Explicitly pass resources to child processes
Benjamin Petersone711caf2008-06-11 16:44:04 +00002250
2251 On Unix a child process can make use of a shared resource created in a
2252 parent process using a global resource. However, it is better to pass the
2253 object as an argument to the constructor for the child process.
2254
2255 Apart from making the code (potentially) compatible with Windows this also
2256 ensures that as long as the child process is still alive the object will not
2257 be garbage collected in the parent process. This might be important if some
2258 resource is freed when the object is garbage collected in the parent
2259 process.
2260
2261 So for instance ::
2262
2263 from multiprocessing import Process, Lock
2264
2265 def f():
2266 ... do something using "lock" ...
2267
2268 if __name__ == '__main__':
2269 lock = Lock()
2270 for i in range(10):
2271 Process(target=f).start()
2272
2273 should be rewritten as ::
2274
2275 from multiprocessing import Process, Lock
2276
2277 def f(l):
2278 ... do something using "l" ...
2279
2280 if __name__ == '__main__':
2281 lock = Lock()
2282 for i in range(10):
2283 Process(target=f, args=(lock,)).start()
2284
Eli Benderskyd08effe2011-12-31 07:20:26 +02002285Beware of replacing :data:`sys.stdin` with a "file like object"
Alexandre Vassalottic57a84f2009-07-17 12:07:01 +00002286
2287 :mod:`multiprocessing` originally unconditionally called::
2288
2289 os.close(sys.stdin.fileno())
2290
2291 in the :meth:`multiprocessing.Process._bootstrap` method --- this resulted
2292 in issues with processes-in-processes. This has been changed to::
2293
2294 sys.stdin.close()
2295 sys.stdin = open(os.devnull)
2296
2297 Which solves the fundamental issue of processes colliding with each other
2298 resulting in a bad file descriptor error, but introduces a potential danger
2299 to applications which replace :func:`sys.stdin` with a "file-like object"
2300 with output buffering. This danger is that if multiple processes call
2301 :func:`close()` on this file-like object, it could result in the same
2302 data being flushed to the object multiple times, resulting in corruption.
2303
2304 If you write a file-like object and implement your own caching, you can
2305 make it fork-safe by storing the pid whenever you append to the cache,
2306 and discarding the cache when the pid changes. For example::
2307
2308 @property
2309 def cache(self):
2310 pid = os.getpid()
2311 if pid != self._pid:
2312 self._pid = pid
2313 self._cache = []
2314 return self._cache
2315
2316 For more information, see :issue:`5155`, :issue:`5313` and :issue:`5331`
Benjamin Petersone711caf2008-06-11 16:44:04 +00002317
2318Windows
2319~~~~~~~
2320
2321Since Windows lacks :func:`os.fork` it has a few extra restrictions:
2322
2323More picklability
2324
2325 Ensure that all arguments to :meth:`Process.__init__` are picklable. This
2326 means, in particular, that bound or unbound methods cannot be used directly
2327 as the ``target`` argument on Windows --- just define a function and use
2328 that instead.
2329
2330 Also, if you subclass :class:`Process` then make sure that instances will be
2331 picklable when the :meth:`Process.start` method is called.
2332
2333Global variables
2334
2335 Bear in mind that if code run in a child process tries to access a global
2336 variable, then the value it sees (if any) may not be the same as the value
2337 in the parent process at the time that :meth:`Process.start` was called.
2338
2339 However, global variables which are just module level constants cause no
2340 problems.
2341
2342Safe importing of main module
2343
2344 Make sure that the main module can be safely imported by a new Python
2345 interpreter without causing unintended side effects (such a starting a new
2346 process).
2347
2348 For example, under Windows running the following module would fail with a
2349 :exc:`RuntimeError`::
2350
2351 from multiprocessing import Process
2352
2353 def foo():
Georg Brandl49702152008-09-29 06:43:45 +00002354 print('hello')
Benjamin Petersone711caf2008-06-11 16:44:04 +00002355
2356 p = Process(target=foo)
2357 p.start()
2358
2359 Instead one should protect the "entry point" of the program by using ``if
2360 __name__ == '__main__':`` as follows::
2361
2362 from multiprocessing import Process, freeze_support
2363
2364 def foo():
Georg Brandl49702152008-09-29 06:43:45 +00002365 print('hello')
Benjamin Petersone711caf2008-06-11 16:44:04 +00002366
2367 if __name__ == '__main__':
2368 freeze_support()
2369 p = Process(target=foo)
2370 p.start()
2371
Benjamin Peterson5289b2b2008-06-28 00:40:54 +00002372 (The ``freeze_support()`` line can be omitted if the program will be run
Benjamin Petersone711caf2008-06-11 16:44:04 +00002373 normally instead of frozen.)
2374
2375 This allows the newly spawned Python interpreter to safely import the module
2376 and then run the module's ``foo()`` function.
2377
2378 Similar restrictions apply if a pool or manager is created in the main
2379 module.
2380
2381
2382.. _multiprocessing-examples:
2383
2384Examples
2385--------
2386
2387Demonstration of how to create and use customized managers and proxies:
2388
2389.. literalinclude:: ../includes/mp_newtype.py
Ezio Melottif86b28e2012-04-13 20:50:48 -06002390 :language: python3
Benjamin Petersone711caf2008-06-11 16:44:04 +00002391
2392
2393Using :class:`Pool`:
2394
2395.. literalinclude:: ../includes/mp_pool.py
Ezio Melottif86b28e2012-04-13 20:50:48 -06002396 :language: python3
Benjamin Petersone711caf2008-06-11 16:44:04 +00002397
2398
2399Synchronization types like locks, conditions and queues:
2400
2401.. literalinclude:: ../includes/mp_synchronize.py
Ezio Melottif86b28e2012-04-13 20:50:48 -06002402 :language: python3
Benjamin Petersone711caf2008-06-11 16:44:04 +00002403
2404
Georg Brandl0b37b332010-09-03 22:49:27 +00002405An example showing how to use queues to feed tasks to a collection of worker
Eli Benderskyd08effe2011-12-31 07:20:26 +02002406processes and collect the results:
Benjamin Petersone711caf2008-06-11 16:44:04 +00002407
2408.. literalinclude:: ../includes/mp_workers.py
2409
2410
2411An example of how a pool of worker processes can each run a
Georg Brandl47d48bb2010-07-10 11:51:06 +00002412:class:`~http.server.SimpleHTTPRequestHandler` instance while sharing a single
2413listening socket.
Benjamin Petersone711caf2008-06-11 16:44:04 +00002414
2415.. literalinclude:: ../includes/mp_webserver.py
2416
2417
2418Some simple benchmarks comparing :mod:`multiprocessing` with :mod:`threading`:
2419
2420.. literalinclude:: ../includes/mp_benchmarks.py
2421