blob: 0ee920bc6fef8e33ee8f502e776d3ef5c81b141a [file] [log] [blame]
Alexandre Vassalottif260e442008-05-11 19:59:59 +00001:mod:`queue` --- A synchronized queue class
Georg Brandl116aa622007-08-15 14:28:22 +00002===========================================
3
Alexandre Vassalottif260e442008-05-11 19:59:59 +00004.. module:: queue
Georg Brandl116aa622007-08-15 14:28:22 +00005 :synopsis: A synchronized queue class.
6
7
Alexandre Vassalottif260e442008-05-11 19:59:59 +00008The :mod:`queue` module implements multi-producer, multi-consumer queues.
Thomas Wouters89d996e2007-09-08 17:39:28 +00009It is especially useful in threaded programming when information must be
Georg Brandl116aa622007-08-15 14:28:22 +000010exchanged safely between multiple threads. The :class:`Queue` class in this
11module implements all the required locking semantics. It depends on the
Thomas Wouters89d996e2007-09-08 17:39:28 +000012availability of thread support in Python; see the :mod:`threading`
13module.
Georg Brandl116aa622007-08-15 14:28:22 +000014
Raymond Hettinger35641462008-01-17 00:13:27 +000015Implements three types of queue whose only difference is the order that
16the entries are retrieved. In a FIFO queue, the first tasks added are
17the first retrieved. In a LIFO queue, the most recently added entry is
18the first retrieved (operating like a stack). With a priority queue,
19the entries are kept sorted (using the :mod:`heapq` module) and the
20lowest valued entry is retrieved first.
Georg Brandl116aa622007-08-15 14:28:22 +000021
Alexandre Vassalottif260e442008-05-11 19:59:59 +000022The :mod:`queue` module defines the following classes and exceptions:
Georg Brandl116aa622007-08-15 14:28:22 +000023
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000024.. class:: Queue(maxsize=0)
Georg Brandl116aa622007-08-15 14:28:22 +000025
Raymond Hettinger35641462008-01-17 00:13:27 +000026 Constructor for a FIFO queue. *maxsize* is an integer that sets the upperbound
Georg Brandl116aa622007-08-15 14:28:22 +000027 limit on the number of items that can be placed in the queue. Insertion will
28 block once this size has been reached, until queue items are consumed. If
29 *maxsize* is less than or equal to zero, the queue size is infinite.
30
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000031.. class:: LifoQueue(maxsize=0)
Raymond Hettinger35641462008-01-17 00:13:27 +000032
33 Constructor for a LIFO queue. *maxsize* is an integer that sets the upperbound
34 limit on the number of items that can be placed in the queue. Insertion will
35 block once this size has been reached, until queue items are consumed. If
36 *maxsize* is less than or equal to zero, the queue size is infinite.
37
Christian Heimes679db4a2008-01-18 09:56:22 +000038
Andrew M. Kuchling2b600e52010-02-26 13:35:56 +000039.. class:: PriorityQueue(maxsize=0)
Raymond Hettinger35641462008-01-17 00:13:27 +000040
41 Constructor for a priority queue. *maxsize* is an integer that sets the upperbound
42 limit on the number of items that can be placed in the queue. Insertion will
43 block once this size has been reached, until queue items are consumed. If
44 *maxsize* is less than or equal to zero, the queue size is infinite.
45
46 The lowest valued entries are retrieved first (the lowest valued entry is the
47 one returned by ``sorted(list(entries))[0]``). A typical pattern for entries
48 is a tuple in the form: ``(priority_number, data)``.
Georg Brandl116aa622007-08-15 14:28:22 +000049
Christian Heimes679db4a2008-01-18 09:56:22 +000050
Georg Brandl116aa622007-08-15 14:28:22 +000051.. exception:: Empty
52
53 Exception raised when non-blocking :meth:`get` (or :meth:`get_nowait`) is called
54 on a :class:`Queue` object which is empty.
55
56
57.. exception:: Full
58
59 Exception raised when non-blocking :meth:`put` (or :meth:`put_nowait`) is called
60 on a :class:`Queue` object which is full.
61
Raymond Hettinger1d7f4b22009-03-09 12:54:35 +000062.. seealso::
63
64 :class:`collections.deque` is an alternative implementation of unbounded
65 queues with fast atomic :func:`append` and :func:`popleft` operations that
66 do not require locking.
67
Georg Brandl116aa622007-08-15 14:28:22 +000068
69.. _queueobjects:
70
71Queue Objects
72-------------
73
Christian Heimes292d3512008-02-03 16:51:08 +000074Queue objects (:class:`Queue`, :class:`LifoQueue`, or :class:`PriorityQueue`)
Georg Brandl48310cd2009-01-03 21:18:54 +000075provide the public methods described below.
Georg Brandl116aa622007-08-15 14:28:22 +000076
77
78.. method:: Queue.qsize()
79
Guido van Rossum7736b5b2008-01-15 21:44:53 +000080 Return the approximate size of the queue. Note, qsize() > 0 doesn't
81 guarantee that a subsequent get() will not block, nor will qsize() < maxsize
82 guarantee that put() will not block.
Georg Brandl116aa622007-08-15 14:28:22 +000083
84
Raymond Hettinger47aa9892009-03-07 14:07:37 +000085.. method:: Queue.empty()
86
87 Return ``True`` if the queue is empty, ``False`` otherwise. If empty()
88 returns ``True`` it doesn't guarantee that a subsequent call to put()
89 will not block. Similarly, if empty() returns ``False`` it doesn't
90 guarantee that a subsequent call to get() will not block.
91
92
93.. method:: Queue.full()
94
95 Return ``True`` if the queue is full, ``False`` otherwise. If full()
96 returns ``True`` it doesn't guarantee that a subsequent call to get()
97 will not block. Similarly, if full() returns ``False`` it doesn't
98 guarantee that a subsequent call to put() will not block.
99
100
Georg Brandl18244152009-09-02 20:34:52 +0000101.. method:: Queue.put(item, block=True, timeout=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000102
103 Put *item* into the queue. If optional args *block* is true and *timeout* is
104 None (the default), block if necessary until a free slot is available. If
105 *timeout* is a positive number, it blocks at most *timeout* seconds and raises
106 the :exc:`Full` exception if no free slot was available within that time.
107 Otherwise (*block* is false), put an item on the queue if a free slot is
108 immediately available, else raise the :exc:`Full` exception (*timeout* is
109 ignored in that case).
110
Georg Brandl116aa622007-08-15 14:28:22 +0000111
112.. method:: Queue.put_nowait(item)
113
114 Equivalent to ``put(item, False)``.
115
116
Georg Brandl18244152009-09-02 20:34:52 +0000117.. method:: Queue.get(block=True, timeout=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000118
119 Remove and return an item from the queue. If optional args *block* is true and
120 *timeout* is None (the default), block if necessary until an item is available.
121 If *timeout* is a positive number, it blocks at most *timeout* seconds and
122 raises the :exc:`Empty` exception if no item was available within that time.
123 Otherwise (*block* is false), return an item if one is immediately available,
124 else raise the :exc:`Empty` exception (*timeout* is ignored in that case).
125
Georg Brandl116aa622007-08-15 14:28:22 +0000126
127.. method:: Queue.get_nowait()
128
129 Equivalent to ``get(False)``.
130
131Two methods are offered to support tracking whether enqueued tasks have been
132fully processed by daemon consumer threads.
133
134
135.. method:: Queue.task_done()
136
137 Indicate that a formerly enqueued task is complete. Used by queue consumer
138 threads. For each :meth:`get` used to fetch a task, a subsequent call to
139 :meth:`task_done` tells the queue that the processing on the task is complete.
140
141 If a :meth:`join` is currently blocking, it will resume when all items have been
142 processed (meaning that a :meth:`task_done` call was received for every item
143 that had been :meth:`put` into the queue).
144
145 Raises a :exc:`ValueError` if called more times than there were items placed in
146 the queue.
147
Georg Brandl116aa622007-08-15 14:28:22 +0000148
149.. method:: Queue.join()
150
151 Blocks until all items in the queue have been gotten and processed.
152
153 The count of unfinished tasks goes up whenever an item is added to the queue.
154 The count goes down whenever a consumer thread calls :meth:`task_done` to
155 indicate that the item was retrieved and all work on it is complete. When the
Raymond Hettinger28c013d2009-03-10 00:07:25 +0000156 count of unfinished tasks drops to zero, :meth:`join` unblocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000157
Georg Brandl116aa622007-08-15 14:28:22 +0000158
159Example of how to wait for enqueued tasks to be completed::
160
Georg Brandl48310cd2009-01-03 21:18:54 +0000161 def worker():
162 while True:
163 item = q.get()
164 do_work(item)
165 q.task_done()
Georg Brandl116aa622007-08-15 14:28:22 +0000166
Georg Brandl48310cd2009-01-03 21:18:54 +0000167 q = Queue()
168 for i in range(num_worker_threads):
Georg Brandl116aa622007-08-15 14:28:22 +0000169 t = Thread(target=worker)
Benjamin Petersone8fcbf62009-01-30 02:29:43 +0000170 t.daemon = True
Georg Brandl48310cd2009-01-03 21:18:54 +0000171 t.start()
Georg Brandl116aa622007-08-15 14:28:22 +0000172
173 for item in source():
Georg Brandl48310cd2009-01-03 21:18:54 +0000174 q.put(item)
Georg Brandl116aa622007-08-15 14:28:22 +0000175
176 q.join() # block until all tasks are done
177