blob: 18e81a793858b01b2a14ed04d27ae6818cc9f685 [file] [log] [blame]
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001:mod:`multiprocessing` --- Process-based "threading" interface
2==============================================================
3
4.. module:: multiprocessing
5 :synopsis: Process-based "threading" interface.
6
7.. versionadded:: 2.6
8
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00009
Benjamin Peterson190d56e2008-06-11 02:40:25 +000010Introduction
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +000011----------------------
Benjamin Peterson190d56e2008-06-11 02:40:25 +000012
Benjamin Peterson910c2ab2008-06-27 23:22:06 +000013:mod:`multiprocessing` is a package that supports spawning processes using an
14API similar to the :mod:`threading` module. The :mod:`multiprocessing` package
15offers both local and remote concurrency, effectively side-stepping the
16:term:`Global Interpreter Lock` by using subprocesses instead of threads. Due
17to this, the :mod:`multiprocessing` module allows the programmer to fully
18leverage multiple processors on a given machine. It runs on both Unix and
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +000019Windows.
Benjamin Peterson190d56e2008-06-11 02:40:25 +000020
Jesse Noller37040cd2008-09-30 00:15:45 +000021.. warning::
22
Andrew M. Kuchling83b39102008-09-30 12:31:07 +000023 Some of this package's functionality requires a functioning shared semaphore
Georg Brandlc62ef8b2009-01-03 20:55:06 +000024 implementation on the host operating system. Without one, the
25 :mod:`multiprocessing.synchronize` module will be disabled, and attempts to
26 import it will result in an :exc:`ImportError`. See
Andrew M. Kuchling83b39102008-09-30 12:31:07 +000027 :issue:`3770` for additional information.
Benjamin Peterson910c2ab2008-06-27 23:22:06 +000028
Jesse Nollera280fd72008-11-28 18:22:54 +000029.. note::
30
31 Functionality within this package requires that the ``__main__`` method be
32 importable by the children. This is covered in :ref:`multiprocessing-programming`
33 however it is worth pointing out here. This means that some examples, such
34 as the :class:`multiprocessing.Pool` examples will not work in the
35 interactive interpreter. For example::
36
37 >>> from multiprocessing import Pool
38 >>> p = Pool(5)
39 >>> def f(x):
Georg Brandl7044b112009-01-03 21:04:55 +000040 ... return x*x
Georg Brandlc62ef8b2009-01-03 20:55:06 +000041 ...
Jesse Nollera280fd72008-11-28 18:22:54 +000042 >>> p.map(f, [1,2,3])
43 Process PoolWorker-1:
44 Process PoolWorker-2:
45 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
51
Benjamin Peterson190d56e2008-06-11 02:40:25 +000052The :class:`Process` class
53~~~~~~~~~~~~~~~~~~~~~~~~~~
54
55In :mod:`multiprocessing`, processes are spawned by creating a :class:`Process`
Benjamin Peterson910c2ab2008-06-27 23:22:06 +000056object and then calling its :meth:`~Process.start` method. :class:`Process`
Benjamin Peterson190d56e2008-06-11 02:40:25 +000057follows the API of :class:`threading.Thread`. A trivial example of a
58multiprocess program is ::
59
Jesse Nollera280fd72008-11-28 18:22:54 +000060 from multiprocessing import Process
Benjamin Peterson190d56e2008-06-11 02:40:25 +000061
Jesse Nollera280fd72008-11-28 18:22:54 +000062 def f(name):
63 print 'hello', name
Benjamin Peterson190d56e2008-06-11 02:40:25 +000064
Jesse Nollera280fd72008-11-28 18:22:54 +000065 if __name__ == '__main__':
66 p = Process(target=f, args=('bob',))
67 p.start()
68 p.join()
Benjamin Peterson190d56e2008-06-11 02:40:25 +000069
Jesse Nollera280fd72008-11-28 18:22:54 +000070To show the individual process IDs involved, here is an expanded example::
71
72 from multiprocessing import Process
73 import os
74
75 def info(title):
76 print title
77 print 'module name:', __name__
78 print 'parent process:', os.getppid()
79 print 'process id:', os.getpid()
Georg Brandlc62ef8b2009-01-03 20:55:06 +000080
Jesse Nollera280fd72008-11-28 18:22:54 +000081 def f(name):
82 info('function f')
83 print 'hello', name
Georg Brandlc62ef8b2009-01-03 20:55:06 +000084
Jesse Nollera280fd72008-11-28 18:22:54 +000085 if __name__ == '__main__':
86 info('main line')
87 p = Process(target=f, args=('bob',))
88 p.start()
89 p.join()
Benjamin Peterson190d56e2008-06-11 02:40:25 +000090
91For an explanation of why (on Windows) the ``if __name__ == '__main__'`` part is
92necessary, see :ref:`multiprocessing-programming`.
93
94
95
96Exchanging objects between processes
97~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
98
99:mod:`multiprocessing` supports two types of communication channel between
100processes:
101
102**Queues**
103
104 The :class:`Queue` class is a near clone of :class:`Queue.Queue`. For
105 example::
106
107 from multiprocessing import Process, Queue
108
109 def f(q):
110 q.put([42, None, 'hello'])
111
Georg Brandledd7d952009-01-03 14:29:53 +0000112 if __name__ == '__main__':
113 q = Queue()
114 p = Process(target=f, args=(q,))
115 p.start()
116 print q.get() # prints "[42, None, 'hello']"
117 p.join()
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000118
119 Queues are thread and process safe.
120
121**Pipes**
122
123 The :func:`Pipe` function returns a pair of connection objects connected by a
124 pipe which by default is duplex (two-way). For example::
125
126 from multiprocessing import Process, Pipe
127
128 def f(conn):
129 conn.send([42, None, 'hello'])
130 conn.close()
131
132 if __name__ == '__main__':
133 parent_conn, child_conn = Pipe()
134 p = Process(target=f, args=(child_conn,))
135 p.start()
136 print parent_conn.recv() # prints "[42, None, 'hello']"
137 p.join()
138
139 The two connection objects returned by :func:`Pipe` represent the two ends of
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000140 the pipe. Each connection object has :meth:`~Connection.send` and
141 :meth:`~Connection.recv` methods (among others). Note that data in a pipe
142 may become corrupted if two processes (or threads) try to read from or write
143 to the *same* end of the pipe at the same time. Of course there is no risk
144 of corruption from processes using different ends of the pipe at the same
145 time.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000146
147
148Synchronization between processes
149~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
150
151:mod:`multiprocessing` contains equivalents of all the synchronization
152primitives from :mod:`threading`. For instance one can use a lock to ensure
153that only one process prints to standard output at a time::
154
155 from multiprocessing import Process, Lock
156
157 def f(l, i):
158 l.acquire()
159 print 'hello world', i
160 l.release()
161
162 if __name__ == '__main__':
163 lock = Lock()
164
165 for num in range(10):
166 Process(target=f, args=(lock, num)).start()
167
168Without using the lock output from the different processes is liable to get all
169mixed up.
170
171
172Sharing state between processes
173~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
174
175As mentioned above, when doing concurrent programming it is usually best to
176avoid using shared state as far as possible. This is particularly true when
177using multiple processes.
178
179However, if you really do need to use some shared data then
180:mod:`multiprocessing` provides a couple of ways of doing so.
181
182**Shared memory**
183
184 Data can be stored in a shared memory map using :class:`Value` or
185 :class:`Array`. For example, the following code ::
186
187 from multiprocessing import Process, Value, Array
188
189 def f(n, a):
190 n.value = 3.1415927
191 for i in range(len(a)):
192 a[i] = -a[i]
193
194 if __name__ == '__main__':
195 num = Value('d', 0.0)
196 arr = Array('i', range(10))
197
198 p = Process(target=f, args=(num, arr))
199 p.start()
200 p.join()
201
202 print num.value
203 print arr[:]
204
205 will print ::
206
207 3.1415927
208 [0, -1, -2, -3, -4, -5, -6, -7, -8, -9]
209
210 The ``'d'`` and ``'i'`` arguments used when creating ``num`` and ``arr`` are
211 typecodes of the kind used by the :mod:`array` module: ``'d'`` indicates a
Benjamin Peterson90f36732008-07-12 20:16:19 +0000212 double precision float and ``'i'`` indicates a signed integer. These shared
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000213 objects will be process and thread safe.
214
215 For more flexibility in using shared memory one can use the
216 :mod:`multiprocessing.sharedctypes` module which supports the creation of
217 arbitrary ctypes objects allocated from shared memory.
218
219**Server process**
220
221 A manager object returned by :func:`Manager` controls a server process which
Andrew M. Kuchlingded01d12008-07-14 00:35:32 +0000222 holds Python objects and allows other processes to manipulate them using
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000223 proxies.
224
225 A manager returned by :func:`Manager` will support types :class:`list`,
226 :class:`dict`, :class:`Namespace`, :class:`Lock`, :class:`RLock`,
227 :class:`Semaphore`, :class:`BoundedSemaphore`, :class:`Condition`,
228 :class:`Event`, :class:`Queue`, :class:`Value` and :class:`Array`. For
229 example, ::
230
231 from multiprocessing import Process, Manager
232
233 def f(d, l):
234 d[1] = '1'
235 d['2'] = 2
236 d[0.25] = None
237 l.reverse()
238
239 if __name__ == '__main__':
240 manager = Manager()
241
242 d = manager.dict()
243 l = manager.list(range(10))
244
245 p = Process(target=f, args=(d, l))
246 p.start()
247 p.join()
248
249 print d
250 print l
251
252 will print ::
253
254 {0.25: None, 1: '1', '2': 2}
255 [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
256
257 Server process managers are more flexible than using shared memory objects
258 because they can be made to support arbitrary object types. Also, a single
259 manager can be shared by processes on different computers over a network.
260 They are, however, slower than using shared memory.
261
262
263Using a pool of workers
264~~~~~~~~~~~~~~~~~~~~~~~
265
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000266The :class:`~multiprocessing.pool.Pool` class represents a pool of worker
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000267processes. It has methods which allows tasks to be offloaded to the worker
268processes in a few different ways.
269
270For example::
271
272 from multiprocessing import Pool
273
274 def f(x):
275 return x*x
276
277 if __name__ == '__main__':
278 pool = Pool(processes=4) # start 4 worker processes
Jesse Nollera280fd72008-11-28 18:22:54 +0000279 result = pool.apply_async(f, [10]) # evaluate "f(10)" asynchronously
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000280 print result.get(timeout=1) # prints "100" unless your computer is *very* slow
281 print pool.map(f, range(10)) # prints "[0, 1, 4,..., 81]"
282
283
284Reference
285---------
286
287The :mod:`multiprocessing` package mostly replicates the API of the
288:mod:`threading` module.
289
290
291:class:`Process` and exceptions
292~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
293
294.. class:: Process([group[, target[, name[, args[, kwargs]]]]])
295
296 Process objects represent activity that is run in a separate process. The
297 :class:`Process` class has equivalents of all the methods of
298 :class:`threading.Thread`.
299
300 The constructor should always be called with keyword arguments. *group*
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000301 should always be ``None``; it exists solely for compatibility with
Benjamin Peterson73641d72008-08-20 14:07:59 +0000302 :class:`threading.Thread`. *target* is the callable object to be invoked by
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000303 the :meth:`run()` method. It defaults to ``None``, meaning nothing is
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000304 called. *name* is the process name. By default, a unique name is constructed
305 of the form 'Process-N\ :sub:`1`:N\ :sub:`2`:...:N\ :sub:`k`' where N\
306 :sub:`1`,N\ :sub:`2`,...,N\ :sub:`k` is a sequence of integers whose length
307 is determined by the *generation* of the process. *args* is the argument
308 tuple for the target invocation. *kwargs* is a dictionary of keyword
309 arguments for the target invocation. By default, no arguments are passed to
310 *target*.
311
312 If a subclass overrides the constructor, it must make sure it invokes the
313 base class constructor (:meth:`Process.__init__`) before doing anything else
314 to the process.
315
316 .. method:: run()
317
318 Method representing the process's activity.
319
320 You may override this method in a subclass. The standard :meth:`run`
321 method invokes the callable object passed to the object's constructor as
322 the target argument, if any, with sequential and keyword arguments taken
323 from the *args* and *kwargs* arguments, respectively.
324
325 .. method:: start()
326
327 Start the process's activity.
328
329 This must be called at most once per process object. It arranges for the
330 object's :meth:`run` method to be invoked in a separate process.
331
332 .. method:: join([timeout])
333
334 Block the calling thread until the process whose :meth:`join` method is
335 called terminates or until the optional timeout occurs.
336
337 If *timeout* is ``None`` then there is no timeout.
338
339 A process can be joined many times.
340
341 A process cannot join itself because this would cause a deadlock. It is
342 an error to attempt to join a process before it has been started.
343
Benjamin Peterson73641d72008-08-20 14:07:59 +0000344 .. attribute:: name
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000345
Benjamin Peterson73641d72008-08-20 14:07:59 +0000346 The process's name.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000347
348 The name is a string used for identification purposes only. It has no
349 semantics. Multiple processes may be given the same name. The initial
350 name is set by the constructor.
351
Jesse Nollera280fd72008-11-28 18:22:54 +0000352 .. method:: is_alive
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000353
354 Return whether the process is alive.
355
356 Roughly, a process object is alive from the moment the :meth:`start`
357 method returns until the child process terminates.
358
Benjamin Peterson73641d72008-08-20 14:07:59 +0000359 .. attribute:: daemon
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000360
Georg Brandl3bcb0ce2008-12-30 10:15:49 +0000361 The process's daemon flag, a Boolean value. This must be set before
Benjamin Peterson73641d72008-08-20 14:07:59 +0000362 :meth:`start` is called.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000363
364 The initial value is inherited from the creating process.
365
366 When a process exits, it attempts to terminate all of its daemonic child
367 processes.
368
369 Note that a daemonic process is not allowed to create child processes.
370 Otherwise a daemonic process would leave its children orphaned if it gets
371 terminated when its parent process exits.
372
Brett Cannon971f1022008-08-24 23:15:19 +0000373 In addition to the :class:`Threading.Thread` API, :class:`Process` objects
374 also support the following attributes and methods:
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000375
Benjamin Peterson73641d72008-08-20 14:07:59 +0000376 .. attribute:: pid
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000377
378 Return the process ID. Before the process is spawned, this will be
379 ``None``.
380
Benjamin Peterson73641d72008-08-20 14:07:59 +0000381 .. attribute:: exitcode
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000382
Benjamin Peterson73641d72008-08-20 14:07:59 +0000383 The child's exit code. This will be ``None`` if the process has not yet
384 terminated. A negative value *-N* indicates that the child was terminated
385 by signal *N*.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000386
Benjamin Peterson73641d72008-08-20 14:07:59 +0000387 .. attribute:: authkey
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000388
Benjamin Peterson73641d72008-08-20 14:07:59 +0000389 The process's authentication key (a byte string).
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000390
391 When :mod:`multiprocessing` is initialized the main process is assigned a
392 random string using :func:`os.random`.
393
394 When a :class:`Process` object is created, it will inherit the
Benjamin Peterson73641d72008-08-20 14:07:59 +0000395 authentication key of its parent process, although this may be changed by
396 setting :attr:`authkey` to another byte string.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000397
398 See :ref:`multiprocessing-auth-keys`.
399
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000400 .. method:: terminate()
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000401
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000402 Terminate the process. On Unix this is done using the ``SIGTERM`` signal;
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000403 on Windows :cfunc:`TerminateProcess` is used. Note that exit handlers and
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000404 finally clauses, etc., will not be executed.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000405
406 Note that descendant processes of the process will *not* be terminated --
407 they will simply become orphaned.
408
409 .. warning::
410
411 If this method is used when the associated process is using a pipe or
412 queue then the pipe or queue is liable to become corrupted and may
413 become unusable by other process. Similarly, if the process has
414 acquired a lock or semaphore etc. then terminating it is liable to
415 cause other processes to deadlock.
416
417 Note that the :meth:`start`, :meth:`join`, :meth:`is_alive` and
Benjamin Peterson73641d72008-08-20 14:07:59 +0000418 :attr:`exit_code` methods should only be called by the process that created
419 the process object.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000420
421 Example usage of some of the methods of :class:`Process`::
422
Georg Brandl19cc9442008-10-16 21:36:39 +0000423 >>> import multiprocessing, time, signal
424 >>> p = multiprocessing.Process(target=time.sleep, args=(1000,))
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000425 >>> print p, p.is_alive()
426 <Process(Process-1, initial)> False
427 >>> p.start()
428 >>> print p, p.is_alive()
429 <Process(Process-1, started)> True
430 >>> p.terminate()
431 >>> print p, p.is_alive()
432 <Process(Process-1, stopped[SIGTERM])> False
Benjamin Peterson73641d72008-08-20 14:07:59 +0000433 >>> p.exitcode == -signal.SIGTERM
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000434 True
435
436
437.. exception:: BufferTooShort
438
439 Exception raised by :meth:`Connection.recv_bytes_into()` when the supplied
440 buffer object is too small for the message read.
441
442 If ``e`` is an instance of :exc:`BufferTooShort` then ``e.args[0]`` will give
443 the message as a byte string.
444
445
446Pipes and Queues
447~~~~~~~~~~~~~~~~
448
449When using multiple processes, one generally uses message passing for
450communication between processes and avoids having to use any synchronization
451primitives like locks.
452
453For passing messages one can use :func:`Pipe` (for a connection between two
454processes) or a queue (which allows multiple producers and consumers).
455
456The :class:`Queue` and :class:`JoinableQueue` types are multi-producer,
457multi-consumer FIFO queues modelled on the :class:`Queue.Queue` class in the
458standard library. They differ in that :class:`Queue` lacks the
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000459:meth:`~Queue.Queue.task_done` and :meth:`~Queue.Queue.join` methods introduced
460into Python 2.5's :class:`Queue.Queue` class.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000461
462If you use :class:`JoinableQueue` then you **must** call
463:meth:`JoinableQueue.task_done` for each task removed from the queue or else the
464semaphore used to count the number of unfinished tasks may eventually overflow
465raising an exception.
466
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000467Note that one can also create a shared queue by using a manager object -- see
468:ref:`multiprocessing-managers`.
469
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000470.. note::
471
472 :mod:`multiprocessing` uses the usual :exc:`Queue.Empty` and
473 :exc:`Queue.Full` exceptions to signal a timeout. They are not available in
474 the :mod:`multiprocessing` namespace so you need to import them from
475 :mod:`Queue`.
476
477
478.. warning::
479
480 If a process is killed using :meth:`Process.terminate` or :func:`os.kill`
481 while it is trying to use a :class:`Queue`, then the data in the queue is
482 likely to become corrupted. This may cause any other processes to get an
483 exception when it tries to use the queue later on.
484
485.. warning::
486
487 As mentioned above, if a child process has put items on a queue (and it has
488 not used :meth:`JoinableQueue.cancel_join_thread`), then that process will
489 not terminate until all buffered items have been flushed to the pipe.
490
491 This means that if you try joining that process you may get a deadlock unless
492 you are sure that all items which have been put on the queue have been
493 consumed. Similarly, if the child process is non-daemonic then the parent
Andrew M. Kuchlingded01d12008-07-14 00:35:32 +0000494 process may hang on exit when it tries to join all its non-daemonic children.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000495
496 Note that a queue created using a manager does not have this issue. See
497 :ref:`multiprocessing-programming`.
498
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000499For an example of the usage of queues for interprocess communication see
500:ref:`multiprocessing-examples`.
501
502
503.. function:: Pipe([duplex])
504
505 Returns a pair ``(conn1, conn2)`` of :class:`Connection` objects representing
506 the ends of a pipe.
507
508 If *duplex* is ``True`` (the default) then the pipe is bidirectional. If
509 *duplex* is ``False`` then the pipe is unidirectional: ``conn1`` can only be
510 used for receiving messages and ``conn2`` can only be used for sending
511 messages.
512
513
514.. class:: Queue([maxsize])
515
516 Returns a process shared queue implemented using a pipe and a few
517 locks/semaphores. When a process first puts an item on the queue a feeder
518 thread is started which transfers objects from a buffer into the pipe.
519
520 The usual :exc:`Queue.Empty` and :exc:`Queue.Full` exceptions from the
521 standard library's :mod:`Queue` module are raised to signal timeouts.
522
523 :class:`Queue` implements all the methods of :class:`Queue.Queue` except for
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000524 :meth:`~Queue.Queue.task_done` and :meth:`~Queue.Queue.join`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000525
526 .. method:: qsize()
527
528 Return the approximate size of the queue. Because of
529 multithreading/multiprocessing semantics, this number is not reliable.
530
531 Note that this may raise :exc:`NotImplementedError` on Unix platforms like
Georg Brandl9af94982008-09-13 17:41:16 +0000532 Mac OS X where ``sem_getvalue()`` is not implemented.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000533
534 .. method:: empty()
535
536 Return ``True`` if the queue is empty, ``False`` otherwise. Because of
537 multithreading/multiprocessing semantics, this is not reliable.
538
539 .. method:: full()
540
541 Return ``True`` if the queue is full, ``False`` otherwise. Because of
542 multithreading/multiprocessing semantics, this is not reliable.
543
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000544 .. method:: put(item[, block[, timeout]])
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000545
Georg Brandlc62ef8b2009-01-03 20:55:06 +0000546 Put item into the queue. If the optional argument *block* is ``True``
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +0000547 (the default) and *timeout* is ``None`` (the default), block if necessary until
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000548 a free slot is available. If *timeout* is a positive number, it blocks at
549 most *timeout* seconds and raises the :exc:`Queue.Full` exception if no
550 free slot was available within that time. Otherwise (*block* is
551 ``False``), put an item on the queue if a free slot is immediately
552 available, else raise the :exc:`Queue.Full` exception (*timeout* is
553 ignored in that case).
554
555 .. method:: put_nowait(item)
556
557 Equivalent to ``put(item, False)``.
558
559 .. method:: get([block[, timeout]])
560
561 Remove and return an item from the queue. If optional args *block* is
562 ``True`` (the default) and *timeout* is ``None`` (the default), block if
563 necessary until an item is available. If *timeout* is a positive number,
564 it blocks at most *timeout* seconds and raises the :exc:`Queue.Empty`
565 exception if no item was available within that time. Otherwise (block is
566 ``False``), return an item if one is immediately available, else raise the
567 :exc:`Queue.Empty` exception (*timeout* is ignored in that case).
568
569 .. method:: get_nowait()
570 get_no_wait()
571
572 Equivalent to ``get(False)``.
573
574 :class:`multiprocessing.Queue` has a few additional methods not found in
Andrew M. Kuchlingded01d12008-07-14 00:35:32 +0000575 :class:`Queue.Queue`. These methods are usually unnecessary for most
576 code:
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000577
578 .. method:: close()
579
580 Indicate that no more data will be put on this queue by the current
581 process. The background thread will quit once it has flushed all buffered
582 data to the pipe. This is called automatically when the queue is garbage
583 collected.
584
585 .. method:: join_thread()
586
587 Join the background thread. This can only be used after :meth:`close` has
588 been called. It blocks until the background thread exits, ensuring that
589 all data in the buffer has been flushed to the pipe.
590
591 By default if a process is not the creator of the queue then on exit it
592 will attempt to join the queue's background thread. The process can call
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000593 :meth:`cancel_join_thread` to make :meth:`join_thread` do nothing.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000594
595 .. method:: cancel_join_thread()
596
597 Prevent :meth:`join_thread` from blocking. In particular, this prevents
598 the background thread from being joined automatically when the process
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000599 exits -- see :meth:`join_thread`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000600
601
602.. class:: JoinableQueue([maxsize])
603
604 :class:`JoinableQueue`, a :class:`Queue` subclass, is a queue which
605 additionally has :meth:`task_done` and :meth:`join` methods.
606
607 .. method:: task_done()
608
609 Indicate that a formerly enqueued task is complete. Used by queue consumer
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000610 threads. For each :meth:`~Queue.get` used to fetch a task, a subsequent
611 call to :meth:`task_done` tells the queue that the processing on the task
612 is complete.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000613
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000614 If a :meth:`~Queue.join` is currently blocking, it will resume when all
615 items have been processed (meaning that a :meth:`task_done` call was
616 received for every item that had been :meth:`~Queue.put` into the queue).
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000617
618 Raises a :exc:`ValueError` if called more times than there were items
619 placed in the queue.
620
621
622 .. method:: join()
623
624 Block until all items in the queue have been gotten and processed.
625
626 The count of unfinished tasks goes up whenever an item is added to the
627 queue. The count goes down whenever a consumer thread calls
628 :meth:`task_done` to indicate that the item was retrieved and all work on
629 it is complete. When the count of unfinished tasks drops to zero,
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000630 :meth:`~Queue.join` unblocks.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000631
632
633Miscellaneous
634~~~~~~~~~~~~~
635
636.. function:: active_children()
637
638 Return list of all live children of the current process.
639
640 Calling this has the side affect of "joining" any processes which have
641 already finished.
642
643.. function:: cpu_count()
644
645 Return the number of CPUs in the system. May raise
646 :exc:`NotImplementedError`.
647
648.. function:: current_process()
649
650 Return the :class:`Process` object corresponding to the current process.
651
652 An analogue of :func:`threading.current_thread`.
653
654.. function:: freeze_support()
655
656 Add support for when a program which uses :mod:`multiprocessing` has been
657 frozen to produce a Windows executable. (Has been tested with **py2exe**,
658 **PyInstaller** and **cx_Freeze**.)
659
660 One needs to call this function straight after the ``if __name__ ==
661 '__main__'`` line of the main module. For example::
662
663 from multiprocessing import Process, freeze_support
664
665 def f():
666 print 'hello world!'
667
668 if __name__ == '__main__':
669 freeze_support()
670 Process(target=f).start()
671
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000672 If the ``freeze_support()`` line is missed out then trying to run the frozen
673 executable will raise :exc:`RuntimeError`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000674
675 If the module is being run normally by the Python interpreter then
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000676 :func:`freeze_support` has no effect.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000677
678.. function:: set_executable()
679
680 Sets the path of the python interpreter to use when starting a child process.
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000681 (By default :data:`sys.executable` is used). Embedders will probably need to
682 do some thing like ::
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000683
684 setExecutable(os.path.join(sys.exec_prefix, 'pythonw.exe'))
685
686 before they can create child processes. (Windows only)
687
688
689.. note::
690
691 :mod:`multiprocessing` contains no analogues of
692 :func:`threading.active_count`, :func:`threading.enumerate`,
693 :func:`threading.settrace`, :func:`threading.setprofile`,
694 :class:`threading.Timer`, or :class:`threading.local`.
695
696
697Connection Objects
698~~~~~~~~~~~~~~~~~~
699
700Connection objects allow the sending and receiving of picklable objects or
701strings. They can be thought of as message oriented connected sockets.
702
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000703Connection objects usually created using :func:`Pipe` -- see also
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000704:ref:`multiprocessing-listeners-clients`.
705
706.. class:: Connection
707
708 .. method:: send(obj)
709
710 Send an object to the other end of the connection which should be read
711 using :meth:`recv`.
712
Jesse Noller5053fbb2009-04-02 04:22:09 +0000713 The object must be picklable. Very large pickles (approximately 32 MB+,
714 though it depends on the OS) may raise a ValueError exception.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000715
716 .. method:: recv()
717
718 Return an object sent from the other end of the connection using
719 :meth:`send`. Raises :exc:`EOFError` if there is nothing left to receive
720 and the other end was closed.
721
722 .. method:: fileno()
723
724 Returns the file descriptor or handle used by the connection.
725
726 .. method:: close()
727
728 Close the connection.
729
730 This is called automatically when the connection is garbage collected.
731
732 .. method:: poll([timeout])
733
734 Return whether there is any data available to be read.
735
736 If *timeout* is not specified then it will return immediately. If
737 *timeout* is a number then this specifies the maximum time in seconds to
738 block. If *timeout* is ``None`` then an infinite timeout is used.
739
740 .. method:: send_bytes(buffer[, offset[, size]])
741
742 Send byte data from an object supporting the buffer interface as a
743 complete message.
744
745 If *offset* is given then data is read from that position in *buffer*. If
Jesse Noller5053fbb2009-04-02 04:22:09 +0000746 *size* is given then that many bytes will be read from buffer. Very large
747 buffers (approximately 32 MB+, though it depends on the OS) may raise a
748 ValueError exception
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000749
750 .. method:: recv_bytes([maxlength])
751
752 Return a complete message of byte data sent from the other end of the
753 connection as a string. Raises :exc:`EOFError` if there is nothing left
754 to receive and the other end has closed.
755
756 If *maxlength* is specified and the message is longer than *maxlength*
757 then :exc:`IOError` is raised and the connection will no longer be
758 readable.
759
760 .. method:: recv_bytes_into(buffer[, offset])
761
762 Read into *buffer* a complete message of byte data sent from the other end
763 of the connection and return the number of bytes in the message. Raises
764 :exc:`EOFError` if there is nothing left to receive and the other end was
765 closed.
766
767 *buffer* must be an object satisfying the writable buffer interface. If
768 *offset* is given then the message will be written into the buffer from
769 *that position. Offset must be a non-negative integer less than the
770 *length of *buffer* (in bytes).
771
772 If the buffer is too short then a :exc:`BufferTooShort` exception is
773 raised and the complete message is available as ``e.args[0]`` where ``e``
774 is the exception instance.
775
776
777For example:
778
779 >>> from multiprocessing import Pipe
780 >>> a, b = Pipe()
781 >>> a.send([1, 'hello', None])
782 >>> b.recv()
783 [1, 'hello', None]
784 >>> b.send_bytes('thank you')
785 >>> a.recv_bytes()
786 'thank you'
787 >>> import array
788 >>> arr1 = array.array('i', range(5))
789 >>> arr2 = array.array('i', [0] * 10)
790 >>> a.send_bytes(arr1)
791 >>> count = b.recv_bytes_into(arr2)
792 >>> assert count == len(arr1) * arr1.itemsize
793 >>> arr2
794 array('i', [0, 1, 2, 3, 4, 0, 0, 0, 0, 0])
795
796
797.. warning::
798
799 The :meth:`Connection.recv` method automatically unpickles the data it
800 receives, which can be a security risk unless you can trust the process
801 which sent the message.
802
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000803 Therefore, unless the connection object was produced using :func:`Pipe` you
804 should only use the :meth:`~Connection.recv` and :meth:`~Connection.send`
805 methods after performing some sort of authentication. See
806 :ref:`multiprocessing-auth-keys`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000807
808.. warning::
809
810 If a process is killed while it is trying to read or write to a pipe then
811 the data in the pipe is likely to become corrupted, because it may become
812 impossible to be sure where the message boundaries lie.
813
814
815Synchronization primitives
816~~~~~~~~~~~~~~~~~~~~~~~~~~
817
818Generally synchronization primitives are not as necessary in a multiprocess
Andrew M. Kuchling8ea605c2008-07-14 01:18:16 +0000819program as they are in a multithreaded program. See the documentation for
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000820:mod:`threading` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000821
822Note that one can also create synchronization primitives by using a manager
823object -- see :ref:`multiprocessing-managers`.
824
825.. class:: BoundedSemaphore([value])
826
827 A bounded semaphore object: a clone of :class:`threading.BoundedSemaphore`.
828
Georg Brandl9af94982008-09-13 17:41:16 +0000829 (On Mac OS X this is indistinguishable from :class:`Semaphore` because
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000830 ``sem_getvalue()`` is not implemented on that platform).
831
832.. class:: Condition([lock])
833
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000834 A condition variable: a clone of :class:`threading.Condition`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000835
836 If *lock* is specified then it should be a :class:`Lock` or :class:`RLock`
837 object from :mod:`multiprocessing`.
838
839.. class:: Event()
840
841 A clone of :class:`threading.Event`.
Jesse Noller02cb0eb2009-04-01 03:45:50 +0000842 This method returns the state of the internal semaphore on exit, so it
843 will always return ``True`` except if a timeout is given and the operation
844 times out.
845
846 .. versionchanged:: 2.7
847 Previously, the method always returned ``None``.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000848
849.. class:: Lock()
850
851 A non-recursive lock object: a clone of :class:`threading.Lock`.
852
853.. class:: RLock()
854
855 A recursive lock object: a clone of :class:`threading.RLock`.
856
857.. class:: Semaphore([value])
858
859 A bounded semaphore object: a clone of :class:`threading.Semaphore`.
860
861.. note::
862
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000863 The :meth:`acquire` method of :class:`BoundedSemaphore`, :class:`Lock`,
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000864 :class:`RLock` and :class:`Semaphore` has a timeout parameter not supported
865 by the equivalents in :mod:`threading`. The signature is
866 ``acquire(block=True, timeout=None)`` with keyword parameters being
867 acceptable. If *block* is ``True`` and *timeout* is not ``None`` then it
868 specifies a timeout in seconds. If *block* is ``False`` then *timeout* is
869 ignored.
Georg Brandlc62ef8b2009-01-03 20:55:06 +0000870
Jesse Nollera280fd72008-11-28 18:22:54 +0000871 Note that on OS/X ``sem_timedwait`` is unsupported, so timeout arguments
872 for these will be ignored.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000873
874.. note::
875
876 If the SIGINT signal generated by Ctrl-C arrives while the main thread is
877 blocked by a call to :meth:`BoundedSemaphore.acquire`, :meth:`Lock.acquire`,
878 :meth:`RLock.acquire`, :meth:`Semaphore.acquire`, :meth:`Condition.acquire`
879 or :meth:`Condition.wait` then the call will be immediately interrupted and
880 :exc:`KeyboardInterrupt` will be raised.
881
882 This differs from the behaviour of :mod:`threading` where SIGINT will be
883 ignored while the equivalent blocking calls are in progress.
884
885
886Shared :mod:`ctypes` Objects
887~~~~~~~~~~~~~~~~~~~~~~~~~~~~
888
889It is possible to create shared objects using shared memory which can be
890inherited by child processes.
891
Jesse Noller6ab22152009-01-18 02:45:38 +0000892.. function:: Value(typecode_or_type, *args[, lock])
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000893
894 Return a :mod:`ctypes` object allocated from shared memory. By default the
895 return value is actually a synchronized wrapper for the object.
896
897 *typecode_or_type* determines the type of the returned object: it is either a
898 ctypes type or a one character typecode of the kind used by the :mod:`array`
899 module. *\*args* is passed on to the constructor for the type.
900
901 If *lock* is ``True`` (the default) then a new lock object is created to
902 synchronize access to the value. If *lock* is a :class:`Lock` or
903 :class:`RLock` object then that will be used to synchronize access to the
904 value. If *lock* is ``False`` then access to the returned object will not be
905 automatically protected by a lock, so it will not necessarily be
906 "process-safe".
907
908 Note that *lock* is a keyword-only argument.
909
910.. function:: Array(typecode_or_type, size_or_initializer, *, lock=True)
911
912 Return a ctypes array allocated from shared memory. By default the return
913 value is actually a synchronized wrapper for the array.
914
915 *typecode_or_type* determines the type of the elements of the returned array:
916 it is either a ctypes type or a one character typecode of the kind used by
917 the :mod:`array` module. If *size_or_initializer* is an integer, then it
918 determines the length of the array, and the array will be initially zeroed.
919 Otherwise, *size_or_initializer* is a sequence which is used to initialize
920 the array and whose length determines the length of the array.
921
922 If *lock* is ``True`` (the default) then a new lock object is created to
923 synchronize access to the value. If *lock* is a :class:`Lock` or
924 :class:`RLock` object then that will be used to synchronize access to the
925 value. If *lock* is ``False`` then access to the returned object will not be
926 automatically protected by a lock, so it will not necessarily be
927 "process-safe".
928
929 Note that *lock* is a keyword only argument.
930
Georg Brandlb053f992008-11-22 08:34:14 +0000931 Note that an array of :data:`ctypes.c_char` has *value* and *raw*
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000932 attributes which allow one to use it to store and retrieve strings.
933
934
935The :mod:`multiprocessing.sharedctypes` module
936>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
937
938.. module:: multiprocessing.sharedctypes
939 :synopsis: Allocate ctypes objects from shared memory.
940
941The :mod:`multiprocessing.sharedctypes` module provides functions for allocating
942:mod:`ctypes` objects from shared memory which can be inherited by child
943processes.
944
945.. note::
946
Benjamin Peterson90f36732008-07-12 20:16:19 +0000947 Although it is possible to store a pointer in shared memory remember that
948 this will refer to a location in the address space of a specific process.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000949 However, the pointer is quite likely to be invalid in the context of a second
950 process and trying to dereference the pointer from the second process may
951 cause a crash.
952
953.. function:: RawArray(typecode_or_type, size_or_initializer)
954
955 Return a ctypes array allocated from shared memory.
956
957 *typecode_or_type* determines the type of the elements of the returned array:
958 it is either a ctypes type or a one character typecode of the kind used by
959 the :mod:`array` module. If *size_or_initializer* is an integer then it
960 determines the length of the array, and the array will be initially zeroed.
961 Otherwise *size_or_initializer* is a sequence which is used to initialize the
962 array and whose length determines the length of the array.
963
964 Note that setting and getting an element is potentially non-atomic -- use
965 :func:`Array` instead to make sure that access is automatically synchronized
966 using a lock.
967
968.. function:: RawValue(typecode_or_type, *args)
969
970 Return a ctypes object allocated from shared memory.
971
972 *typecode_or_type* determines the type of the returned object: it is either a
973 ctypes type or a one character typecode of the kind used by the :mod:`array`
Jesse Noller6ab22152009-01-18 02:45:38 +0000974 module. *\*args* is passed on to the constructor for the type.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000975
976 Note that setting and getting the value is potentially non-atomic -- use
977 :func:`Value` instead to make sure that access is automatically synchronized
978 using a lock.
979
Georg Brandlb053f992008-11-22 08:34:14 +0000980 Note that an array of :data:`ctypes.c_char` has ``value`` and ``raw``
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000981 attributes which allow one to use it to store and retrieve strings -- see
982 documentation for :mod:`ctypes`.
983
Jesse Noller6ab22152009-01-18 02:45:38 +0000984.. function:: Array(typecode_or_type, size_or_initializer, *args[, lock])
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000985
986 The same as :func:`RawArray` except that depending on the value of *lock* a
987 process-safe synchronization wrapper may be returned instead of a raw ctypes
988 array.
989
990 If *lock* is ``True`` (the default) then a new lock object is created to
991 synchronize access to the value. If *lock* is a :class:`Lock` or
992 :class:`RLock` object then that will be used to synchronize access to the
993 value. If *lock* is ``False`` then access to the returned object will not be
994 automatically protected by a lock, so it will not necessarily be
995 "process-safe".
996
997 Note that *lock* is a keyword-only argument.
998
999.. function:: Value(typecode_or_type, *args[, lock])
1000
1001 The same as :func:`RawValue` except that depending on the value of *lock* a
1002 process-safe synchronization wrapper may be returned instead of a raw ctypes
1003 object.
1004
1005 If *lock* is ``True`` (the default) then a new lock object is created to
1006 synchronize access to the value. If *lock* is a :class:`Lock` or
1007 :class:`RLock` object then that will be used to synchronize access to the
1008 value. If *lock* is ``False`` then access to the returned object will not be
1009 automatically protected by a lock, so it will not necessarily be
1010 "process-safe".
1011
1012 Note that *lock* is a keyword-only argument.
1013
1014.. function:: copy(obj)
1015
1016 Return a ctypes object allocated from shared memory which is a copy of the
1017 ctypes object *obj*.
1018
1019.. function:: synchronized(obj[, lock])
1020
1021 Return a process-safe wrapper object for a ctypes object which uses *lock* to
1022 synchronize access. If *lock* is ``None`` (the default) then a
1023 :class:`multiprocessing.RLock` object is created automatically.
1024
1025 A synchronized wrapper will have two methods in addition to those of the
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001026 object it wraps: :meth:`get_obj` returns the wrapped object and
1027 :meth:`get_lock` returns the lock object used for synchronization.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001028
1029 Note that accessing the ctypes object through the wrapper can be a lot slower
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001030 than accessing the raw ctypes object.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001031
1032
1033The table below compares the syntax for creating shared ctypes objects from
1034shared memory with the normal ctypes syntax. (In the table ``MyStruct`` is some
1035subclass of :class:`ctypes.Structure`.)
1036
1037==================== ========================== ===========================
1038ctypes sharedctypes using type sharedctypes using typecode
1039==================== ========================== ===========================
1040c_double(2.4) RawValue(c_double, 2.4) RawValue('d', 2.4)
1041MyStruct(4, 6) RawValue(MyStruct, 4, 6)
1042(c_short * 7)() RawArray(c_short, 7) RawArray('h', 7)
1043(c_int * 3)(9, 2, 8) RawArray(c_int, (9, 2, 8)) RawArray('i', (9, 2, 8))
1044==================== ========================== ===========================
1045
1046
1047Below is an example where a number of ctypes objects are modified by a child
1048process::
1049
1050 from multiprocessing import Process, Lock
1051 from multiprocessing.sharedctypes import Value, Array
1052 from ctypes import Structure, c_double
1053
1054 class Point(Structure):
1055 _fields_ = [('x', c_double), ('y', c_double)]
1056
1057 def modify(n, x, s, A):
1058 n.value **= 2
1059 x.value **= 2
1060 s.value = s.value.upper()
1061 for a in A:
1062 a.x **= 2
1063 a.y **= 2
1064
1065 if __name__ == '__main__':
1066 lock = Lock()
1067
1068 n = Value('i', 7)
1069 x = Value(ctypes.c_double, 1.0/3.0, lock=False)
1070 s = Array('c', 'hello world', lock=lock)
1071 A = Array(Point, [(1.875,-6.25), (-5.75,2.0), (2.375,9.5)], lock=lock)
1072
1073 p = Process(target=modify, args=(n, x, s, A))
1074 p.start()
1075 p.join()
1076
1077 print n.value
1078 print x.value
1079 print s.value
1080 print [(a.x, a.y) for a in A]
1081
1082
1083.. highlightlang:: none
1084
1085The results printed are ::
1086
1087 49
1088 0.1111111111111111
1089 HELLO WORLD
1090 [(3.515625, 39.0625), (33.0625, 4.0), (5.640625, 90.25)]
1091
1092.. highlightlang:: python
1093
1094
1095.. _multiprocessing-managers:
1096
1097Managers
1098~~~~~~~~
1099
1100Managers provide a way to create data which can be shared between different
1101processes. A manager object controls a server process which manages *shared
1102objects*. Other processes can access the shared objects by using proxies.
1103
1104.. function:: multiprocessing.Manager()
1105
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001106 Returns a started :class:`~multiprocessing.managers.SyncManager` object which
1107 can be used for sharing objects between processes. The returned manager
1108 object corresponds to a spawned child process and has methods which will
1109 create shared objects and return corresponding proxies.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001110
1111.. module:: multiprocessing.managers
1112 :synopsis: Share data between process with shared objects.
1113
1114Manager processes will be shutdown as soon as they are garbage collected or
1115their parent process exits. The manager classes are defined in the
1116:mod:`multiprocessing.managers` module:
1117
1118.. class:: BaseManager([address[, authkey]])
1119
1120 Create a BaseManager object.
1121
1122 Once created one should call :meth:`start` or :meth:`serve_forever` to ensure
1123 that the manager object refers to a started manager process.
1124
1125 *address* is the address on which the manager process listens for new
1126 connections. If *address* is ``None`` then an arbitrary one is chosen.
1127
1128 *authkey* is the authentication key which will be used to check the validity
1129 of incoming connections to the server process. If *authkey* is ``None`` then
Benjamin Peterson73641d72008-08-20 14:07:59 +00001130 ``current_process().authkey``. Otherwise *authkey* is used and it
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001131 must be a string.
1132
1133 .. method:: start()
1134
1135 Start a subprocess to start the manager.
1136
Andrew M. Kuchlinga2478d92008-07-14 00:40:55 +00001137 .. method:: serve_forever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001138
1139 Run the server in the current process.
1140
1141 .. method:: from_address(address, authkey)
1142
1143 A class method which creates a manager object referring to a pre-existing
1144 server process which is using the given address and authentication key.
1145
Jesse Nollera280fd72008-11-28 18:22:54 +00001146 .. method:: get_server()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001147
Jesse Nollera280fd72008-11-28 18:22:54 +00001148 Returns a :class:`Server` object which represents the actual server under
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001149 the control of the Manager. The :class:`Server` object supports the
Georg Brandlfc29f272009-01-02 20:25:14 +00001150 :meth:`serve_forever` method:
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001151
Georg Brandlfc29f272009-01-02 20:25:14 +00001152 >>> from multiprocessing.managers import BaseManager
1153 >>> m = BaseManager(address=('', 50000), authkey='abc'))
1154 >>> server = m.get_server()
1155 >>> s.serve_forever()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001156
Georg Brandlfc29f272009-01-02 20:25:14 +00001157 :class:`Server` additionally have an :attr:`address` attribute.
Jesse Nollera280fd72008-11-28 18:22:54 +00001158
1159 .. method:: connect()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001160
Georg Brandlfc29f272009-01-02 20:25:14 +00001161 Connect a local manager object to a remote manager process:
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001162
Jesse Nollera280fd72008-11-28 18:22:54 +00001163 >>> from multiprocessing.managers import BaseManager
1164 >>> m = BaseManager(address='127.0.0.1', authkey='abc))
1165 >>> m.connect()
1166
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001167 .. method:: shutdown()
1168
1169 Stop the process used by the manager. This is only available if
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001170 :meth:`start` has been used to start the server process.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001171
1172 This can be called multiple times.
1173
1174 .. method:: register(typeid[, callable[, proxytype[, exposed[, method_to_typeid[, create_method]]]]])
1175
1176 A classmethod which can be used for registering a type or callable with
1177 the manager class.
1178
1179 *typeid* is a "type identifier" which is used to identify a particular
1180 type of shared object. This must be a string.
1181
1182 *callable* is a callable used for creating objects for this type
1183 identifier. If a manager instance will be created using the
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001184 :meth:`from_address` classmethod or if the *create_method* argument is
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001185 ``False`` then this can be left as ``None``.
1186
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001187 *proxytype* is a subclass of :class:`BaseProxy` which is used to create
1188 proxies for shared objects with this *typeid*. If ``None`` then a proxy
1189 class is created automatically.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001190
1191 *exposed* is used to specify a sequence of method names which proxies for
1192 this typeid should be allowed to access using
1193 :meth:`BaseProxy._callMethod`. (If *exposed* is ``None`` then
1194 :attr:`proxytype._exposed_` is used instead if it exists.) In the case
1195 where no exposed list is specified, all "public methods" of the shared
1196 object will be accessible. (Here a "public method" means any attribute
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001197 which has a :meth:`__call__` method and whose name does not begin with
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001198 ``'_'``.)
1199
1200 *method_to_typeid* is a mapping used to specify the return type of those
1201 exposed methods which should return a proxy. It maps method names to
1202 typeid strings. (If *method_to_typeid* is ``None`` then
1203 :attr:`proxytype._method_to_typeid_` is used instead if it exists.) If a
1204 method's name is not a key of this mapping or if the mapping is ``None``
1205 then the object returned by the method will be copied by value.
1206
1207 *create_method* determines whether a method should be created with name
1208 *typeid* which can be used to tell the server process to create a new
1209 shared object and return a proxy for it. By default it is ``True``.
1210
1211 :class:`BaseManager` instances also have one read-only property:
1212
1213 .. attribute:: address
1214
1215 The address used by the manager.
1216
1217
1218.. class:: SyncManager
1219
1220 A subclass of :class:`BaseManager` which can be used for the synchronization
1221 of processes. Objects of this type are returned by
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001222 :func:`multiprocessing.Manager`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001223
1224 It also supports creation of shared lists and dictionaries.
1225
1226 .. method:: BoundedSemaphore([value])
1227
1228 Create a shared :class:`threading.BoundedSemaphore` object and return a
1229 proxy for it.
1230
1231 .. method:: Condition([lock])
1232
1233 Create a shared :class:`threading.Condition` object and return a proxy for
1234 it.
1235
1236 If *lock* is supplied then it should be a proxy for a
1237 :class:`threading.Lock` or :class:`threading.RLock` object.
1238
1239 .. method:: Event()
1240
1241 Create a shared :class:`threading.Event` object and return a proxy for it.
1242
1243 .. method:: Lock()
1244
1245 Create a shared :class:`threading.Lock` object and return a proxy for it.
1246
1247 .. method:: Namespace()
1248
1249 Create a shared :class:`Namespace` object and return a proxy for it.
1250
1251 .. method:: Queue([maxsize])
1252
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001253 Create a shared :class:`Queue.Queue` object and return a proxy for it.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001254
1255 .. method:: RLock()
1256
1257 Create a shared :class:`threading.RLock` object and return a proxy for it.
1258
1259 .. method:: Semaphore([value])
1260
1261 Create a shared :class:`threading.Semaphore` object and return a proxy for
1262 it.
1263
1264 .. method:: Array(typecode, sequence)
1265
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001266 Create an array and return a proxy for it.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001267
1268 .. method:: Value(typecode, value)
1269
1270 Create an object with a writable ``value`` attribute and return a proxy
1271 for it.
1272
1273 .. method:: dict()
1274 dict(mapping)
1275 dict(sequence)
1276
1277 Create a shared ``dict`` object and return a proxy for it.
1278
1279 .. method:: list()
1280 list(sequence)
1281
1282 Create a shared ``list`` object and return a proxy for it.
1283
1284
1285Namespace objects
1286>>>>>>>>>>>>>>>>>
1287
1288A namespace object has no public methods, but does have writable attributes.
1289Its representation shows the values of its attributes.
1290
1291However, when using a proxy for a namespace object, an attribute beginning with
1292``'_'`` will be an attribute of the proxy and not an attribute of the referent::
1293
1294 >>> manager = multiprocessing.Manager()
1295 >>> Global = manager.Namespace()
1296 >>> Global.x = 10
1297 >>> Global.y = 'hello'
1298 >>> Global._z = 12.3 # this is an attribute of the proxy
1299 >>> print Global
1300 Namespace(x=10, y='hello')
1301
1302
1303Customized managers
1304>>>>>>>>>>>>>>>>>>>
1305
1306To create one's own manager, one creates a subclass of :class:`BaseManager` and
Georg Brandlfc29f272009-01-02 20:25:14 +00001307use the :meth:`~BaseManager.register` classmethod to register new types or
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001308callables with the manager class. For example::
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001309
1310 from multiprocessing.managers import BaseManager
1311
1312 class MathsClass(object):
1313 def add(self, x, y):
1314 return x + y
1315 def mul(self, x, y):
1316 return x * y
1317
1318 class MyManager(BaseManager):
1319 pass
1320
1321 MyManager.register('Maths', MathsClass)
1322
1323 if __name__ == '__main__':
1324 manager = MyManager()
1325 manager.start()
1326 maths = manager.Maths()
1327 print maths.add(4, 3) # prints 7
1328 print maths.mul(7, 8) # prints 56
1329
1330
1331Using a remote manager
1332>>>>>>>>>>>>>>>>>>>>>>
1333
1334It is possible to run a manager server on one machine and have clients use it
1335from other machines (assuming that the firewalls involved allow it).
1336
1337Running the following commands creates a server for a single shared queue which
1338remote clients can access::
1339
1340 >>> from multiprocessing.managers import BaseManager
1341 >>> import Queue
1342 >>> queue = Queue.Queue()
1343 >>> class QueueManager(BaseManager): pass
1344 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001345 >>> QueueManager.register('get_queue', callable=lambda:queue)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001346 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
Jesse Nollera280fd72008-11-28 18:22:54 +00001347 >>> s = m.get_server()
1348 >>> s.serveForever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001349
1350One client can access the server as follows::
1351
1352 >>> from multiprocessing.managers import BaseManager
1353 >>> class QueueManager(BaseManager): pass
1354 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001355 >>> QueueManager.register('get_queue')
1356 >>> m = QueueManager(address=('foo.bar.org', 50000), authkey='abracadabra')
1357 >>> m.connect()
1358 >>> queue = m.get_queue()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001359 >>> queue.put('hello')
1360
1361Another client can also use it::
1362
1363 >>> from multiprocessing.managers import BaseManager
1364 >>> class QueueManager(BaseManager): pass
1365 ...
1366 >>> QueueManager.register('getQueue')
1367 >>> m = QueueManager.from_address(address=('foo.bar.org', 50000), authkey='abracadabra')
1368 >>> queue = m.getQueue()
1369 >>> queue.get()
1370 'hello'
1371
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001372Local processes can also access that queue, using the code from above on the
Jesse Nollera280fd72008-11-28 18:22:54 +00001373client to access it remotely::
1374
1375 >>> from multiprocessing import Process, Queue
1376 >>> from multiprocessing.managers import BaseManager
1377 >>> class Worker(Process):
1378 ... def __init__(self, q):
1379 ... self.q = q
1380 ... super(Worker, self).__init__()
1381 ... def run(self):
1382 ... self.q.put('local hello')
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001383 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001384 >>> queue = Queue()
1385 >>> w = Worker(queue)
1386 >>> w.start()
1387 >>> class QueueManager(BaseManager): pass
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001388 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001389 >>> QueueManager.register('get_queue', callable=lambda: queue)
1390 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
1391 >>> s = m.get_server()
1392 >>> s.serve_forever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001393
1394Proxy Objects
1395~~~~~~~~~~~~~
1396
1397A proxy is an object which *refers* to a shared object which lives (presumably)
1398in a different process. The shared object is said to be the *referent* of the
1399proxy. Multiple proxy objects may have the same referent.
1400
1401A proxy object has methods which invoke corresponding methods of its referent
1402(although not every method of the referent will necessarily be available through
1403the proxy). A proxy can usually be used in most of the same ways that its
1404referent can::
1405
1406 >>> from multiprocessing import Manager
1407 >>> manager = Manager()
1408 >>> l = manager.list([i*i for i in range(10)])
1409 >>> print l
1410 [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
1411 >>> print repr(l)
1412 <ListProxy object, typeid 'list' at 0xb799974c>
1413 >>> l[4]
1414 16
1415 >>> l[2:5]
1416 [4, 9, 16]
1417
1418Notice that applying :func:`str` to a proxy will return the representation of
1419the referent, whereas applying :func:`repr` will return the representation of
1420the proxy.
1421
1422An important feature of proxy objects is that they are picklable so they can be
1423passed between processes. Note, however, that if a proxy is sent to the
1424corresponding manager's process then unpickling it will produce the referent
1425itself. This means, for example, that one shared object can contain a second::
1426
1427 >>> a = manager.list()
1428 >>> b = manager.list()
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001429 >>> a.append(b) # referent of a now contains referent of b
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001430 >>> print a, b
1431 [[]] []
1432 >>> b.append('hello')
1433 >>> print a, b
1434 [['hello']] ['hello']
1435
1436.. note::
1437
1438 The proxy types in :mod:`multiprocessing` do nothing to support comparisons
1439 by value. So, for instance, ::
1440
1441 manager.list([1,2,3]) == [1,2,3]
1442
1443 will return ``False``. One should just use a copy of the referent instead
1444 when making comparisons.
1445
1446.. class:: BaseProxy
1447
1448 Proxy objects are instances of subclasses of :class:`BaseProxy`.
1449
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001450 .. method:: _callmethod(methodname[, args[, kwds]])
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001451
1452 Call and return the result of a method of the proxy's referent.
1453
1454 If ``proxy`` is a proxy whose referent is ``obj`` then the expression ::
1455
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001456 proxy._callmethod(methodname, args, kwds)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001457
1458 will evaluate the expression ::
1459
1460 getattr(obj, methodname)(*args, **kwds)
1461
1462 in the manager's process.
1463
1464 The returned value will be a copy of the result of the call or a proxy to
1465 a new shared object -- see documentation for the *method_to_typeid*
1466 argument of :meth:`BaseManager.register`.
1467
1468 If an exception is raised by the call, then then is re-raised by
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001469 :meth:`_callmethod`. If some other exception is raised in the manager's
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001470 process then this is converted into a :exc:`RemoteError` exception and is
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001471 raised by :meth:`_callmethod`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001472
1473 Note in particular that an exception will be raised if *methodname* has
1474 not been *exposed*
1475
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001476 An example of the usage of :meth:`_callmethod`::
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001477
1478 >>> l = manager.list(range(10))
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001479 >>> l._callmethod('__len__')
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001480 10
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001481 >>> l._callmethod('__getslice__', (2, 7)) # equiv to `l[2:7]`
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001482 [2, 3, 4, 5, 6]
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001483 >>> l._callmethod('__getitem__', (20,)) # equiv to `l[20]`
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001484 Traceback (most recent call last):
1485 ...
1486 IndexError: list index out of range
1487
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001488 .. method:: _getvalue()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001489
1490 Return a copy of the referent.
1491
1492 If the referent is unpicklable then this will raise an exception.
1493
1494 .. method:: __repr__
1495
1496 Return a representation of the proxy object.
1497
1498 .. method:: __str__
1499
1500 Return the representation of the referent.
1501
1502
1503Cleanup
1504>>>>>>>
1505
1506A proxy object uses a weakref callback so that when it gets garbage collected it
1507deregisters itself from the manager which owns its referent.
1508
1509A shared object gets deleted from the manager process when there are no longer
1510any proxies referring to it.
1511
1512
1513Process Pools
1514~~~~~~~~~~~~~
1515
1516.. module:: multiprocessing.pool
1517 :synopsis: Create pools of processes.
1518
1519One can create a pool of processes which will carry out tasks submitted to it
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001520with the :class:`Pool` class.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001521
1522.. class:: multiprocessing.Pool([processes[, initializer[, initargs]]])
1523
1524 A process pool object which controls a pool of worker processes to which jobs
1525 can be submitted. It supports asynchronous results with timeouts and
1526 callbacks and has a parallel map implementation.
1527
1528 *processes* is the number of worker processes to use. If *processes* is
1529 ``None`` then the number returned by :func:`cpu_count` is used. If
1530 *initializer* is not ``None`` then each worker process will call
1531 ``initializer(*initargs)`` when it starts.
1532
1533 .. method:: apply(func[, args[, kwds]])
1534
1535 Equivalent of the :func:`apply` builtin function. It blocks till the
Jesse Noller403c6632009-01-22 21:53:22 +00001536 result is ready. Given this blocks - :meth:`apply_async` is better suited
1537 for performing work in parallel. Additionally, the passed
1538 in function is only executed in one of the workers of the pool.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001539
1540 .. method:: apply_async(func[, args[, kwds[, callback]]])
1541
1542 A variant of the :meth:`apply` method which returns a result object.
1543
1544 If *callback* is specified then it should be a callable which accepts a
1545 single argument. When the result becomes ready *callback* is applied to
1546 it (unless the call failed). *callback* should complete immediately since
1547 otherwise the thread which handles the results will get blocked.
1548
1549 .. method:: map(func, iterable[, chunksize])
1550
1551 A parallel equivalent of the :func:`map` builtin function. It blocks till
1552 the result is ready.
1553
1554 This method chops the iterable into a number of chunks which it submits to
1555 the process pool as separate tasks. The (approximate) size of these
1556 chunks can be specified by setting *chunksize* to a positive integer.
1557
1558 .. method:: map_async(func, iterable[, chunksize[, callback]])
1559
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001560 A variant of the :meth:`map` method which returns a result object.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001561
1562 If *callback* is specified then it should be a callable which accepts a
1563 single argument. When the result becomes ready *callback* is applied to
1564 it (unless the call failed). *callback* should complete immediately since
1565 otherwise the thread which handles the results will get blocked.
1566
1567 .. method:: imap(func, iterable[, chunksize])
1568
1569 An equivalent of :func:`itertools.imap`.
1570
1571 The *chunksize* argument is the same as the one used by the :meth:`.map`
1572 method. For very long iterables using a large value for *chunksize* can
1573 make make the job complete **much** faster than using the default value of
1574 ``1``.
1575
1576 Also if *chunksize* is ``1`` then the :meth:`next` method of the iterator
1577 returned by the :meth:`imap` method has an optional *timeout* parameter:
1578 ``next(timeout)`` will raise :exc:`multiprocessing.TimeoutError` if the
1579 result cannot be returned within *timeout* seconds.
1580
1581 .. method:: imap_unordered(func, iterable[, chunksize])
1582
1583 The same as :meth:`imap` except that the ordering of the results from the
1584 returned iterator should be considered arbitrary. (Only when there is
1585 only one worker process is the order guaranteed to be "correct".)
1586
1587 .. method:: close()
1588
1589 Prevents any more tasks from being submitted to the pool. Once all the
1590 tasks have been completed the worker processes will exit.
1591
1592 .. method:: terminate()
1593
1594 Stops the worker processes immediately without completing outstanding
1595 work. When the pool object is garbage collected :meth:`terminate` will be
1596 called immediately.
1597
1598 .. method:: join()
1599
1600 Wait for the worker processes to exit. One must call :meth:`close` or
1601 :meth:`terminate` before using :meth:`join`.
1602
1603
1604.. class:: AsyncResult
1605
1606 The class of the result returned by :meth:`Pool.apply_async` and
1607 :meth:`Pool.map_async`.
1608
Jesse Nollera280fd72008-11-28 18:22:54 +00001609 .. method:: get([timeout])
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001610
1611 Return the result when it arrives. If *timeout* is not ``None`` and the
1612 result does not arrive within *timeout* seconds then
1613 :exc:`multiprocessing.TimeoutError` is raised. If the remote call raised
1614 an exception then that exception will be reraised by :meth:`get`.
1615
1616 .. method:: wait([timeout])
1617
1618 Wait until the result is available or until *timeout* seconds pass.
1619
1620 .. method:: ready()
1621
1622 Return whether the call has completed.
1623
1624 .. method:: successful()
1625
1626 Return whether the call completed without raising an exception. Will
1627 raise :exc:`AssertionError` if the result is not ready.
1628
1629The following example demonstrates the use of a pool::
1630
1631 from multiprocessing import Pool
1632
1633 def f(x):
1634 return x*x
1635
1636 if __name__ == '__main__':
1637 pool = Pool(processes=4) # start 4 worker processes
1638
Jesse Nollera280fd72008-11-28 18:22:54 +00001639 result = pool.apply_async(f, (10,)) # evaluate "f(10)" asynchronously
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001640 print result.get(timeout=1) # prints "100" unless your computer is *very* slow
1641
1642 print pool.map(f, range(10)) # prints "[0, 1, 4,..., 81]"
1643
1644 it = pool.imap(f, range(10))
1645 print it.next() # prints "0"
1646 print it.next() # prints "1"
1647 print it.next(timeout=1) # prints "4" unless your computer is *very* slow
1648
1649 import time
Jesse Nollera280fd72008-11-28 18:22:54 +00001650 result = pool.apply_async(time.sleep, (10,))
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001651 print result.get(timeout=1) # raises TimeoutError
1652
1653
1654.. _multiprocessing-listeners-clients:
1655
1656Listeners and Clients
1657~~~~~~~~~~~~~~~~~~~~~
1658
1659.. module:: multiprocessing.connection
1660 :synopsis: API for dealing with sockets.
1661
1662Usually message passing between processes is done using queues or by using
1663:class:`Connection` objects returned by :func:`Pipe`.
1664
1665However, the :mod:`multiprocessing.connection` module allows some extra
1666flexibility. It basically gives a high level message oriented API for dealing
1667with sockets or Windows named pipes, and also has support for *digest
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001668authentication* using the :mod:`hmac` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001669
1670
1671.. function:: deliver_challenge(connection, authkey)
1672
1673 Send a randomly generated message to the other end of the connection and wait
1674 for a reply.
1675
1676 If the reply matches the digest of the message using *authkey* as the key
1677 then a welcome message is sent to the other end of the connection. Otherwise
1678 :exc:`AuthenticationError` is raised.
1679
1680.. function:: answerChallenge(connection, authkey)
1681
1682 Receive a message, calculate the digest of the message using *authkey* as the
1683 key, and then send the digest back.
1684
1685 If a welcome message is not received, then :exc:`AuthenticationError` is
1686 raised.
1687
1688.. function:: Client(address[, family[, authenticate[, authkey]]])
1689
1690 Attempt to set up a connection to the listener which is using address
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001691 *address*, returning a :class:`~multiprocessing.Connection`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001692
1693 The type of the connection is determined by *family* argument, but this can
1694 generally be omitted since it can usually be inferred from the format of
1695 *address*. (See :ref:`multiprocessing-address-formats`)
1696
1697 If *authentication* is ``True`` or *authkey* is a string then digest
1698 authentication is used. The key used for authentication will be either
Benjamin Peterson73641d72008-08-20 14:07:59 +00001699 *authkey* or ``current_process().authkey)`` if *authkey* is ``None``.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001700 If authentication fails then :exc:`AuthenticationError` is raised. See
1701 :ref:`multiprocessing-auth-keys`.
1702
1703.. class:: Listener([address[, family[, backlog[, authenticate[, authkey]]]]])
1704
1705 A wrapper for a bound socket or Windows named pipe which is 'listening' for
1706 connections.
1707
1708 *address* is the address to be used by the bound socket or named pipe of the
1709 listener object.
1710
Jesse Nollerb12e79d2009-04-01 16:42:19 +00001711 .. note::
1712
1713 If an address of '0.0.0.0' is used, the address will not be a connectable
1714 end point on Windows. If you require a connectable end-point,
1715 you should use '127.0.0.1'.
1716
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001717 *family* is the type of socket (or named pipe) to use. This can be one of
1718 the strings ``'AF_INET'`` (for a TCP socket), ``'AF_UNIX'`` (for a Unix
1719 domain socket) or ``'AF_PIPE'`` (for a Windows named pipe). Of these only
1720 the first is guaranteed to be available. If *family* is ``None`` then the
1721 family is inferred from the format of *address*. If *address* is also
1722 ``None`` then a default is chosen. This default is the family which is
1723 assumed to be the fastest available. See
1724 :ref:`multiprocessing-address-formats`. Note that if *family* is
1725 ``'AF_UNIX'`` and address is ``None`` then the socket will be created in a
1726 private temporary directory created using :func:`tempfile.mkstemp`.
1727
1728 If the listener object uses a socket then *backlog* (1 by default) is passed
1729 to the :meth:`listen` method of the socket once it has been bound.
1730
1731 If *authenticate* is ``True`` (``False`` by default) or *authkey* is not
1732 ``None`` then digest authentication is used.
1733
1734 If *authkey* is a string then it will be used as the authentication key;
1735 otherwise it must be *None*.
1736
1737 If *authkey* is ``None`` and *authenticate* is ``True`` then
Benjamin Peterson73641d72008-08-20 14:07:59 +00001738 ``current_process().authkey`` is used as the authentication key. If
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001739 *authkey* is ``None`` and *authentication* is ``False`` then no
1740 authentication is done. If authentication fails then
1741 :exc:`AuthenticationError` is raised. See :ref:`multiprocessing-auth-keys`.
1742
1743 .. method:: accept()
1744
1745 Accept a connection on the bound socket or named pipe of the listener
1746 object and return a :class:`Connection` object. If authentication is
1747 attempted and fails, then :exc:`AuthenticationError` is raised.
1748
1749 .. method:: close()
1750
1751 Close the bound socket or named pipe of the listener object. This is
1752 called automatically when the listener is garbage collected. However it
1753 is advisable to call it explicitly.
1754
1755 Listener objects have the following read-only properties:
1756
1757 .. attribute:: address
1758
1759 The address which is being used by the Listener object.
1760
1761 .. attribute:: last_accepted
1762
1763 The address from which the last accepted connection came. If this is
1764 unavailable then it is ``None``.
1765
1766
1767The module defines two exceptions:
1768
1769.. exception:: AuthenticationError
1770
1771 Exception raised when there is an authentication error.
1772
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001773
1774**Examples**
1775
1776The following server code creates a listener which uses ``'secret password'`` as
1777an authentication key. It then waits for a connection and sends some data to
1778the client::
1779
1780 from multiprocessing.connection import Listener
1781 from array import array
1782
1783 address = ('localhost', 6000) # family is deduced to be 'AF_INET'
1784 listener = Listener(address, authkey='secret password')
1785
1786 conn = listener.accept()
1787 print 'connection accepted from', listener.last_accepted
1788
1789 conn.send([2.25, None, 'junk', float])
1790
1791 conn.send_bytes('hello')
1792
1793 conn.send_bytes(array('i', [42, 1729]))
1794
1795 conn.close()
1796 listener.close()
1797
1798The following code connects to the server and receives some data from the
1799server::
1800
1801 from multiprocessing.connection import Client
1802 from array import array
1803
1804 address = ('localhost', 6000)
1805 conn = Client(address, authkey='secret password')
1806
1807 print conn.recv() # => [2.25, None, 'junk', float]
1808
1809 print conn.recv_bytes() # => 'hello'
1810
1811 arr = array('i', [0, 0, 0, 0, 0])
1812 print conn.recv_bytes_into(arr) # => 8
1813 print arr # => array('i', [42, 1729, 0, 0, 0])
1814
1815 conn.close()
1816
1817
1818.. _multiprocessing-address-formats:
1819
1820Address Formats
1821>>>>>>>>>>>>>>>
1822
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001823* An ``'AF_INET'`` address is a tuple of the form ``(hostname, port)`` where
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001824 *hostname* is a string and *port* is an integer.
1825
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001826* An ``'AF_UNIX'`` address is a string representing a filename on the
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001827 filesystem.
1828
1829* An ``'AF_PIPE'`` address is a string of the form
Georg Brandl6b28f392008-12-27 19:06:04 +00001830 :samp:`r'\\\\.\\pipe\\{PipeName}'`. To use :func:`Client` to connect to a named
Georg Brandlfc29f272009-01-02 20:25:14 +00001831 pipe on a remote computer called *ServerName* one should use an address of the
Georg Brandldd7e3132009-01-04 10:24:09 +00001832 form :samp:`r'\\\\{ServerName}\\pipe\\{PipeName}'` instead.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001833
1834Note that any string beginning with two backslashes is assumed by default to be
1835an ``'AF_PIPE'`` address rather than an ``'AF_UNIX'`` address.
1836
1837
1838.. _multiprocessing-auth-keys:
1839
1840Authentication keys
1841~~~~~~~~~~~~~~~~~~~
1842
1843When one uses :meth:`Connection.recv`, the data received is automatically
1844unpickled. Unfortunately unpickling data from an untrusted source is a security
1845risk. Therefore :class:`Listener` and :func:`Client` use the :mod:`hmac` module
1846to provide digest authentication.
1847
1848An authentication key is a string which can be thought of as a password: once a
1849connection is established both ends will demand proof that the other knows the
1850authentication key. (Demonstrating that both ends are using the same key does
1851**not** involve sending the key over the connection.)
1852
1853If authentication is requested but do authentication key is specified then the
Benjamin Peterson73641d72008-08-20 14:07:59 +00001854return value of ``current_process().authkey`` is used (see
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001855:class:`~multiprocessing.Process`). This value will automatically inherited by
1856any :class:`~multiprocessing.Process` object that the current process creates.
1857This means that (by default) all processes of a multi-process program will share
1858a single authentication key which can be used when setting up connections
1859between the themselves.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001860
1861Suitable authentication keys can also be generated by using :func:`os.urandom`.
1862
1863
1864Logging
1865~~~~~~~
1866
1867Some support for logging is available. Note, however, that the :mod:`logging`
1868package does not use process shared locks so it is possible (depending on the
1869handler type) for messages from different processes to get mixed up.
1870
1871.. currentmodule:: multiprocessing
1872.. function:: get_logger()
1873
1874 Returns the logger used by :mod:`multiprocessing`. If necessary, a new one
1875 will be created.
1876
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001877 When first created the logger has level :data:`logging.NOTSET` and no
1878 default handler. Messages sent to this logger will not by default propagate
1879 to the root logger.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001880
1881 Note that on Windows child processes will only inherit the level of the
1882 parent process's logger -- any other customization of the logger will not be
1883 inherited.
1884
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001885.. currentmodule:: multiprocessing
1886.. function:: log_to_stderr()
1887
1888 This function performs a call to :func:`get_logger` but in addition to
1889 returning the logger created by get_logger, it adds a handler which sends
1890 output to :data:`sys.stderr` using format
1891 ``'[%(levelname)s/%(processName)s] %(message)s'``.
1892
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001893Below is an example session with logging turned on::
1894
Georg Brandl19cc9442008-10-16 21:36:39 +00001895 >>> import multiprocessing, logging
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001896 >>> logger = multiprocessing.log_to_stderr()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001897 >>> logger.setLevel(logging.INFO)
1898 >>> logger.warning('doomed')
1899 [WARNING/MainProcess] doomed
Georg Brandl19cc9442008-10-16 21:36:39 +00001900 >>> m = multiprocessing.Manager()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001901 [INFO/SyncManager-1] child process calling self.run()
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001902 [INFO/SyncManager-1] created temp directory /.../pymp-Wh47O_
1903 [INFO/SyncManager-1] manager serving at '/.../listener-lWsERs'
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001904 >>> del m
1905 [INFO/MainProcess] sending shutdown message to manager
1906 [INFO/SyncManager-1] manager exiting with exitcode 0
1907
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001908In addition to having these two logging functions, the multiprocessing also
1909exposes two additional logging level attributes. These are :const:`SUBWARNING`
1910and :const:`SUBDEBUG`. The table below illustrates where theses fit in the
1911normal level hierarchy.
1912
1913+----------------+----------------+
1914| Level | Numeric value |
1915+================+================+
1916| ``SUBWARNING`` | 25 |
1917+----------------+----------------+
1918| ``SUBDEBUG`` | 5 |
1919+----------------+----------------+
1920
1921For a full table of logging levels, see the :mod:`logging` module.
1922
1923These additional logging levels are used primarily for certain debug messages
1924within the multiprocessing module. Below is the same example as above, except
1925with :const:`SUBDEBUG` enabled::
1926
1927 >>> import multiprocessing, logging
1928 >>> logger = multiprocessing.log_to_stderr()
1929 >>> logger.setLevel(multiprocessing.SUBDEBUG)
1930 >>> logger.warning('doomed')
1931 [WARNING/MainProcess] doomed
1932 >>> m = multiprocessing.Manager()
1933 [INFO/SyncManager-1] child process calling self.run()
1934 [INFO/SyncManager-1] created temp directory /.../pymp-djGBXN
1935 [INFO/SyncManager-1] manager serving at '/.../pymp-djGBXN/listener-knBYGe'
1936 >>> del m
1937 [SUBDEBUG/MainProcess] finalizer calling ...
1938 [INFO/MainProcess] sending shutdown message to manager
1939 [DEBUG/SyncManager-1] manager received shutdown message
1940 [SUBDEBUG/SyncManager-1] calling <Finalize object, callback=unlink, ...
1941 [SUBDEBUG/SyncManager-1] finalizer calling <built-in function unlink> ...
1942 [SUBDEBUG/SyncManager-1] calling <Finalize object, dead>
1943 [SUBDEBUG/SyncManager-1] finalizer calling <function rmtree at 0x5aa730> ...
1944 [INFO/SyncManager-1] manager exiting with exitcode 0
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001945
1946The :mod:`multiprocessing.dummy` module
1947~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1948
1949.. module:: multiprocessing.dummy
1950 :synopsis: Dumb wrapper around threading.
1951
1952:mod:`multiprocessing.dummy` replicates the API of :mod:`multiprocessing` but is
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001953no more than a wrapper around the :mod:`threading` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001954
1955
1956.. _multiprocessing-programming:
1957
1958Programming guidelines
1959----------------------
1960
1961There are certain guidelines and idioms which should be adhered to when using
1962:mod:`multiprocessing`.
1963
1964
1965All platforms
1966~~~~~~~~~~~~~
1967
1968Avoid shared state
1969
1970 As far as possible one should try to avoid shifting large amounts of data
1971 between processes.
1972
1973 It is probably best to stick to using queues or pipes for communication
1974 between processes rather than using the lower level synchronization
1975 primitives from the :mod:`threading` module.
1976
1977Picklability
1978
1979 Ensure that the arguments to the methods of proxies are picklable.
1980
1981Thread safety of proxies
1982
1983 Do not use a proxy object from more than one thread unless you protect it
1984 with a lock.
1985
1986 (There is never a problem with different processes using the *same* proxy.)
1987
1988Joining zombie processes
1989
1990 On Unix when a process finishes but has not been joined it becomes a zombie.
1991 There should never be very many because each time a new process starts (or
1992 :func:`active_children` is called) all completed processes which have not
1993 yet been joined will be joined. Also calling a finished process's
1994 :meth:`Process.is_alive` will join the process. Even so it is probably good
1995 practice to explicitly join all the processes that you start.
1996
1997Better to inherit than pickle/unpickle
1998
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001999 On Windows many types from :mod:`multiprocessing` need to be picklable so
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002000 that child processes can use them. However, one should generally avoid
2001 sending shared objects to other processes using pipes or queues. Instead
2002 you should arrange the program so that a process which need access to a
2003 shared resource created elsewhere can inherit it from an ancestor process.
2004
2005Avoid terminating processes
2006
2007 Using the :meth:`Process.terminate` method to stop a process is liable to
2008 cause any shared resources (such as locks, semaphores, pipes and queues)
2009 currently being used by the process to become broken or unavailable to other
2010 processes.
2011
2012 Therefore it is probably best to only consider using
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002013 :meth:`Process.terminate` on processes which never use any shared resources.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002014
2015Joining processes that use queues
2016
2017 Bear in mind that a process that has put items in a queue will wait before
2018 terminating until all the buffered items are fed by the "feeder" thread to
2019 the underlying pipe. (The child process can call the
Jesse Nollerd5ff5b22008-09-06 01:20:11 +00002020 :meth:`Queue.cancel_join_thread` method of the queue to avoid this behaviour.)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002021
2022 This means that whenever you use a queue you need to make sure that all
2023 items which have been put on the queue will eventually be removed before the
2024 process is joined. Otherwise you cannot be sure that processes which have
2025 put items on the queue will terminate. Remember also that non-daemonic
2026 processes will be automatically be joined.
2027
2028 An example which will deadlock is the following::
2029
2030 from multiprocessing import Process, Queue
2031
2032 def f(q):
2033 q.put('X' * 1000000)
2034
2035 if __name__ == '__main__':
2036 queue = Queue()
2037 p = Process(target=f, args=(queue,))
2038 p.start()
2039 p.join() # this deadlocks
2040 obj = queue.get()
2041
2042 A fix here would be to swap the last two lines round (or simply remove the
2043 ``p.join()`` line).
2044
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00002045Explicitly pass resources to child processes
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002046
2047 On Unix a child process can make use of a shared resource created in a
2048 parent process using a global resource. However, it is better to pass the
2049 object as an argument to the constructor for the child process.
2050
2051 Apart from making the code (potentially) compatible with Windows this also
2052 ensures that as long as the child process is still alive the object will not
2053 be garbage collected in the parent process. This might be important if some
2054 resource is freed when the object is garbage collected in the parent
2055 process.
2056
2057 So for instance ::
2058
2059 from multiprocessing import Process, Lock
2060
2061 def f():
2062 ... do something using "lock" ...
2063
2064 if __name__ == '__main__':
2065 lock = Lock()
2066 for i in range(10):
2067 Process(target=f).start()
2068
2069 should be rewritten as ::
2070
2071 from multiprocessing import Process, Lock
2072
2073 def f(l):
2074 ... do something using "l" ...
2075
2076 if __name__ == '__main__':
2077 lock = Lock()
2078 for i in range(10):
2079 Process(target=f, args=(lock,)).start()
2080
2081
2082Windows
2083~~~~~~~
2084
2085Since Windows lacks :func:`os.fork` it has a few extra restrictions:
2086
2087More picklability
2088
2089 Ensure that all arguments to :meth:`Process.__init__` are picklable. This
2090 means, in particular, that bound or unbound methods cannot be used directly
2091 as the ``target`` argument on Windows --- just define a function and use
2092 that instead.
2093
2094 Also, if you subclass :class:`Process` then make sure that instances will be
2095 picklable when the :meth:`Process.start` method is called.
2096
2097Global variables
2098
2099 Bear in mind that if code run in a child process tries to access a global
2100 variable, then the value it sees (if any) may not be the same as the value
2101 in the parent process at the time that :meth:`Process.start` was called.
2102
2103 However, global variables which are just module level constants cause no
2104 problems.
2105
2106Safe importing of main module
2107
2108 Make sure that the main module can be safely imported by a new Python
2109 interpreter without causing unintended side effects (such a starting a new
2110 process).
2111
2112 For example, under Windows running the following module would fail with a
2113 :exc:`RuntimeError`::
2114
2115 from multiprocessing import Process
2116
2117 def foo():
2118 print 'hello'
2119
2120 p = Process(target=foo)
2121 p.start()
2122
2123 Instead one should protect the "entry point" of the program by using ``if
2124 __name__ == '__main__':`` as follows::
2125
2126 from multiprocessing import Process, freeze_support
2127
2128 def foo():
2129 print 'hello'
2130
2131 if __name__ == '__main__':
2132 freeze_support()
2133 p = Process(target=foo)
2134 p.start()
2135
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002136 (The ``freeze_support()`` line can be omitted if the program will be run
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002137 normally instead of frozen.)
2138
2139 This allows the newly spawned Python interpreter to safely import the module
2140 and then run the module's ``foo()`` function.
2141
2142 Similar restrictions apply if a pool or manager is created in the main
2143 module.
2144
2145
2146.. _multiprocessing-examples:
2147
2148Examples
2149--------
2150
2151Demonstration of how to create and use customized managers and proxies:
2152
2153.. literalinclude:: ../includes/mp_newtype.py
2154
2155
2156Using :class:`Pool`:
2157
2158.. literalinclude:: ../includes/mp_pool.py
2159
2160
2161Synchronization types like locks, conditions and queues:
2162
2163.. literalinclude:: ../includes/mp_synchronize.py
2164
2165
2166An showing how to use queues to feed tasks to a collection of worker process and
2167collect the results:
2168
2169.. literalinclude:: ../includes/mp_workers.py
2170
2171
2172An example of how a pool of worker processes can each run a
2173:class:`SimpleHTTPServer.HttpServer` instance while sharing a single listening
2174socket.
2175
2176.. literalinclude:: ../includes/mp_webserver.py
2177
2178
2179Some simple benchmarks comparing :mod:`multiprocessing` with :mod:`threading`:
2180
2181.. literalinclude:: ../includes/mp_benchmarks.py
2182
2183An example/demo of how to use the :class:`managers.SyncManager`, :class:`Process`
Georg Brandlc62ef8b2009-01-03 20:55:06 +00002184and others to build a system which can distribute processes and work via a
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002185distributed queue to a "cluster" of machines on a network, accessible via SSH.
2186You will need to have private key authentication for all hosts configured for
2187this to work.
2188
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002189.. literalinclude:: ../includes/mp_distributing.py