blob: 02f6ac1c4712a368485442bafb3ceccc481537f7 [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
713 The object must be picklable.
714
715 .. method:: recv()
716
717 Return an object sent from the other end of the connection using
718 :meth:`send`. Raises :exc:`EOFError` if there is nothing left to receive
719 and the other end was closed.
720
721 .. method:: fileno()
722
723 Returns the file descriptor or handle used by the connection.
724
725 .. method:: close()
726
727 Close the connection.
728
729 This is called automatically when the connection is garbage collected.
730
731 .. method:: poll([timeout])
732
733 Return whether there is any data available to be read.
734
735 If *timeout* is not specified then it will return immediately. If
736 *timeout* is a number then this specifies the maximum time in seconds to
737 block. If *timeout* is ``None`` then an infinite timeout is used.
738
739 .. method:: send_bytes(buffer[, offset[, size]])
740
741 Send byte data from an object supporting the buffer interface as a
742 complete message.
743
744 If *offset* is given then data is read from that position in *buffer*. If
745 *size* is given then that many bytes will be read from buffer.
746
747 .. method:: recv_bytes([maxlength])
748
749 Return a complete message of byte data sent from the other end of the
750 connection as a string. Raises :exc:`EOFError` if there is nothing left
751 to receive and the other end has closed.
752
753 If *maxlength* is specified and the message is longer than *maxlength*
754 then :exc:`IOError` is raised and the connection will no longer be
755 readable.
756
757 .. method:: recv_bytes_into(buffer[, offset])
758
759 Read into *buffer* a complete message of byte data sent from the other end
760 of the connection and return the number of bytes in the message. Raises
761 :exc:`EOFError` if there is nothing left to receive and the other end was
762 closed.
763
764 *buffer* must be an object satisfying the writable buffer interface. If
765 *offset* is given then the message will be written into the buffer from
766 *that position. Offset must be a non-negative integer less than the
767 *length of *buffer* (in bytes).
768
769 If the buffer is too short then a :exc:`BufferTooShort` exception is
770 raised and the complete message is available as ``e.args[0]`` where ``e``
771 is the exception instance.
772
773
774For example:
775
776 >>> from multiprocessing import Pipe
777 >>> a, b = Pipe()
778 >>> a.send([1, 'hello', None])
779 >>> b.recv()
780 [1, 'hello', None]
781 >>> b.send_bytes('thank you')
782 >>> a.recv_bytes()
783 'thank you'
784 >>> import array
785 >>> arr1 = array.array('i', range(5))
786 >>> arr2 = array.array('i', [0] * 10)
787 >>> a.send_bytes(arr1)
788 >>> count = b.recv_bytes_into(arr2)
789 >>> assert count == len(arr1) * arr1.itemsize
790 >>> arr2
791 array('i', [0, 1, 2, 3, 4, 0, 0, 0, 0, 0])
792
793
794.. warning::
795
796 The :meth:`Connection.recv` method automatically unpickles the data it
797 receives, which can be a security risk unless you can trust the process
798 which sent the message.
799
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000800 Therefore, unless the connection object was produced using :func:`Pipe` you
801 should only use the :meth:`~Connection.recv` and :meth:`~Connection.send`
802 methods after performing some sort of authentication. See
803 :ref:`multiprocessing-auth-keys`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000804
805.. warning::
806
807 If a process is killed while it is trying to read or write to a pipe then
808 the data in the pipe is likely to become corrupted, because it may become
809 impossible to be sure where the message boundaries lie.
810
811
812Synchronization primitives
813~~~~~~~~~~~~~~~~~~~~~~~~~~
814
815Generally synchronization primitives are not as necessary in a multiprocess
Andrew M. Kuchling8ea605c2008-07-14 01:18:16 +0000816program as they are in a multithreaded program. See the documentation for
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000817:mod:`threading` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000818
819Note that one can also create synchronization primitives by using a manager
820object -- see :ref:`multiprocessing-managers`.
821
822.. class:: BoundedSemaphore([value])
823
824 A bounded semaphore object: a clone of :class:`threading.BoundedSemaphore`.
825
Georg Brandl9af94982008-09-13 17:41:16 +0000826 (On Mac OS X this is indistinguishable from :class:`Semaphore` because
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000827 ``sem_getvalue()`` is not implemented on that platform).
828
829.. class:: Condition([lock])
830
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000831 A condition variable: a clone of :class:`threading.Condition`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000832
833 If *lock* is specified then it should be a :class:`Lock` or :class:`RLock`
834 object from :mod:`multiprocessing`.
835
836.. class:: Event()
837
838 A clone of :class:`threading.Event`.
Jesse Noller02cb0eb2009-04-01 03:45:50 +0000839 This method returns the state of the internal semaphore on exit, so it
840 will always return ``True`` except if a timeout is given and the operation
841 times out.
842
843 .. versionchanged:: 2.7
844 Previously, the method always returned ``None``.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000845
846.. class:: Lock()
847
848 A non-recursive lock object: a clone of :class:`threading.Lock`.
849
850.. class:: RLock()
851
852 A recursive lock object: a clone of :class:`threading.RLock`.
853
854.. class:: Semaphore([value])
855
856 A bounded semaphore object: a clone of :class:`threading.Semaphore`.
857
858.. note::
859
Benjamin Peterson910c2ab2008-06-27 23:22:06 +0000860 The :meth:`acquire` method of :class:`BoundedSemaphore`, :class:`Lock`,
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000861 :class:`RLock` and :class:`Semaphore` has a timeout parameter not supported
862 by the equivalents in :mod:`threading`. The signature is
863 ``acquire(block=True, timeout=None)`` with keyword parameters being
864 acceptable. If *block* is ``True`` and *timeout* is not ``None`` then it
865 specifies a timeout in seconds. If *block* is ``False`` then *timeout* is
866 ignored.
Georg Brandlc62ef8b2009-01-03 20:55:06 +0000867
Jesse Nollera280fd72008-11-28 18:22:54 +0000868 Note that on OS/X ``sem_timedwait`` is unsupported, so timeout arguments
869 for these will be ignored.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000870
871.. note::
872
873 If the SIGINT signal generated by Ctrl-C arrives while the main thread is
874 blocked by a call to :meth:`BoundedSemaphore.acquire`, :meth:`Lock.acquire`,
875 :meth:`RLock.acquire`, :meth:`Semaphore.acquire`, :meth:`Condition.acquire`
876 or :meth:`Condition.wait` then the call will be immediately interrupted and
877 :exc:`KeyboardInterrupt` will be raised.
878
879 This differs from the behaviour of :mod:`threading` where SIGINT will be
880 ignored while the equivalent blocking calls are in progress.
881
882
883Shared :mod:`ctypes` Objects
884~~~~~~~~~~~~~~~~~~~~~~~~~~~~
885
886It is possible to create shared objects using shared memory which can be
887inherited by child processes.
888
Jesse Noller6ab22152009-01-18 02:45:38 +0000889.. function:: Value(typecode_or_type, *args[, lock])
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000890
891 Return a :mod:`ctypes` object allocated from shared memory. By default the
892 return value is actually a synchronized wrapper for the object.
893
894 *typecode_or_type* determines the type of the returned object: it is either a
895 ctypes type or a one character typecode of the kind used by the :mod:`array`
896 module. *\*args* is passed on to the constructor for the type.
897
898 If *lock* is ``True`` (the default) then a new lock object is created to
899 synchronize access to the value. If *lock* is a :class:`Lock` or
900 :class:`RLock` object then that will be used to synchronize access to the
901 value. If *lock* is ``False`` then access to the returned object will not be
902 automatically protected by a lock, so it will not necessarily be
903 "process-safe".
904
905 Note that *lock* is a keyword-only argument.
906
907.. function:: Array(typecode_or_type, size_or_initializer, *, lock=True)
908
909 Return a ctypes array allocated from shared memory. By default the return
910 value is actually a synchronized wrapper for the array.
911
912 *typecode_or_type* determines the type of the elements of the returned array:
913 it is either a ctypes type or a one character typecode of the kind used by
914 the :mod:`array` module. If *size_or_initializer* is an integer, then it
915 determines the length of the array, and the array will be initially zeroed.
916 Otherwise, *size_or_initializer* is a sequence which is used to initialize
917 the array and whose length determines the length of the array.
918
919 If *lock* is ``True`` (the default) then a new lock object is created to
920 synchronize access to the value. If *lock* is a :class:`Lock` or
921 :class:`RLock` object then that will be used to synchronize access to the
922 value. If *lock* is ``False`` then access to the returned object will not be
923 automatically protected by a lock, so it will not necessarily be
924 "process-safe".
925
926 Note that *lock* is a keyword only argument.
927
Georg Brandlb053f992008-11-22 08:34:14 +0000928 Note that an array of :data:`ctypes.c_char` has *value* and *raw*
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000929 attributes which allow one to use it to store and retrieve strings.
930
931
932The :mod:`multiprocessing.sharedctypes` module
933>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
934
935.. module:: multiprocessing.sharedctypes
936 :synopsis: Allocate ctypes objects from shared memory.
937
938The :mod:`multiprocessing.sharedctypes` module provides functions for allocating
939:mod:`ctypes` objects from shared memory which can be inherited by child
940processes.
941
942.. note::
943
Benjamin Peterson90f36732008-07-12 20:16:19 +0000944 Although it is possible to store a pointer in shared memory remember that
945 this will refer to a location in the address space of a specific process.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000946 However, the pointer is quite likely to be invalid in the context of a second
947 process and trying to dereference the pointer from the second process may
948 cause a crash.
949
950.. function:: RawArray(typecode_or_type, size_or_initializer)
951
952 Return a ctypes array allocated from shared memory.
953
954 *typecode_or_type* determines the type of the elements of the returned array:
955 it is either a ctypes type or a one character typecode of the kind used by
956 the :mod:`array` module. If *size_or_initializer* is an integer then it
957 determines the length of the array, and the array will be initially zeroed.
958 Otherwise *size_or_initializer* is a sequence which is used to initialize the
959 array and whose length determines the length of the array.
960
961 Note that setting and getting an element is potentially non-atomic -- use
962 :func:`Array` instead to make sure that access is automatically synchronized
963 using a lock.
964
965.. function:: RawValue(typecode_or_type, *args)
966
967 Return a ctypes object allocated from shared memory.
968
969 *typecode_or_type* determines the type of the returned object: it is either a
970 ctypes type or a one character typecode of the kind used by the :mod:`array`
Jesse Noller6ab22152009-01-18 02:45:38 +0000971 module. *\*args* is passed on to the constructor for the type.
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000972
973 Note that setting and getting the value is potentially non-atomic -- use
974 :func:`Value` instead to make sure that access is automatically synchronized
975 using a lock.
976
Georg Brandlb053f992008-11-22 08:34:14 +0000977 Note that an array of :data:`ctypes.c_char` has ``value`` and ``raw``
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000978 attributes which allow one to use it to store and retrieve strings -- see
979 documentation for :mod:`ctypes`.
980
Jesse Noller6ab22152009-01-18 02:45:38 +0000981.. function:: Array(typecode_or_type, size_or_initializer, *args[, lock])
Benjamin Peterson190d56e2008-06-11 02:40:25 +0000982
983 The same as :func:`RawArray` except that depending on the value of *lock* a
984 process-safe synchronization wrapper may be returned instead of a raw ctypes
985 array.
986
987 If *lock* is ``True`` (the default) then a new lock object is created to
988 synchronize access to the value. If *lock* is a :class:`Lock` or
989 :class:`RLock` object then that will be used to synchronize access to the
990 value. If *lock* is ``False`` then access to the returned object will not be
991 automatically protected by a lock, so it will not necessarily be
992 "process-safe".
993
994 Note that *lock* is a keyword-only argument.
995
996.. function:: Value(typecode_or_type, *args[, lock])
997
998 The same as :func:`RawValue` except that depending on the value of *lock* a
999 process-safe synchronization wrapper may be returned instead of a raw ctypes
1000 object.
1001
1002 If *lock* is ``True`` (the default) then a new lock object is created to
1003 synchronize access to the value. If *lock* is a :class:`Lock` or
1004 :class:`RLock` object then that will be used to synchronize access to the
1005 value. If *lock* is ``False`` then access to the returned object will not be
1006 automatically protected by a lock, so it will not necessarily be
1007 "process-safe".
1008
1009 Note that *lock* is a keyword-only argument.
1010
1011.. function:: copy(obj)
1012
1013 Return a ctypes object allocated from shared memory which is a copy of the
1014 ctypes object *obj*.
1015
1016.. function:: synchronized(obj[, lock])
1017
1018 Return a process-safe wrapper object for a ctypes object which uses *lock* to
1019 synchronize access. If *lock* is ``None`` (the default) then a
1020 :class:`multiprocessing.RLock` object is created automatically.
1021
1022 A synchronized wrapper will have two methods in addition to those of the
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001023 object it wraps: :meth:`get_obj` returns the wrapped object and
1024 :meth:`get_lock` returns the lock object used for synchronization.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001025
1026 Note that accessing the ctypes object through the wrapper can be a lot slower
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001027 than accessing the raw ctypes object.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001028
1029
1030The table below compares the syntax for creating shared ctypes objects from
1031shared memory with the normal ctypes syntax. (In the table ``MyStruct`` is some
1032subclass of :class:`ctypes.Structure`.)
1033
1034==================== ========================== ===========================
1035ctypes sharedctypes using type sharedctypes using typecode
1036==================== ========================== ===========================
1037c_double(2.4) RawValue(c_double, 2.4) RawValue('d', 2.4)
1038MyStruct(4, 6) RawValue(MyStruct, 4, 6)
1039(c_short * 7)() RawArray(c_short, 7) RawArray('h', 7)
1040(c_int * 3)(9, 2, 8) RawArray(c_int, (9, 2, 8)) RawArray('i', (9, 2, 8))
1041==================== ========================== ===========================
1042
1043
1044Below is an example where a number of ctypes objects are modified by a child
1045process::
1046
1047 from multiprocessing import Process, Lock
1048 from multiprocessing.sharedctypes import Value, Array
1049 from ctypes import Structure, c_double
1050
1051 class Point(Structure):
1052 _fields_ = [('x', c_double), ('y', c_double)]
1053
1054 def modify(n, x, s, A):
1055 n.value **= 2
1056 x.value **= 2
1057 s.value = s.value.upper()
1058 for a in A:
1059 a.x **= 2
1060 a.y **= 2
1061
1062 if __name__ == '__main__':
1063 lock = Lock()
1064
1065 n = Value('i', 7)
1066 x = Value(ctypes.c_double, 1.0/3.0, lock=False)
1067 s = Array('c', 'hello world', lock=lock)
1068 A = Array(Point, [(1.875,-6.25), (-5.75,2.0), (2.375,9.5)], lock=lock)
1069
1070 p = Process(target=modify, args=(n, x, s, A))
1071 p.start()
1072 p.join()
1073
1074 print n.value
1075 print x.value
1076 print s.value
1077 print [(a.x, a.y) for a in A]
1078
1079
1080.. highlightlang:: none
1081
1082The results printed are ::
1083
1084 49
1085 0.1111111111111111
1086 HELLO WORLD
1087 [(3.515625, 39.0625), (33.0625, 4.0), (5.640625, 90.25)]
1088
1089.. highlightlang:: python
1090
1091
1092.. _multiprocessing-managers:
1093
1094Managers
1095~~~~~~~~
1096
1097Managers provide a way to create data which can be shared between different
1098processes. A manager object controls a server process which manages *shared
1099objects*. Other processes can access the shared objects by using proxies.
1100
1101.. function:: multiprocessing.Manager()
1102
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001103 Returns a started :class:`~multiprocessing.managers.SyncManager` object which
1104 can be used for sharing objects between processes. The returned manager
1105 object corresponds to a spawned child process and has methods which will
1106 create shared objects and return corresponding proxies.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001107
1108.. module:: multiprocessing.managers
1109 :synopsis: Share data between process with shared objects.
1110
1111Manager processes will be shutdown as soon as they are garbage collected or
1112their parent process exits. The manager classes are defined in the
1113:mod:`multiprocessing.managers` module:
1114
1115.. class:: BaseManager([address[, authkey]])
1116
1117 Create a BaseManager object.
1118
1119 Once created one should call :meth:`start` or :meth:`serve_forever` to ensure
1120 that the manager object refers to a started manager process.
1121
1122 *address* is the address on which the manager process listens for new
1123 connections. If *address* is ``None`` then an arbitrary one is chosen.
1124
1125 *authkey* is the authentication key which will be used to check the validity
1126 of incoming connections to the server process. If *authkey* is ``None`` then
Benjamin Peterson73641d72008-08-20 14:07:59 +00001127 ``current_process().authkey``. Otherwise *authkey* is used and it
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001128 must be a string.
1129
1130 .. method:: start()
1131
1132 Start a subprocess to start the manager.
1133
Andrew M. Kuchlinga2478d92008-07-14 00:40:55 +00001134 .. method:: serve_forever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001135
1136 Run the server in the current process.
1137
1138 .. method:: from_address(address, authkey)
1139
1140 A class method which creates a manager object referring to a pre-existing
1141 server process which is using the given address and authentication key.
1142
Jesse Nollera280fd72008-11-28 18:22:54 +00001143 .. method:: get_server()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001144
Jesse Nollera280fd72008-11-28 18:22:54 +00001145 Returns a :class:`Server` object which represents the actual server under
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001146 the control of the Manager. The :class:`Server` object supports the
Georg Brandlfc29f272009-01-02 20:25:14 +00001147 :meth:`serve_forever` method:
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001148
Georg Brandlfc29f272009-01-02 20:25:14 +00001149 >>> from multiprocessing.managers import BaseManager
1150 >>> m = BaseManager(address=('', 50000), authkey='abc'))
1151 >>> server = m.get_server()
1152 >>> s.serve_forever()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001153
Georg Brandlfc29f272009-01-02 20:25:14 +00001154 :class:`Server` additionally have an :attr:`address` attribute.
Jesse Nollera280fd72008-11-28 18:22:54 +00001155
1156 .. method:: connect()
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001157
Georg Brandlfc29f272009-01-02 20:25:14 +00001158 Connect a local manager object to a remote manager process:
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001159
Jesse Nollera280fd72008-11-28 18:22:54 +00001160 >>> from multiprocessing.managers import BaseManager
1161 >>> m = BaseManager(address='127.0.0.1', authkey='abc))
1162 >>> m.connect()
1163
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001164 .. method:: shutdown()
1165
1166 Stop the process used by the manager. This is only available if
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001167 :meth:`start` has been used to start the server process.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001168
1169 This can be called multiple times.
1170
1171 .. method:: register(typeid[, callable[, proxytype[, exposed[, method_to_typeid[, create_method]]]]])
1172
1173 A classmethod which can be used for registering a type or callable with
1174 the manager class.
1175
1176 *typeid* is a "type identifier" which is used to identify a particular
1177 type of shared object. This must be a string.
1178
1179 *callable* is a callable used for creating objects for this type
1180 identifier. If a manager instance will be created using the
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001181 :meth:`from_address` classmethod or if the *create_method* argument is
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001182 ``False`` then this can be left as ``None``.
1183
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001184 *proxytype* is a subclass of :class:`BaseProxy` which is used to create
1185 proxies for shared objects with this *typeid*. If ``None`` then a proxy
1186 class is created automatically.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001187
1188 *exposed* is used to specify a sequence of method names which proxies for
1189 this typeid should be allowed to access using
1190 :meth:`BaseProxy._callMethod`. (If *exposed* is ``None`` then
1191 :attr:`proxytype._exposed_` is used instead if it exists.) In the case
1192 where no exposed list is specified, all "public methods" of the shared
1193 object will be accessible. (Here a "public method" means any attribute
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001194 which has a :meth:`__call__` method and whose name does not begin with
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001195 ``'_'``.)
1196
1197 *method_to_typeid* is a mapping used to specify the return type of those
1198 exposed methods which should return a proxy. It maps method names to
1199 typeid strings. (If *method_to_typeid* is ``None`` then
1200 :attr:`proxytype._method_to_typeid_` is used instead if it exists.) If a
1201 method's name is not a key of this mapping or if the mapping is ``None``
1202 then the object returned by the method will be copied by value.
1203
1204 *create_method* determines whether a method should be created with name
1205 *typeid* which can be used to tell the server process to create a new
1206 shared object and return a proxy for it. By default it is ``True``.
1207
1208 :class:`BaseManager` instances also have one read-only property:
1209
1210 .. attribute:: address
1211
1212 The address used by the manager.
1213
1214
1215.. class:: SyncManager
1216
1217 A subclass of :class:`BaseManager` which can be used for the synchronization
1218 of processes. Objects of this type are returned by
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001219 :func:`multiprocessing.Manager`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001220
1221 It also supports creation of shared lists and dictionaries.
1222
1223 .. method:: BoundedSemaphore([value])
1224
1225 Create a shared :class:`threading.BoundedSemaphore` object and return a
1226 proxy for it.
1227
1228 .. method:: Condition([lock])
1229
1230 Create a shared :class:`threading.Condition` object and return a proxy for
1231 it.
1232
1233 If *lock* is supplied then it should be a proxy for a
1234 :class:`threading.Lock` or :class:`threading.RLock` object.
1235
1236 .. method:: Event()
1237
1238 Create a shared :class:`threading.Event` object and return a proxy for it.
1239
1240 .. method:: Lock()
1241
1242 Create a shared :class:`threading.Lock` object and return a proxy for it.
1243
1244 .. method:: Namespace()
1245
1246 Create a shared :class:`Namespace` object and return a proxy for it.
1247
1248 .. method:: Queue([maxsize])
1249
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001250 Create a shared :class:`Queue.Queue` object and return a proxy for it.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001251
1252 .. method:: RLock()
1253
1254 Create a shared :class:`threading.RLock` object and return a proxy for it.
1255
1256 .. method:: Semaphore([value])
1257
1258 Create a shared :class:`threading.Semaphore` object and return a proxy for
1259 it.
1260
1261 .. method:: Array(typecode, sequence)
1262
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001263 Create an array and return a proxy for it.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001264
1265 .. method:: Value(typecode, value)
1266
1267 Create an object with a writable ``value`` attribute and return a proxy
1268 for it.
1269
1270 .. method:: dict()
1271 dict(mapping)
1272 dict(sequence)
1273
1274 Create a shared ``dict`` object and return a proxy for it.
1275
1276 .. method:: list()
1277 list(sequence)
1278
1279 Create a shared ``list`` object and return a proxy for it.
1280
1281
1282Namespace objects
1283>>>>>>>>>>>>>>>>>
1284
1285A namespace object has no public methods, but does have writable attributes.
1286Its representation shows the values of its attributes.
1287
1288However, when using a proxy for a namespace object, an attribute beginning with
1289``'_'`` will be an attribute of the proxy and not an attribute of the referent::
1290
1291 >>> manager = multiprocessing.Manager()
1292 >>> Global = manager.Namespace()
1293 >>> Global.x = 10
1294 >>> Global.y = 'hello'
1295 >>> Global._z = 12.3 # this is an attribute of the proxy
1296 >>> print Global
1297 Namespace(x=10, y='hello')
1298
1299
1300Customized managers
1301>>>>>>>>>>>>>>>>>>>
1302
1303To create one's own manager, one creates a subclass of :class:`BaseManager` and
Georg Brandlfc29f272009-01-02 20:25:14 +00001304use the :meth:`~BaseManager.register` classmethod to register new types or
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001305callables with the manager class. For example::
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001306
1307 from multiprocessing.managers import BaseManager
1308
1309 class MathsClass(object):
1310 def add(self, x, y):
1311 return x + y
1312 def mul(self, x, y):
1313 return x * y
1314
1315 class MyManager(BaseManager):
1316 pass
1317
1318 MyManager.register('Maths', MathsClass)
1319
1320 if __name__ == '__main__':
1321 manager = MyManager()
1322 manager.start()
1323 maths = manager.Maths()
1324 print maths.add(4, 3) # prints 7
1325 print maths.mul(7, 8) # prints 56
1326
1327
1328Using a remote manager
1329>>>>>>>>>>>>>>>>>>>>>>
1330
1331It is possible to run a manager server on one machine and have clients use it
1332from other machines (assuming that the firewalls involved allow it).
1333
1334Running the following commands creates a server for a single shared queue which
1335remote clients can access::
1336
1337 >>> from multiprocessing.managers import BaseManager
1338 >>> import Queue
1339 >>> queue = Queue.Queue()
1340 >>> class QueueManager(BaseManager): pass
1341 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001342 >>> QueueManager.register('get_queue', callable=lambda:queue)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001343 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
Jesse Nollera280fd72008-11-28 18:22:54 +00001344 >>> s = m.get_server()
1345 >>> s.serveForever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001346
1347One client can access the server as follows::
1348
1349 >>> from multiprocessing.managers import BaseManager
1350 >>> class QueueManager(BaseManager): pass
1351 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001352 >>> QueueManager.register('get_queue')
1353 >>> m = QueueManager(address=('foo.bar.org', 50000), authkey='abracadabra')
1354 >>> m.connect()
1355 >>> queue = m.get_queue()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001356 >>> queue.put('hello')
1357
1358Another client can also use it::
1359
1360 >>> from multiprocessing.managers import BaseManager
1361 >>> class QueueManager(BaseManager): pass
1362 ...
1363 >>> QueueManager.register('getQueue')
1364 >>> m = QueueManager.from_address(address=('foo.bar.org', 50000), authkey='abracadabra')
1365 >>> queue = m.getQueue()
1366 >>> queue.get()
1367 'hello'
1368
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001369Local processes can also access that queue, using the code from above on the
Jesse Nollera280fd72008-11-28 18:22:54 +00001370client to access it remotely::
1371
1372 >>> from multiprocessing import Process, Queue
1373 >>> from multiprocessing.managers import BaseManager
1374 >>> class Worker(Process):
1375 ... def __init__(self, q):
1376 ... self.q = q
1377 ... super(Worker, self).__init__()
1378 ... def run(self):
1379 ... self.q.put('local hello')
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001380 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001381 >>> queue = Queue()
1382 >>> w = Worker(queue)
1383 >>> w.start()
1384 >>> class QueueManager(BaseManager): pass
Georg Brandlc62ef8b2009-01-03 20:55:06 +00001385 ...
Jesse Nollera280fd72008-11-28 18:22:54 +00001386 >>> QueueManager.register('get_queue', callable=lambda: queue)
1387 >>> m = QueueManager(address=('', 50000), authkey='abracadabra')
1388 >>> s = m.get_server()
1389 >>> s.serve_forever()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001390
1391Proxy Objects
1392~~~~~~~~~~~~~
1393
1394A proxy is an object which *refers* to a shared object which lives (presumably)
1395in a different process. The shared object is said to be the *referent* of the
1396proxy. Multiple proxy objects may have the same referent.
1397
1398A proxy object has methods which invoke corresponding methods of its referent
1399(although not every method of the referent will necessarily be available through
1400the proxy). A proxy can usually be used in most of the same ways that its
1401referent can::
1402
1403 >>> from multiprocessing import Manager
1404 >>> manager = Manager()
1405 >>> l = manager.list([i*i for i in range(10)])
1406 >>> print l
1407 [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
1408 >>> print repr(l)
1409 <ListProxy object, typeid 'list' at 0xb799974c>
1410 >>> l[4]
1411 16
1412 >>> l[2:5]
1413 [4, 9, 16]
1414
1415Notice that applying :func:`str` to a proxy will return the representation of
1416the referent, whereas applying :func:`repr` will return the representation of
1417the proxy.
1418
1419An important feature of proxy objects is that they are picklable so they can be
1420passed between processes. Note, however, that if a proxy is sent to the
1421corresponding manager's process then unpickling it will produce the referent
1422itself. This means, for example, that one shared object can contain a second::
1423
1424 >>> a = manager.list()
1425 >>> b = manager.list()
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001426 >>> a.append(b) # referent of a now contains referent of b
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001427 >>> print a, b
1428 [[]] []
1429 >>> b.append('hello')
1430 >>> print a, b
1431 [['hello']] ['hello']
1432
1433.. note::
1434
1435 The proxy types in :mod:`multiprocessing` do nothing to support comparisons
1436 by value. So, for instance, ::
1437
1438 manager.list([1,2,3]) == [1,2,3]
1439
1440 will return ``False``. One should just use a copy of the referent instead
1441 when making comparisons.
1442
1443.. class:: BaseProxy
1444
1445 Proxy objects are instances of subclasses of :class:`BaseProxy`.
1446
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001447 .. method:: _callmethod(methodname[, args[, kwds]])
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001448
1449 Call and return the result of a method of the proxy's referent.
1450
1451 If ``proxy`` is a proxy whose referent is ``obj`` then the expression ::
1452
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001453 proxy._callmethod(methodname, args, kwds)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001454
1455 will evaluate the expression ::
1456
1457 getattr(obj, methodname)(*args, **kwds)
1458
1459 in the manager's process.
1460
1461 The returned value will be a copy of the result of the call or a proxy to
1462 a new shared object -- see documentation for the *method_to_typeid*
1463 argument of :meth:`BaseManager.register`.
1464
1465 If an exception is raised by the call, then then is re-raised by
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001466 :meth:`_callmethod`. If some other exception is raised in the manager's
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001467 process then this is converted into a :exc:`RemoteError` exception and is
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001468 raised by :meth:`_callmethod`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001469
1470 Note in particular that an exception will be raised if *methodname* has
1471 not been *exposed*
1472
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001473 An example of the usage of :meth:`_callmethod`::
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001474
1475 >>> l = manager.list(range(10))
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001476 >>> l._callmethod('__len__')
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001477 10
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001478 >>> l._callmethod('__getslice__', (2, 7)) # equiv to `l[2:7]`
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001479 [2, 3, 4, 5, 6]
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001480 >>> l._callmethod('__getitem__', (20,)) # equiv to `l[20]`
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001481 Traceback (most recent call last):
1482 ...
1483 IndexError: list index out of range
1484
Benjamin Peterson2b97b712008-12-19 02:31:35 +00001485 .. method:: _getvalue()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001486
1487 Return a copy of the referent.
1488
1489 If the referent is unpicklable then this will raise an exception.
1490
1491 .. method:: __repr__
1492
1493 Return a representation of the proxy object.
1494
1495 .. method:: __str__
1496
1497 Return the representation of the referent.
1498
1499
1500Cleanup
1501>>>>>>>
1502
1503A proxy object uses a weakref callback so that when it gets garbage collected it
1504deregisters itself from the manager which owns its referent.
1505
1506A shared object gets deleted from the manager process when there are no longer
1507any proxies referring to it.
1508
1509
1510Process Pools
1511~~~~~~~~~~~~~
1512
1513.. module:: multiprocessing.pool
1514 :synopsis: Create pools of processes.
1515
1516One can create a pool of processes which will carry out tasks submitted to it
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001517with the :class:`Pool` class.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001518
1519.. class:: multiprocessing.Pool([processes[, initializer[, initargs]]])
1520
1521 A process pool object which controls a pool of worker processes to which jobs
1522 can be submitted. It supports asynchronous results with timeouts and
1523 callbacks and has a parallel map implementation.
1524
1525 *processes* is the number of worker processes to use. If *processes* is
1526 ``None`` then the number returned by :func:`cpu_count` is used. If
1527 *initializer* is not ``None`` then each worker process will call
1528 ``initializer(*initargs)`` when it starts.
1529
1530 .. method:: apply(func[, args[, kwds]])
1531
1532 Equivalent of the :func:`apply` builtin function. It blocks till the
Jesse Noller403c6632009-01-22 21:53:22 +00001533 result is ready. Given this blocks - :meth:`apply_async` is better suited
1534 for performing work in parallel. Additionally, the passed
1535 in function is only executed in one of the workers of the pool.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001536
1537 .. method:: apply_async(func[, args[, kwds[, callback]]])
1538
1539 A variant of the :meth:`apply` method which returns a result object.
1540
1541 If *callback* is specified then it should be a callable which accepts a
1542 single argument. When the result becomes ready *callback* is applied to
1543 it (unless the call failed). *callback* should complete immediately since
1544 otherwise the thread which handles the results will get blocked.
1545
1546 .. method:: map(func, iterable[, chunksize])
1547
1548 A parallel equivalent of the :func:`map` builtin function. It blocks till
1549 the result is ready.
1550
1551 This method chops the iterable into a number of chunks which it submits to
1552 the process pool as separate tasks. The (approximate) size of these
1553 chunks can be specified by setting *chunksize* to a positive integer.
1554
1555 .. method:: map_async(func, iterable[, chunksize[, callback]])
1556
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001557 A variant of the :meth:`map` method which returns a result object.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001558
1559 If *callback* is specified then it should be a callable which accepts a
1560 single argument. When the result becomes ready *callback* is applied to
1561 it (unless the call failed). *callback* should complete immediately since
1562 otherwise the thread which handles the results will get blocked.
1563
1564 .. method:: imap(func, iterable[, chunksize])
1565
1566 An equivalent of :func:`itertools.imap`.
1567
1568 The *chunksize* argument is the same as the one used by the :meth:`.map`
1569 method. For very long iterables using a large value for *chunksize* can
1570 make make the job complete **much** faster than using the default value of
1571 ``1``.
1572
1573 Also if *chunksize* is ``1`` then the :meth:`next` method of the iterator
1574 returned by the :meth:`imap` method has an optional *timeout* parameter:
1575 ``next(timeout)`` will raise :exc:`multiprocessing.TimeoutError` if the
1576 result cannot be returned within *timeout* seconds.
1577
1578 .. method:: imap_unordered(func, iterable[, chunksize])
1579
1580 The same as :meth:`imap` except that the ordering of the results from the
1581 returned iterator should be considered arbitrary. (Only when there is
1582 only one worker process is the order guaranteed to be "correct".)
1583
1584 .. method:: close()
1585
1586 Prevents any more tasks from being submitted to the pool. Once all the
1587 tasks have been completed the worker processes will exit.
1588
1589 .. method:: terminate()
1590
1591 Stops the worker processes immediately without completing outstanding
1592 work. When the pool object is garbage collected :meth:`terminate` will be
1593 called immediately.
1594
1595 .. method:: join()
1596
1597 Wait for the worker processes to exit. One must call :meth:`close` or
1598 :meth:`terminate` before using :meth:`join`.
1599
1600
1601.. class:: AsyncResult
1602
1603 The class of the result returned by :meth:`Pool.apply_async` and
1604 :meth:`Pool.map_async`.
1605
Jesse Nollera280fd72008-11-28 18:22:54 +00001606 .. method:: get([timeout])
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001607
1608 Return the result when it arrives. If *timeout* is not ``None`` and the
1609 result does not arrive within *timeout* seconds then
1610 :exc:`multiprocessing.TimeoutError` is raised. If the remote call raised
1611 an exception then that exception will be reraised by :meth:`get`.
1612
1613 .. method:: wait([timeout])
1614
1615 Wait until the result is available or until *timeout* seconds pass.
1616
1617 .. method:: ready()
1618
1619 Return whether the call has completed.
1620
1621 .. method:: successful()
1622
1623 Return whether the call completed without raising an exception. Will
1624 raise :exc:`AssertionError` if the result is not ready.
1625
1626The following example demonstrates the use of a pool::
1627
1628 from multiprocessing import Pool
1629
1630 def f(x):
1631 return x*x
1632
1633 if __name__ == '__main__':
1634 pool = Pool(processes=4) # start 4 worker processes
1635
Jesse Nollera280fd72008-11-28 18:22:54 +00001636 result = pool.apply_async(f, (10,)) # evaluate "f(10)" asynchronously
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001637 print result.get(timeout=1) # prints "100" unless your computer is *very* slow
1638
1639 print pool.map(f, range(10)) # prints "[0, 1, 4,..., 81]"
1640
1641 it = pool.imap(f, range(10))
1642 print it.next() # prints "0"
1643 print it.next() # prints "1"
1644 print it.next(timeout=1) # prints "4" unless your computer is *very* slow
1645
1646 import time
Jesse Nollera280fd72008-11-28 18:22:54 +00001647 result = pool.apply_async(time.sleep, (10,))
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001648 print result.get(timeout=1) # raises TimeoutError
1649
1650
1651.. _multiprocessing-listeners-clients:
1652
1653Listeners and Clients
1654~~~~~~~~~~~~~~~~~~~~~
1655
1656.. module:: multiprocessing.connection
1657 :synopsis: API for dealing with sockets.
1658
1659Usually message passing between processes is done using queues or by using
1660:class:`Connection` objects returned by :func:`Pipe`.
1661
1662However, the :mod:`multiprocessing.connection` module allows some extra
1663flexibility. It basically gives a high level message oriented API for dealing
1664with sockets or Windows named pipes, and also has support for *digest
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001665authentication* using the :mod:`hmac` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001666
1667
1668.. function:: deliver_challenge(connection, authkey)
1669
1670 Send a randomly generated message to the other end of the connection and wait
1671 for a reply.
1672
1673 If the reply matches the digest of the message using *authkey* as the key
1674 then a welcome message is sent to the other end of the connection. Otherwise
1675 :exc:`AuthenticationError` is raised.
1676
1677.. function:: answerChallenge(connection, authkey)
1678
1679 Receive a message, calculate the digest of the message using *authkey* as the
1680 key, and then send the digest back.
1681
1682 If a welcome message is not received, then :exc:`AuthenticationError` is
1683 raised.
1684
1685.. function:: Client(address[, family[, authenticate[, authkey]]])
1686
1687 Attempt to set up a connection to the listener which is using address
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001688 *address*, returning a :class:`~multiprocessing.Connection`.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001689
1690 The type of the connection is determined by *family* argument, but this can
1691 generally be omitted since it can usually be inferred from the format of
1692 *address*. (See :ref:`multiprocessing-address-formats`)
1693
1694 If *authentication* is ``True`` or *authkey* is a string then digest
1695 authentication is used. The key used for authentication will be either
Benjamin Peterson73641d72008-08-20 14:07:59 +00001696 *authkey* or ``current_process().authkey)`` if *authkey* is ``None``.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001697 If authentication fails then :exc:`AuthenticationError` is raised. See
1698 :ref:`multiprocessing-auth-keys`.
1699
1700.. class:: Listener([address[, family[, backlog[, authenticate[, authkey]]]]])
1701
1702 A wrapper for a bound socket or Windows named pipe which is 'listening' for
1703 connections.
1704
1705 *address* is the address to be used by the bound socket or named pipe of the
1706 listener object.
1707
1708 *family* is the type of socket (or named pipe) to use. This can be one of
1709 the strings ``'AF_INET'`` (for a TCP socket), ``'AF_UNIX'`` (for a Unix
1710 domain socket) or ``'AF_PIPE'`` (for a Windows named pipe). Of these only
1711 the first is guaranteed to be available. If *family* is ``None`` then the
1712 family is inferred from the format of *address*. If *address* is also
1713 ``None`` then a default is chosen. This default is the family which is
1714 assumed to be the fastest available. See
1715 :ref:`multiprocessing-address-formats`. Note that if *family* is
1716 ``'AF_UNIX'`` and address is ``None`` then the socket will be created in a
1717 private temporary directory created using :func:`tempfile.mkstemp`.
1718
1719 If the listener object uses a socket then *backlog* (1 by default) is passed
1720 to the :meth:`listen` method of the socket once it has been bound.
1721
1722 If *authenticate* is ``True`` (``False`` by default) or *authkey* is not
1723 ``None`` then digest authentication is used.
1724
1725 If *authkey* is a string then it will be used as the authentication key;
1726 otherwise it must be *None*.
1727
1728 If *authkey* is ``None`` and *authenticate* is ``True`` then
Benjamin Peterson73641d72008-08-20 14:07:59 +00001729 ``current_process().authkey`` is used as the authentication key. If
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001730 *authkey* is ``None`` and *authentication* is ``False`` then no
1731 authentication is done. If authentication fails then
1732 :exc:`AuthenticationError` is raised. See :ref:`multiprocessing-auth-keys`.
1733
1734 .. method:: accept()
1735
1736 Accept a connection on the bound socket or named pipe of the listener
1737 object and return a :class:`Connection` object. If authentication is
1738 attempted and fails, then :exc:`AuthenticationError` is raised.
1739
1740 .. method:: close()
1741
1742 Close the bound socket or named pipe of the listener object. This is
1743 called automatically when the listener is garbage collected. However it
1744 is advisable to call it explicitly.
1745
1746 Listener objects have the following read-only properties:
1747
1748 .. attribute:: address
1749
1750 The address which is being used by the Listener object.
1751
1752 .. attribute:: last_accepted
1753
1754 The address from which the last accepted connection came. If this is
1755 unavailable then it is ``None``.
1756
1757
1758The module defines two exceptions:
1759
1760.. exception:: AuthenticationError
1761
1762 Exception raised when there is an authentication error.
1763
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001764
1765**Examples**
1766
1767The following server code creates a listener which uses ``'secret password'`` as
1768an authentication key. It then waits for a connection and sends some data to
1769the client::
1770
1771 from multiprocessing.connection import Listener
1772 from array import array
1773
1774 address = ('localhost', 6000) # family is deduced to be 'AF_INET'
1775 listener = Listener(address, authkey='secret password')
1776
1777 conn = listener.accept()
1778 print 'connection accepted from', listener.last_accepted
1779
1780 conn.send([2.25, None, 'junk', float])
1781
1782 conn.send_bytes('hello')
1783
1784 conn.send_bytes(array('i', [42, 1729]))
1785
1786 conn.close()
1787 listener.close()
1788
1789The following code connects to the server and receives some data from the
1790server::
1791
1792 from multiprocessing.connection import Client
1793 from array import array
1794
1795 address = ('localhost', 6000)
1796 conn = Client(address, authkey='secret password')
1797
1798 print conn.recv() # => [2.25, None, 'junk', float]
1799
1800 print conn.recv_bytes() # => 'hello'
1801
1802 arr = array('i', [0, 0, 0, 0, 0])
1803 print conn.recv_bytes_into(arr) # => 8
1804 print arr # => array('i', [42, 1729, 0, 0, 0])
1805
1806 conn.close()
1807
1808
1809.. _multiprocessing-address-formats:
1810
1811Address Formats
1812>>>>>>>>>>>>>>>
1813
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001814* An ``'AF_INET'`` address is a tuple of the form ``(hostname, port)`` where
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001815 *hostname* is a string and *port* is an integer.
1816
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001817* An ``'AF_UNIX'`` address is a string representing a filename on the
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001818 filesystem.
1819
1820* An ``'AF_PIPE'`` address is a string of the form
Georg Brandl6b28f392008-12-27 19:06:04 +00001821 :samp:`r'\\\\.\\pipe\\{PipeName}'`. To use :func:`Client` to connect to a named
Georg Brandlfc29f272009-01-02 20:25:14 +00001822 pipe on a remote computer called *ServerName* one should use an address of the
Georg Brandldd7e3132009-01-04 10:24:09 +00001823 form :samp:`r'\\\\{ServerName}\\pipe\\{PipeName}'` instead.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001824
1825Note that any string beginning with two backslashes is assumed by default to be
1826an ``'AF_PIPE'`` address rather than an ``'AF_UNIX'`` address.
1827
1828
1829.. _multiprocessing-auth-keys:
1830
1831Authentication keys
1832~~~~~~~~~~~~~~~~~~~
1833
1834When one uses :meth:`Connection.recv`, the data received is automatically
1835unpickled. Unfortunately unpickling data from an untrusted source is a security
1836risk. Therefore :class:`Listener` and :func:`Client` use the :mod:`hmac` module
1837to provide digest authentication.
1838
1839An authentication key is a string which can be thought of as a password: once a
1840connection is established both ends will demand proof that the other knows the
1841authentication key. (Demonstrating that both ends are using the same key does
1842**not** involve sending the key over the connection.)
1843
1844If authentication is requested but do authentication key is specified then the
Benjamin Peterson73641d72008-08-20 14:07:59 +00001845return value of ``current_process().authkey`` is used (see
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001846:class:`~multiprocessing.Process`). This value will automatically inherited by
1847any :class:`~multiprocessing.Process` object that the current process creates.
1848This means that (by default) all processes of a multi-process program will share
1849a single authentication key which can be used when setting up connections
1850between the themselves.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001851
1852Suitable authentication keys can also be generated by using :func:`os.urandom`.
1853
1854
1855Logging
1856~~~~~~~
1857
1858Some support for logging is available. Note, however, that the :mod:`logging`
1859package does not use process shared locks so it is possible (depending on the
1860handler type) for messages from different processes to get mixed up.
1861
1862.. currentmodule:: multiprocessing
1863.. function:: get_logger()
1864
1865 Returns the logger used by :mod:`multiprocessing`. If necessary, a new one
1866 will be created.
1867
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001868 When first created the logger has level :data:`logging.NOTSET` and no
1869 default handler. Messages sent to this logger will not by default propagate
1870 to the root logger.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001871
1872 Note that on Windows child processes will only inherit the level of the
1873 parent process's logger -- any other customization of the logger will not be
1874 inherited.
1875
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001876.. currentmodule:: multiprocessing
1877.. function:: log_to_stderr()
1878
1879 This function performs a call to :func:`get_logger` but in addition to
1880 returning the logger created by get_logger, it adds a handler which sends
1881 output to :data:`sys.stderr` using format
1882 ``'[%(levelname)s/%(processName)s] %(message)s'``.
1883
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001884Below is an example session with logging turned on::
1885
Georg Brandl19cc9442008-10-16 21:36:39 +00001886 >>> import multiprocessing, logging
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001887 >>> logger = multiprocessing.log_to_stderr()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001888 >>> logger.setLevel(logging.INFO)
1889 >>> logger.warning('doomed')
1890 [WARNING/MainProcess] doomed
Georg Brandl19cc9442008-10-16 21:36:39 +00001891 >>> m = multiprocessing.Manager()
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001892 [INFO/SyncManager-1] child process calling self.run()
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001893 [INFO/SyncManager-1] created temp directory /.../pymp-Wh47O_
1894 [INFO/SyncManager-1] manager serving at '/.../listener-lWsERs'
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001895 >>> del m
1896 [INFO/MainProcess] sending shutdown message to manager
1897 [INFO/SyncManager-1] manager exiting with exitcode 0
1898
Jesse Nollerb5a4b0a2009-01-25 03:36:13 +00001899In addition to having these two logging functions, the multiprocessing also
1900exposes two additional logging level attributes. These are :const:`SUBWARNING`
1901and :const:`SUBDEBUG`. The table below illustrates where theses fit in the
1902normal level hierarchy.
1903
1904+----------------+----------------+
1905| Level | Numeric value |
1906+================+================+
1907| ``SUBWARNING`` | 25 |
1908+----------------+----------------+
1909| ``SUBDEBUG`` | 5 |
1910+----------------+----------------+
1911
1912For a full table of logging levels, see the :mod:`logging` module.
1913
1914These additional logging levels are used primarily for certain debug messages
1915within the multiprocessing module. Below is the same example as above, except
1916with :const:`SUBDEBUG` enabled::
1917
1918 >>> import multiprocessing, logging
1919 >>> logger = multiprocessing.log_to_stderr()
1920 >>> logger.setLevel(multiprocessing.SUBDEBUG)
1921 >>> logger.warning('doomed')
1922 [WARNING/MainProcess] doomed
1923 >>> m = multiprocessing.Manager()
1924 [INFO/SyncManager-1] child process calling self.run()
1925 [INFO/SyncManager-1] created temp directory /.../pymp-djGBXN
1926 [INFO/SyncManager-1] manager serving at '/.../pymp-djGBXN/listener-knBYGe'
1927 >>> del m
1928 [SUBDEBUG/MainProcess] finalizer calling ...
1929 [INFO/MainProcess] sending shutdown message to manager
1930 [DEBUG/SyncManager-1] manager received shutdown message
1931 [SUBDEBUG/SyncManager-1] calling <Finalize object, callback=unlink, ...
1932 [SUBDEBUG/SyncManager-1] finalizer calling <built-in function unlink> ...
1933 [SUBDEBUG/SyncManager-1] calling <Finalize object, dead>
1934 [SUBDEBUG/SyncManager-1] finalizer calling <function rmtree at 0x5aa730> ...
1935 [INFO/SyncManager-1] manager exiting with exitcode 0
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001936
1937The :mod:`multiprocessing.dummy` module
1938~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1939
1940.. module:: multiprocessing.dummy
1941 :synopsis: Dumb wrapper around threading.
1942
1943:mod:`multiprocessing.dummy` replicates the API of :mod:`multiprocessing` but is
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00001944no more than a wrapper around the :mod:`threading` module.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001945
1946
1947.. _multiprocessing-programming:
1948
1949Programming guidelines
1950----------------------
1951
1952There are certain guidelines and idioms which should be adhered to when using
1953:mod:`multiprocessing`.
1954
1955
1956All platforms
1957~~~~~~~~~~~~~
1958
1959Avoid shared state
1960
1961 As far as possible one should try to avoid shifting large amounts of data
1962 between processes.
1963
1964 It is probably best to stick to using queues or pipes for communication
1965 between processes rather than using the lower level synchronization
1966 primitives from the :mod:`threading` module.
1967
1968Picklability
1969
1970 Ensure that the arguments to the methods of proxies are picklable.
1971
1972Thread safety of proxies
1973
1974 Do not use a proxy object from more than one thread unless you protect it
1975 with a lock.
1976
1977 (There is never a problem with different processes using the *same* proxy.)
1978
1979Joining zombie processes
1980
1981 On Unix when a process finishes but has not been joined it becomes a zombie.
1982 There should never be very many because each time a new process starts (or
1983 :func:`active_children` is called) all completed processes which have not
1984 yet been joined will be joined. Also calling a finished process's
1985 :meth:`Process.is_alive` will join the process. Even so it is probably good
1986 practice to explicitly join all the processes that you start.
1987
1988Better to inherit than pickle/unpickle
1989
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00001990 On Windows many types from :mod:`multiprocessing` need to be picklable so
Benjamin Peterson190d56e2008-06-11 02:40:25 +00001991 that child processes can use them. However, one should generally avoid
1992 sending shared objects to other processes using pipes or queues. Instead
1993 you should arrange the program so that a process which need access to a
1994 shared resource created elsewhere can inherit it from an ancestor process.
1995
1996Avoid terminating processes
1997
1998 Using the :meth:`Process.terminate` method to stop a process is liable to
1999 cause any shared resources (such as locks, semaphores, pipes and queues)
2000 currently being used by the process to become broken or unavailable to other
2001 processes.
2002
2003 Therefore it is probably best to only consider using
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002004 :meth:`Process.terminate` on processes which never use any shared resources.
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002005
2006Joining processes that use queues
2007
2008 Bear in mind that a process that has put items in a queue will wait before
2009 terminating until all the buffered items are fed by the "feeder" thread to
2010 the underlying pipe. (The child process can call the
Jesse Nollerd5ff5b22008-09-06 01:20:11 +00002011 :meth:`Queue.cancel_join_thread` method of the queue to avoid this behaviour.)
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002012
2013 This means that whenever you use a queue you need to make sure that all
2014 items which have been put on the queue will eventually be removed before the
2015 process is joined. Otherwise you cannot be sure that processes which have
2016 put items on the queue will terminate. Remember also that non-daemonic
2017 processes will be automatically be joined.
2018
2019 An example which will deadlock is the following::
2020
2021 from multiprocessing import Process, Queue
2022
2023 def f(q):
2024 q.put('X' * 1000000)
2025
2026 if __name__ == '__main__':
2027 queue = Queue()
2028 p = Process(target=f, args=(queue,))
2029 p.start()
2030 p.join() # this deadlocks
2031 obj = queue.get()
2032
2033 A fix here would be to swap the last two lines round (or simply remove the
2034 ``p.join()`` line).
2035
Andrew M. Kuchlingbe504f12008-06-19 19:48:42 +00002036Explicitly pass resources to child processes
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002037
2038 On Unix a child process can make use of a shared resource created in a
2039 parent process using a global resource. However, it is better to pass the
2040 object as an argument to the constructor for the child process.
2041
2042 Apart from making the code (potentially) compatible with Windows this also
2043 ensures that as long as the child process is still alive the object will not
2044 be garbage collected in the parent process. This might be important if some
2045 resource is freed when the object is garbage collected in the parent
2046 process.
2047
2048 So for instance ::
2049
2050 from multiprocessing import Process, Lock
2051
2052 def f():
2053 ... do something using "lock" ...
2054
2055 if __name__ == '__main__':
2056 lock = Lock()
2057 for i in range(10):
2058 Process(target=f).start()
2059
2060 should be rewritten as ::
2061
2062 from multiprocessing import Process, Lock
2063
2064 def f(l):
2065 ... do something using "l" ...
2066
2067 if __name__ == '__main__':
2068 lock = Lock()
2069 for i in range(10):
2070 Process(target=f, args=(lock,)).start()
2071
2072
2073Windows
2074~~~~~~~
2075
2076Since Windows lacks :func:`os.fork` it has a few extra restrictions:
2077
2078More picklability
2079
2080 Ensure that all arguments to :meth:`Process.__init__` are picklable. This
2081 means, in particular, that bound or unbound methods cannot be used directly
2082 as the ``target`` argument on Windows --- just define a function and use
2083 that instead.
2084
2085 Also, if you subclass :class:`Process` then make sure that instances will be
2086 picklable when the :meth:`Process.start` method is called.
2087
2088Global variables
2089
2090 Bear in mind that if code run in a child process tries to access a global
2091 variable, then the value it sees (if any) may not be the same as the value
2092 in the parent process at the time that :meth:`Process.start` was called.
2093
2094 However, global variables which are just module level constants cause no
2095 problems.
2096
2097Safe importing of main module
2098
2099 Make sure that the main module can be safely imported by a new Python
2100 interpreter without causing unintended side effects (such a starting a new
2101 process).
2102
2103 For example, under Windows running the following module would fail with a
2104 :exc:`RuntimeError`::
2105
2106 from multiprocessing import Process
2107
2108 def foo():
2109 print 'hello'
2110
2111 p = Process(target=foo)
2112 p.start()
2113
2114 Instead one should protect the "entry point" of the program by using ``if
2115 __name__ == '__main__':`` as follows::
2116
2117 from multiprocessing import Process, freeze_support
2118
2119 def foo():
2120 print 'hello'
2121
2122 if __name__ == '__main__':
2123 freeze_support()
2124 p = Process(target=foo)
2125 p.start()
2126
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002127 (The ``freeze_support()`` line can be omitted if the program will be run
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002128 normally instead of frozen.)
2129
2130 This allows the newly spawned Python interpreter to safely import the module
2131 and then run the module's ``foo()`` function.
2132
2133 Similar restrictions apply if a pool or manager is created in the main
2134 module.
2135
2136
2137.. _multiprocessing-examples:
2138
2139Examples
2140--------
2141
2142Demonstration of how to create and use customized managers and proxies:
2143
2144.. literalinclude:: ../includes/mp_newtype.py
2145
2146
2147Using :class:`Pool`:
2148
2149.. literalinclude:: ../includes/mp_pool.py
2150
2151
2152Synchronization types like locks, conditions and queues:
2153
2154.. literalinclude:: ../includes/mp_synchronize.py
2155
2156
2157An showing how to use queues to feed tasks to a collection of worker process and
2158collect the results:
2159
2160.. literalinclude:: ../includes/mp_workers.py
2161
2162
2163An example of how a pool of worker processes can each run a
2164:class:`SimpleHTTPServer.HttpServer` instance while sharing a single listening
2165socket.
2166
2167.. literalinclude:: ../includes/mp_webserver.py
2168
2169
2170Some simple benchmarks comparing :mod:`multiprocessing` with :mod:`threading`:
2171
2172.. literalinclude:: ../includes/mp_benchmarks.py
2173
2174An example/demo of how to use the :class:`managers.SyncManager`, :class:`Process`
Georg Brandlc62ef8b2009-01-03 20:55:06 +00002175and others to build a system which can distribute processes and work via a
Benjamin Peterson190d56e2008-06-11 02:40:25 +00002176distributed queue to a "cluster" of machines on a network, accessible via SSH.
2177You will need to have private key authentication for all hosts configured for
2178this to work.
2179
Benjamin Peterson910c2ab2008-06-27 23:22:06 +00002180.. literalinclude:: ../includes/mp_distributing.py