blob: 7b3d63dd211ea49e06fed4c79d57533854b4457c [file] [log] [blame]
Jeremy Hylton92bb6e72002-08-14 19:25:42 +00001"""Thread module emulating a subset of Java's threading model."""
Guido van Rossum7f5013a1998-04-09 22:01:42 +00002
Antoine Pitrou4a8bcdf2017-05-28 14:02:26 +02003import os as _os
Fred Drakea8725952002-12-30 23:32:50 +00004import sys as _sys
Georg Brandl2067bfd2008-05-25 13:05:15 +00005import _thread
Kyle Stanleyb61b8182020-03-27 15:31:22 -04006import functools
Fred Drakea8725952002-12-30 23:32:50 +00007
Victor Stinnerae586492014-09-02 23:18:25 +02008from time import monotonic as _time
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02009from _weakrefset import WeakSet
R David Murrayb186f1df2014-10-04 17:43:54 -040010from itertools import islice as _islice, count as _count
Raymond Hettingerec4b1742013-03-10 17:57:28 -070011try:
Raymond Hettingerec4b1742013-03-10 17:57:28 -070012 from _collections import deque as _deque
Brett Cannoncd171c82013-07-04 17:43:24 -040013except ImportError:
Raymond Hettingerec4b1742013-03-10 17:57:28 -070014 from collections import deque as _deque
Guido van Rossum7f5013a1998-04-09 22:01:42 +000015
Benjamin Petersonb3085c92008-09-01 23:09:31 +000016# Note regarding PEP 8 compliant names
17# This threading model was originally inspired by Java, and inherited
18# the convention of camelCase function and method names from that
Ezio Melotti30b9d5d2013-08-17 15:50:46 +030019# language. Those original names are not in any imminent danger of
Benjamin Petersonb3085c92008-09-01 23:09:31 +000020# being deprecated (even for Py3k),so this module provides them as an
21# alias for the PEP 8 compliant names
22# Note that using the new PEP 8 compliant names facilitates substitution
23# with the multiprocessing module, which doesn't provide the old
24# Java inspired names.
25
Victor Stinnerd12e7572019-05-21 12:44:57 +020026__all__ = ['get_ident', 'active_count', 'Condition', 'current_thread',
27 'enumerate', 'main_thread', 'TIMEOUT_MAX',
Martin Panter19e69c52015-11-14 12:46:42 +000028 'Event', 'Lock', 'RLock', 'Semaphore', 'BoundedSemaphore', 'Thread',
29 'Barrier', 'BrokenBarrierError', 'Timer', 'ThreadError',
Victor Stinnercd590a72019-05-28 00:39:52 +020030 'setprofile', 'settrace', 'local', 'stack_size',
Mario Corchero0001a1b2020-11-04 10:27:43 +010031 'excepthook', 'ExceptHookArgs', 'gettrace', 'getprofile']
Guido van Rossum7f5013a1998-04-09 22:01:42 +000032
Raymond Hettinger5cee47f2011-01-11 19:59:46 +000033# Rename some stuff so "from threading import *" is safe
Georg Brandl2067bfd2008-05-25 13:05:15 +000034_start_new_thread = _thread.start_new_thread
35_allocate_lock = _thread.allocate_lock
Antoine Pitrou7b476992013-09-07 23:38:37 +020036_set_sentinel = _thread._set_sentinel
Victor Stinner2a129742011-05-30 23:02:52 +020037get_ident = _thread.get_ident
Jake Teslerb121f632019-05-22 08:43:17 -070038try:
39 get_native_id = _thread.get_native_id
40 _HAVE_THREAD_NATIVE_ID = True
41 __all__.append('get_native_id')
42except AttributeError:
43 _HAVE_THREAD_NATIVE_ID = False
Georg Brandl2067bfd2008-05-25 13:05:15 +000044ThreadError = _thread.error
Antoine Pitrou434736a2009-11-10 18:46:01 +000045try:
46 _CRLock = _thread.RLock
47except AttributeError:
48 _CRLock = None
Antoine Pitrou7c3e5772010-04-14 15:44:10 +000049TIMEOUT_MAX = _thread.TIMEOUT_MAX
Georg Brandl2067bfd2008-05-25 13:05:15 +000050del _thread
Guido van Rossum7f5013a1998-04-09 22:01:42 +000051
Guido van Rossum7f5013a1998-04-09 22:01:42 +000052
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000053# Support for profile and trace hooks
54
55_profile_hook = None
56_trace_hook = None
57
58def setprofile(func):
Georg Brandlc30b59f2013-10-13 10:43:59 +020059 """Set a profile function for all threads started from the threading module.
60
61 The func will be passed to sys.setprofile() for each thread, before its
62 run() method is called.
63
64 """
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000065 global _profile_hook
66 _profile_hook = func
Tim Petersd1b108b2003-06-29 17:24:17 +000067
Mario Corchero0001a1b2020-11-04 10:27:43 +010068def getprofile():
69 """Get the profiler function as set by threading.setprofile()."""
70 return _profile_hook
71
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000072def settrace(func):
Georg Brandlc30b59f2013-10-13 10:43:59 +020073 """Set a trace function for all threads started from the threading module.
74
75 The func will be passed to sys.settrace() for each thread, before its run()
76 method is called.
77
78 """
Jeremy Hyltonbfccb352003-06-29 16:58:41 +000079 global _trace_hook
80 _trace_hook = func
Guido van Rossum7f5013a1998-04-09 22:01:42 +000081
Mario Corchero0001a1b2020-11-04 10:27:43 +010082def gettrace():
83 """Get the trace function as set by threading.settrace()."""
84 return _trace_hook
85
Guido van Rossum7f5013a1998-04-09 22:01:42 +000086# Synchronization classes
87
88Lock = _allocate_lock
89
Victor Stinner135b6d82012-03-03 01:32:57 +010090def RLock(*args, **kwargs):
Georg Brandlc30b59f2013-10-13 10:43:59 +020091 """Factory function that returns a new reentrant lock.
92
93 A reentrant lock must be released by the thread that acquired it. Once a
94 thread has acquired a reentrant lock, the same thread may acquire it again
95 without blocking; the thread must release it once for each time it has
96 acquired it.
97
98 """
Victor Stinner135b6d82012-03-03 01:32:57 +010099 if _CRLock is None:
100 return _PyRLock(*args, **kwargs)
Antoine Pitrou434736a2009-11-10 18:46:01 +0000101 return _CRLock(*args, **kwargs)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000102
Victor Stinner135b6d82012-03-03 01:32:57 +0100103class _RLock:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200104 """This class implements reentrant lock objects.
105
106 A reentrant lock must be released by the thread that acquired it. Once a
107 thread has acquired a reentrant lock, the same thread may acquire it
108 again without blocking; the thread must release it once for each time it
109 has acquired it.
110
111 """
Tim Petersb90f89a2001-01-15 03:26:36 +0000112
Victor Stinner135b6d82012-03-03 01:32:57 +0100113 def __init__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000114 self._block = _allocate_lock()
115 self._owner = None
116 self._count = 0
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000117
118 def __repr__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000119 owner = self._owner
Antoine Pitroub0872682009-11-09 16:08:16 +0000120 try:
121 owner = _active[owner].name
122 except KeyError:
123 pass
Raymond Hettinger62f4dad2014-05-25 18:22:35 -0700124 return "<%s %s.%s object owner=%r count=%d at %s>" % (
125 "locked" if self._block.locked() else "unlocked",
126 self.__class__.__module__,
127 self.__class__.__qualname__,
128 owner,
129 self._count,
130 hex(id(self))
131 )
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000132
Victor Stinner87255be2020-04-07 23:11:49 +0200133 def _at_fork_reinit(self):
134 self._block._at_fork_reinit()
135 self._owner = None
136 self._count = 0
137
Antoine Pitrou7c3e5772010-04-14 15:44:10 +0000138 def acquire(self, blocking=True, timeout=-1):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200139 """Acquire a lock, blocking or non-blocking.
140
141 When invoked without arguments: if this thread already owns the lock,
142 increment the recursion level by one, and return immediately. Otherwise,
143 if another thread owns the lock, block until the lock is unlocked. Once
144 the lock is unlocked (not owned by any thread), then grab ownership, set
145 the recursion level to one, and return. If more than one thread is
146 blocked waiting until the lock is unlocked, only one at a time will be
147 able to grab ownership of the lock. There is no return value in this
148 case.
149
150 When invoked with the blocking argument set to true, do the same thing
151 as when called without arguments, and return true.
152
153 When invoked with the blocking argument set to false, do not block. If a
154 call without an argument would block, return false immediately;
155 otherwise, do the same thing as when called without arguments, and
156 return true.
157
158 When invoked with the floating-point timeout argument set to a positive
159 value, block for at most the number of seconds specified by timeout
160 and as long as the lock cannot be acquired. Return true if the lock has
161 been acquired, false if the timeout has elapsed.
162
163 """
Victor Stinner2a129742011-05-30 23:02:52 +0200164 me = get_ident()
Antoine Pitroub0872682009-11-09 16:08:16 +0000165 if self._owner == me:
Raymond Hettinger720da572013-03-10 15:13:35 -0700166 self._count += 1
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000167 return 1
Antoine Pitrou7c3e5772010-04-14 15:44:10 +0000168 rc = self._block.acquire(blocking, timeout)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000169 if rc:
Guido van Rossumd0648992007-08-20 19:25:41 +0000170 self._owner = me
171 self._count = 1
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000172 return rc
173
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000174 __enter__ = acquire
175
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000176 def release(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200177 """Release a lock, decrementing the recursion level.
178
179 If after the decrement it is zero, reset the lock to unlocked (not owned
180 by any thread), and if any other threads are blocked waiting for the
181 lock to become unlocked, allow exactly one of them to proceed. If after
182 the decrement the recursion level is still nonzero, the lock remains
183 locked and owned by the calling thread.
184
185 Only call this method when the calling thread owns the lock. A
186 RuntimeError is raised if this method is called when the lock is
187 unlocked.
188
189 There is no return value.
190
191 """
Victor Stinner2a129742011-05-30 23:02:52 +0200192 if self._owner != get_ident():
Georg Brandl495f7b52009-10-27 15:28:25 +0000193 raise RuntimeError("cannot release un-acquired lock")
Guido van Rossumd0648992007-08-20 19:25:41 +0000194 self._count = count = self._count - 1
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000195 if not count:
Guido van Rossumd0648992007-08-20 19:25:41 +0000196 self._owner = None
197 self._block.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000198
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000199 def __exit__(self, t, v, tb):
200 self.release()
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000201
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000202 # Internal methods used by condition variables
203
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000204 def _acquire_restore(self, state):
Guido van Rossumd0648992007-08-20 19:25:41 +0000205 self._block.acquire()
206 self._count, self._owner = state
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000207
208 def _release_save(self):
Victor Stinnerc2824d42011-04-24 23:41:33 +0200209 if self._count == 0:
210 raise RuntimeError("cannot release un-acquired lock")
Guido van Rossumd0648992007-08-20 19:25:41 +0000211 count = self._count
212 self._count = 0
213 owner = self._owner
214 self._owner = None
215 self._block.release()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000216 return (count, owner)
217
218 def _is_owned(self):
Victor Stinner2a129742011-05-30 23:02:52 +0200219 return self._owner == get_ident()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000220
Antoine Pitrou434736a2009-11-10 18:46:01 +0000221_PyRLock = _RLock
222
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000223
Victor Stinner135b6d82012-03-03 01:32:57 +0100224class Condition:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200225 """Class that implements a condition variable.
226
227 A condition variable allows one or more threads to wait until they are
228 notified by another thread.
229
230 If the lock argument is given and not None, it must be a Lock or RLock
231 object, and it is used as the underlying lock. Otherwise, a new RLock object
232 is created and used as the underlying lock.
233
234 """
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000235
Victor Stinner135b6d82012-03-03 01:32:57 +0100236 def __init__(self, lock=None):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000237 if lock is None:
238 lock = RLock()
Guido van Rossumd0648992007-08-20 19:25:41 +0000239 self._lock = lock
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000240 # Export the lock's acquire() and release() methods
241 self.acquire = lock.acquire
242 self.release = lock.release
243 # If the lock defines _release_save() and/or _acquire_restore(),
244 # these override the default implementations (which just call
245 # release() and acquire() on the lock). Ditto for _is_owned().
246 try:
247 self._release_save = lock._release_save
248 except AttributeError:
249 pass
250 try:
251 self._acquire_restore = lock._acquire_restore
252 except AttributeError:
253 pass
254 try:
255 self._is_owned = lock._is_owned
256 except AttributeError:
257 pass
Raymond Hettingerec4b1742013-03-10 17:57:28 -0700258 self._waiters = _deque()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000259
Victor Stinner87255be2020-04-07 23:11:49 +0200260 def _at_fork_reinit(self):
261 self._lock._at_fork_reinit()
262 self._waiters.clear()
263
Thomas Wouters477c8d52006-05-27 19:21:47 +0000264 def __enter__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000265 return self._lock.__enter__()
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000266
Thomas Wouters477c8d52006-05-27 19:21:47 +0000267 def __exit__(self, *args):
Guido van Rossumd0648992007-08-20 19:25:41 +0000268 return self._lock.__exit__(*args)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000269
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000270 def __repr__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000271 return "<Condition(%s, %d)>" % (self._lock, len(self._waiters))
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000272
273 def _release_save(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000274 self._lock.release() # No state to save
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000275
276 def _acquire_restore(self, x):
Guido van Rossumd0648992007-08-20 19:25:41 +0000277 self._lock.acquire() # Ignore saved state
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000278
279 def _is_owned(self):
Benjamin Peterson672b8032008-06-11 19:14:14 +0000280 # Return True if lock is owned by current_thread.
Serhiy Storchaka52005c22014-09-21 22:08:13 +0300281 # This method is called only if _lock doesn't have _is_owned().
Serhiy Storchaka1f21eaa2019-09-01 12:16:51 +0300282 if self._lock.acquire(False):
Guido van Rossumd0648992007-08-20 19:25:41 +0000283 self._lock.release()
Tim Petersbc0e9102002-04-04 22:55:58 +0000284 return False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000285 else:
Tim Petersbc0e9102002-04-04 22:55:58 +0000286 return True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000287
288 def wait(self, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200289 """Wait until notified or until a timeout occurs.
290
291 If the calling thread has not acquired the lock when this method is
292 called, a RuntimeError is raised.
293
294 This method releases the underlying lock, and then blocks until it is
295 awakened by a notify() or notify_all() call for the same condition
296 variable in another thread, or until the optional timeout occurs. Once
297 awakened or timed out, it re-acquires the lock and returns.
298
299 When the timeout argument is present and not None, it should be a
300 floating point number specifying a timeout for the operation in seconds
301 (or fractions thereof).
302
303 When the underlying lock is an RLock, it is not released using its
304 release() method, since this may not actually unlock the lock when it
305 was acquired multiple times recursively. Instead, an internal interface
306 of the RLock class is used, which really unlocks it even when it has
307 been recursively acquired several times. Another internal interface is
308 then used to restore the recursion level when the lock is reacquired.
309
310 """
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000311 if not self._is_owned():
Georg Brandl495f7b52009-10-27 15:28:25 +0000312 raise RuntimeError("cannot wait on un-acquired lock")
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000313 waiter = _allocate_lock()
314 waiter.acquire()
Guido van Rossumd0648992007-08-20 19:25:41 +0000315 self._waiters.append(waiter)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000316 saved_state = self._release_save()
Antoine Pitroua64b92e2014-08-29 23:26:36 +0200317 gotit = False
Tim Petersc951bf92001-04-02 20:15:57 +0000318 try: # restore state no matter what (e.g., KeyboardInterrupt)
319 if timeout is None:
320 waiter.acquire()
Georg Brandlb9a43912010-10-28 09:03:20 +0000321 gotit = True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000322 else:
Antoine Pitrou7c3e5772010-04-14 15:44:10 +0000323 if timeout > 0:
324 gotit = waiter.acquire(True, timeout)
325 else:
326 gotit = waiter.acquire(False)
Georg Brandlb9a43912010-10-28 09:03:20 +0000327 return gotit
Tim Petersc951bf92001-04-02 20:15:57 +0000328 finally:
329 self._acquire_restore(saved_state)
Antoine Pitroua64b92e2014-08-29 23:26:36 +0200330 if not gotit:
331 try:
332 self._waiters.remove(waiter)
333 except ValueError:
334 pass
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000335
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000336 def wait_for(self, predicate, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200337 """Wait until a condition evaluates to True.
338
339 predicate should be a callable which result will be interpreted as a
340 boolean value. A timeout may be provided giving the maximum time to
341 wait.
342
343 """
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000344 endtime = None
345 waittime = timeout
346 result = predicate()
347 while not result:
348 if waittime is not None:
349 if endtime is None:
350 endtime = _time() + waittime
351 else:
352 waittime = endtime - _time()
353 if waittime <= 0:
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000354 break
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000355 self.wait(waittime)
356 result = predicate()
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000357 return result
358
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000359 def notify(self, n=1):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200360 """Wake up one or more threads waiting on this condition, if any.
361
362 If the calling thread has not acquired the lock when this method is
363 called, a RuntimeError is raised.
364
365 This method wakes up at most n of the threads waiting for the condition
366 variable; it is a no-op if no threads are waiting.
367
368 """
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000369 if not self._is_owned():
Georg Brandl495f7b52009-10-27 15:28:25 +0000370 raise RuntimeError("cannot notify on un-acquired lock")
Raymond Hettingerb65e5792013-03-10 20:34:16 -0700371 all_waiters = self._waiters
372 waiters_to_notify = _deque(_islice(all_waiters, n))
373 if not waiters_to_notify:
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000374 return
Raymond Hettingerb65e5792013-03-10 20:34:16 -0700375 for waiter in waiters_to_notify:
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000376 waiter.release()
377 try:
Raymond Hettingerb65e5792013-03-10 20:34:16 -0700378 all_waiters.remove(waiter)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000379 except ValueError:
380 pass
381
Benjamin Peterson672b8032008-06-11 19:14:14 +0000382 def notify_all(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200383 """Wake up all threads waiting on this condition.
384
385 If the calling thread has not acquired the lock when this method
386 is called, a RuntimeError is raised.
387
388 """
Guido van Rossumd0648992007-08-20 19:25:41 +0000389 self.notify(len(self._waiters))
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000390
Benjamin Petersonb3085c92008-09-01 23:09:31 +0000391 notifyAll = notify_all
392
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000393
Victor Stinner135b6d82012-03-03 01:32:57 +0100394class Semaphore:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200395 """This class implements semaphore objects.
396
397 Semaphores manage a counter representing the number of release() calls minus
398 the number of acquire() calls, plus an initial value. The acquire() method
399 blocks if necessary until it can return without making the counter
400 negative. If not given, value defaults to 1.
401
402 """
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000403
Andrew M. Kuchling39d3bfc2000-02-29 00:10:24 +0000404 # After Tim Peters' semaphore class, but not quite the same (no maximum)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000405
Victor Stinner135b6d82012-03-03 01:32:57 +0100406 def __init__(self, value=1):
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000407 if value < 0:
408 raise ValueError("semaphore initial value must be >= 0")
Guido van Rossumd0648992007-08-20 19:25:41 +0000409 self._cond = Condition(Lock())
410 self._value = value
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000411
Antoine Pitrou0454af92010-04-17 23:51:58 +0000412 def acquire(self, blocking=True, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200413 """Acquire a semaphore, decrementing the internal counter by one.
414
415 When invoked without arguments: if the internal counter is larger than
416 zero on entry, decrement it by one and return immediately. If it is zero
417 on entry, block, waiting until some other thread has called release() to
418 make it larger than zero. This is done with proper interlocking so that
419 if multiple acquire() calls are blocked, release() will wake exactly one
420 of them up. The implementation may pick one at random, so the order in
421 which blocked threads are awakened should not be relied on. There is no
422 return value in this case.
423
424 When invoked with blocking set to true, do the same thing as when called
425 without arguments, and return true.
426
427 When invoked with blocking set to false, do not block. If a call without
428 an argument would block, return false immediately; otherwise, do the
429 same thing as when called without arguments, and return true.
430
431 When invoked with a timeout other than None, it will block for at
432 most timeout seconds. If acquire does not complete successfully in
433 that interval, return false. Return true otherwise.
434
435 """
Antoine Pitrou0454af92010-04-17 23:51:58 +0000436 if not blocking and timeout is not None:
437 raise ValueError("can't specify timeout for non-blocking acquire")
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000438 rc = False
Antoine Pitrou0454af92010-04-17 23:51:58 +0000439 endtime = None
Serhiy Storchaka81a58552013-04-22 22:51:43 +0300440 with self._cond:
441 while self._value == 0:
442 if not blocking:
443 break
444 if timeout is not None:
445 if endtime is None:
446 endtime = _time() + timeout
447 else:
448 timeout = endtime - _time()
449 if timeout <= 0:
450 break
451 self._cond.wait(timeout)
452 else:
Serhiy Storchakab00b5962013-04-22 22:54:16 +0300453 self._value -= 1
Serhiy Storchaka81a58552013-04-22 22:51:43 +0300454 rc = True
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000455 return rc
456
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000457 __enter__ = acquire
458
Raymond Hettinger35f63012019-08-29 01:45:19 -0700459 def release(self, n=1):
460 """Release a semaphore, incrementing the internal counter by one or more.
Georg Brandlc30b59f2013-10-13 10:43:59 +0200461
462 When the counter is zero on entry and another thread is waiting for it
463 to become larger than zero again, wake up that thread.
464
465 """
Raymond Hettinger35f63012019-08-29 01:45:19 -0700466 if n < 1:
467 raise ValueError('n must be one or more')
Serhiy Storchaka81a58552013-04-22 22:51:43 +0300468 with self._cond:
Raymond Hettinger35f63012019-08-29 01:45:19 -0700469 self._value += n
470 for i in range(n):
471 self._cond.notify()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000472
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000473 def __exit__(self, t, v, tb):
474 self.release()
Guido van Rossum1a5e21e2006-02-28 21:57:43 +0000475
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000476
Éric Araujo0cdd4452011-07-28 00:28:28 +0200477class BoundedSemaphore(Semaphore):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200478 """Implements a bounded semaphore.
479
480 A bounded semaphore checks to make sure its current value doesn't exceed its
481 initial value. If it does, ValueError is raised. In most situations
482 semaphores are used to guard resources with limited capacity.
483
484 If the semaphore is released too many times it's a sign of a bug. If not
485 given, value defaults to 1.
486
487 Like regular semaphores, bounded semaphores manage a counter representing
488 the number of release() calls minus the number of acquire() calls, plus an
489 initial value. The acquire() method blocks if necessary until it can return
490 without making the counter negative. If not given, value defaults to 1.
491
492 """
493
Victor Stinner135b6d82012-03-03 01:32:57 +0100494 def __init__(self, value=1):
495 Semaphore.__init__(self, value)
Skip Montanaroe428bb72001-08-20 20:27:58 +0000496 self._initial_value = value
497
Raymond Hettinger35f63012019-08-29 01:45:19 -0700498 def release(self, n=1):
499 """Release a semaphore, incrementing the internal counter by one or more.
Georg Brandlc30b59f2013-10-13 10:43:59 +0200500
501 When the counter is zero on entry and another thread is waiting for it
502 to become larger than zero again, wake up that thread.
503
504 If the number of releases exceeds the number of acquires,
505 raise a ValueError.
506
507 """
Raymond Hettinger35f63012019-08-29 01:45:19 -0700508 if n < 1:
509 raise ValueError('n must be one or more')
Tim Peters7634e1c2013-10-08 20:55:51 -0500510 with self._cond:
Raymond Hettinger35f63012019-08-29 01:45:19 -0700511 if self._value + n > self._initial_value:
Tim Peters7634e1c2013-10-08 20:55:51 -0500512 raise ValueError("Semaphore released too many times")
Raymond Hettinger35f63012019-08-29 01:45:19 -0700513 self._value += n
514 for i in range(n):
515 self._cond.notify()
Skip Montanaroe428bb72001-08-20 20:27:58 +0000516
517
Victor Stinner135b6d82012-03-03 01:32:57 +0100518class Event:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200519 """Class implementing event objects.
520
521 Events manage a flag that can be set to true with the set() method and reset
522 to false with the clear() method. The wait() method blocks until the flag is
523 true. The flag is initially false.
524
525 """
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000526
527 # After Tim Peters' event class (without is_posted())
528
Victor Stinner135b6d82012-03-03 01:32:57 +0100529 def __init__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000530 self._cond = Condition(Lock())
531 self._flag = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000532
Victor Stinner87255be2020-04-07 23:11:49 +0200533 def _at_fork_reinit(self):
534 # Private method called by Thread._reset_internal_locks()
535 self._cond._at_fork_reinit()
Gregory P. Smith9bd4a242011-01-04 18:33:38 +0000536
Benjamin Peterson672b8032008-06-11 19:14:14 +0000537 def is_set(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200538 """Return true if and only if the internal flag is true."""
Guido van Rossumd0648992007-08-20 19:25:41 +0000539 return self._flag
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000540
Benjamin Petersonb3085c92008-09-01 23:09:31 +0000541 isSet = is_set
Benjamin Petersonf0923f52008-08-18 22:10:13 +0000542
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000543 def set(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200544 """Set the internal flag to true.
545
546 All threads waiting for it to become true are awakened. Threads
547 that call wait() once the flag is true will not block at all.
548
549 """
Benjamin Peterson414918a2015-10-10 19:34:46 -0700550 with self._cond:
Guido van Rossumd0648992007-08-20 19:25:41 +0000551 self._flag = True
Benjamin Peterson672b8032008-06-11 19:14:14 +0000552 self._cond.notify_all()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000553
554 def clear(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200555 """Reset the internal flag to false.
556
557 Subsequently, threads calling wait() will block until set() is called to
558 set the internal flag to true again.
559
560 """
Benjamin Peterson414918a2015-10-10 19:34:46 -0700561 with self._cond:
Guido van Rossumd0648992007-08-20 19:25:41 +0000562 self._flag = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000563
564 def wait(self, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200565 """Block until the internal flag is true.
566
567 If the internal flag is true on entry, return immediately. Otherwise,
568 block until another thread calls set() to set the flag to true, or until
569 the optional timeout occurs.
570
571 When the timeout argument is present and not None, it should be a
572 floating point number specifying a timeout for the operation in seconds
573 (or fractions thereof).
574
575 This method returns the internal flag on exit, so it will always return
576 True except if a timeout is given and the operation times out.
577
578 """
Benjamin Peterson414918a2015-10-10 19:34:46 -0700579 with self._cond:
Charles-François Natalided03482012-01-07 18:24:56 +0100580 signaled = self._flag
581 if not signaled:
582 signaled = self._cond.wait(timeout)
583 return signaled
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000584
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000585
586# A barrier class. Inspired in part by the pthread_barrier_* api and
587# the CyclicBarrier class from Java. See
588# http://sourceware.org/pthreads-win32/manual/pthread_barrier_init.html and
589# http://java.sun.com/j2se/1.5.0/docs/api/java/util/concurrent/
590# CyclicBarrier.html
591# for information.
592# We maintain two main states, 'filling' and 'draining' enabling the barrier
593# to be cyclic. Threads are not allowed into it until it has fully drained
594# since the previous cycle. In addition, a 'resetting' state exists which is
595# similar to 'draining' except that threads leave with a BrokenBarrierError,
Ezio Melottie130a522011-10-19 10:58:56 +0300596# and a 'broken' state in which all threads get the exception.
Victor Stinner135b6d82012-03-03 01:32:57 +0100597class Barrier:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200598 """Implements a Barrier.
599
600 Useful for synchronizing a fixed number of threads at known synchronization
Carl Bordum Hansen62fa51f2019-03-09 18:38:05 +0100601 points. Threads block on 'wait()' and are simultaneously awoken once they
602 have all made that call.
Georg Brandlc30b59f2013-10-13 10:43:59 +0200603
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000604 """
Georg Brandlc30b59f2013-10-13 10:43:59 +0200605
Victor Stinner135b6d82012-03-03 01:32:57 +0100606 def __init__(self, parties, action=None, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200607 """Create a barrier, initialised to 'parties' threads.
608
609 'action' is a callable which, when supplied, will be called by one of
610 the threads after they have all entered the barrier and just prior to
Carl Bordum Hansen62fa51f2019-03-09 18:38:05 +0100611 releasing them all. If a 'timeout' is provided, it is used as the
Georg Brandlc30b59f2013-10-13 10:43:59 +0200612 default for all subsequent 'wait()' calls.
613
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000614 """
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000615 self._cond = Condition(Lock())
616 self._action = action
617 self._timeout = timeout
618 self._parties = parties
619 self._state = 0 #0 filling, 1, draining, -1 resetting, -2 broken
620 self._count = 0
621
622 def wait(self, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200623 """Wait for the barrier.
624
625 When the specified number of threads have started waiting, they are all
626 simultaneously awoken. If an 'action' was provided for the barrier, one
627 of the threads will have executed that callback prior to returning.
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000628 Returns an individual index number from 0 to 'parties-1'.
Georg Brandlc30b59f2013-10-13 10:43:59 +0200629
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000630 """
631 if timeout is None:
632 timeout = self._timeout
633 with self._cond:
634 self._enter() # Block while the barrier drains.
635 index = self._count
636 self._count += 1
637 try:
638 if index + 1 == self._parties:
639 # We release the barrier
640 self._release()
641 else:
642 # We wait until someone releases us
643 self._wait(timeout)
644 return index
645 finally:
646 self._count -= 1
647 # Wake up any threads waiting for barrier to drain.
648 self._exit()
649
650 # Block until the barrier is ready for us, or raise an exception
651 # if it is broken.
652 def _enter(self):
653 while self._state in (-1, 1):
654 # It is draining or resetting, wait until done
655 self._cond.wait()
656 #see if the barrier is in a broken state
657 if self._state < 0:
658 raise BrokenBarrierError
659 assert self._state == 0
660
661 # Optionally run the 'action' and release the threads waiting
662 # in the barrier.
663 def _release(self):
664 try:
665 if self._action:
666 self._action()
667 # enter draining state
668 self._state = 1
669 self._cond.notify_all()
670 except:
671 #an exception during the _action handler. Break and reraise
672 self._break()
673 raise
674
Martin Panter69332c12016-08-04 13:07:31 +0000675 # Wait in the barrier until we are released. Raise an exception
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000676 # if the barrier is reset or broken.
677 def _wait(self, timeout):
Kristján Valur Jónsson63315202010-11-18 12:46:39 +0000678 if not self._cond.wait_for(lambda : self._state != 0, timeout):
679 #timed out. Break the barrier
680 self._break()
681 raise BrokenBarrierError
682 if self._state < 0:
683 raise BrokenBarrierError
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000684 assert self._state == 1
685
686 # If we are the last thread to exit the barrier, signal any threads
687 # waiting for the barrier to drain.
688 def _exit(self):
689 if self._count == 0:
690 if self._state in (-1, 1):
691 #resetting or draining
692 self._state = 0
693 self._cond.notify_all()
694
695 def reset(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200696 """Reset the barrier to the initial state.
697
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000698 Any threads currently waiting will get the BrokenBarrier exception
699 raised.
Georg Brandlc30b59f2013-10-13 10:43:59 +0200700
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000701 """
702 with self._cond:
703 if self._count > 0:
704 if self._state == 0:
705 #reset the barrier, waking up threads
706 self._state = -1
707 elif self._state == -2:
708 #was broken, set it to reset state
709 #which clears when the last thread exits
710 self._state = -1
711 else:
712 self._state = 0
713 self._cond.notify_all()
714
715 def abort(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200716 """Place the barrier into a 'broken' state.
717
718 Useful in case of error. Any currently waiting threads and threads
719 attempting to 'wait()' will have BrokenBarrierError raised.
720
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000721 """
722 with self._cond:
723 self._break()
724
725 def _break(self):
726 # An internal error was detected. The barrier is set to
727 # a broken state all parties awakened.
728 self._state = -2
729 self._cond.notify_all()
730
731 @property
732 def parties(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200733 """Return the number of threads required to trip the barrier."""
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000734 return self._parties
735
736 @property
737 def n_waiting(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200738 """Return the number of threads currently waiting at the barrier."""
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000739 # We don't need synchronization here since this is an ephemeral result
740 # anyway. It returns the correct value in the steady state.
741 if self._state == 0:
742 return self._count
743 return 0
744
745 @property
746 def broken(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200747 """Return True if the barrier is in a broken state."""
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000748 return self._state == -2
749
Georg Brandlc30b59f2013-10-13 10:43:59 +0200750# exception raised by the Barrier class
751class BrokenBarrierError(RuntimeError):
752 pass
Kristján Valur Jónsson3be00032010-10-28 09:43:10 +0000753
754
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000755# Helper to generate new thread names
Victor Stinner98c16c92020-09-23 23:21:19 +0200756_counter = _count(1).__next__
757def _newname(name_template):
758 return name_template % _counter()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000759
760# Active thread administration
761_active_limbo_lock = _allocate_lock()
Tim Peters711906e2005-01-08 07:30:42 +0000762_active = {} # maps thread id to Thread object
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000763_limbo = {}
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200764_dangling = WeakSet()
Victor Stinner468e5fe2019-06-13 01:30:17 +0200765# Set of Thread._tstate_lock locks of non-daemon threads used by _shutdown()
766# to wait until all Python thread states get deleted:
767# see Thread._set_tstate_lock().
768_shutdown_locks_lock = _allocate_lock()
769_shutdown_locks = set()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000770
771# Main class for threads
772
Victor Stinner135b6d82012-03-03 01:32:57 +0100773class Thread:
Georg Brandlc30b59f2013-10-13 10:43:59 +0200774 """A class that represents a thread of control.
775
776 This class can be safely subclassed in a limited fashion. There are two ways
777 to specify the activity: by passing a callable object to the constructor, or
778 by overriding the run() method in a subclass.
779
780 """
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000781
Serhiy Storchaka52005c22014-09-21 22:08:13 +0300782 _initialized = False
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000783
784 def __init__(self, group=None, target=None, name=None,
Victor Stinner135b6d82012-03-03 01:32:57 +0100785 args=(), kwargs=None, *, daemon=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200786 """This constructor should always be called with keyword arguments. Arguments are:
787
788 *group* should be None; reserved for future extension when a ThreadGroup
789 class is implemented.
790
791 *target* is the callable object to be invoked by the run()
792 method. Defaults to None, meaning nothing is called.
793
794 *name* is the thread name. By default, a unique name is constructed of
795 the form "Thread-N" where N is a small decimal number.
796
797 *args* is the argument tuple for the target invocation. Defaults to ().
798
799 *kwargs* is a dictionary of keyword arguments for the target
800 invocation. Defaults to {}.
801
802 If a subclass overrides the constructor, it must make sure to invoke
803 the base class constructor (Thread.__init__()) before doing anything
804 else to the thread.
805
806 """
Guido van Rossum5a43e1a1998-06-09 19:04:26 +0000807 assert group is None, "group argument must be None for now"
Georg Brandla4a8b822005-07-15 09:13:21 +0000808 if kwargs is None:
809 kwargs = {}
Victor Stinner98c16c92020-09-23 23:21:19 +0200810 if name:
811 name = str(name)
812 else:
813 name = _newname("Thread-%d")
814 if target is not None:
815 try:
816 target_name = target.__name__
817 name += f" ({target_name})"
818 except AttributeError:
819 pass
820
Guido van Rossumd0648992007-08-20 19:25:41 +0000821 self._target = target
Victor Stinner98c16c92020-09-23 23:21:19 +0200822 self._name = name
Guido van Rossumd0648992007-08-20 19:25:41 +0000823 self._args = args
824 self._kwargs = kwargs
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +0000825 if daemon is not None:
826 self._daemonic = daemon
827 else:
828 self._daemonic = current_thread().daemon
Georg Brandl0c77a822008-06-10 16:37:50 +0000829 self._ident = None
Jake Teslerb121f632019-05-22 08:43:17 -0700830 if _HAVE_THREAD_NATIVE_ID:
831 self._native_id = None
Antoine Pitrou7b476992013-09-07 23:38:37 +0200832 self._tstate_lock = None
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000833 self._started = Event()
Tim Petersc363a232013-09-08 18:44:40 -0500834 self._is_stopped = False
Guido van Rossumd0648992007-08-20 19:25:41 +0000835 self._initialized = True
Victor Stinnercd590a72019-05-28 00:39:52 +0200836 # Copy of sys.stderr used by self._invoke_excepthook()
Guido van Rossumd0648992007-08-20 19:25:41 +0000837 self._stderr = _sys.stderr
Victor Stinnercd590a72019-05-28 00:39:52 +0200838 self._invoke_excepthook = _make_invoke_excepthook()
Antoine Pitrou5da7e792013-09-08 13:19:06 +0200839 # For debugging and _after_fork()
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200840 _dangling.add(self)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000841
Antoine Pitrou7b476992013-09-07 23:38:37 +0200842 def _reset_internal_locks(self, is_alive):
Gregory P. Smith9bd4a242011-01-04 18:33:38 +0000843 # private! Called by _after_fork() to reset our internal locks as
844 # they may be in an invalid state leading to a deadlock or crash.
Victor Stinner87255be2020-04-07 23:11:49 +0200845 self._started._at_fork_reinit()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200846 if is_alive:
Victor Stinner5909a492020-11-16 15:20:34 +0100847 # bpo-42350: If the fork happens when the thread is already stopped
848 # (ex: after threading._shutdown() has been called), _tstate_lock
849 # is None. Do nothing in this case.
850 if self._tstate_lock is not None:
851 self._tstate_lock._at_fork_reinit()
852 self._tstate_lock.acquire()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200853 else:
854 # The thread isn't alive after fork: it doesn't have a tstate
855 # anymore.
Tim Petersb5e9ac92013-09-09 14:41:50 -0500856 self._is_stopped = True
Antoine Pitrou7b476992013-09-07 23:38:37 +0200857 self._tstate_lock = None
Gregory P. Smith9bd4a242011-01-04 18:33:38 +0000858
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000859 def __repr__(self):
Guido van Rossumd0648992007-08-20 19:25:41 +0000860 assert self._initialized, "Thread.__init__() was not called"
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000861 status = "initial"
Benjamin Peterson672b8032008-06-11 19:14:14 +0000862 if self._started.is_set():
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000863 status = "started"
Tim Peters72460fa2013-09-09 18:48:24 -0500864 self.is_alive() # easy way to get ._is_stopped set when appropriate
Tim Petersc363a232013-09-08 18:44:40 -0500865 if self._is_stopped:
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000866 status = "stopped"
Guido van Rossumd0648992007-08-20 19:25:41 +0000867 if self._daemonic:
Georg Brandl0c77a822008-06-10 16:37:50 +0000868 status += " daemon"
869 if self._ident is not None:
870 status += " %s" % self._ident
Guido van Rossumd0648992007-08-20 19:25:41 +0000871 return "<%s(%s, %s)>" % (self.__class__.__name__, self._name, status)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000872
873 def start(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200874 """Start the thread's activity.
875
876 It must be called at most once per thread object. It arranges for the
877 object's run() method to be invoked in a separate thread of control.
878
879 This method will raise a RuntimeError if called more than once on the
880 same thread object.
881
882 """
Guido van Rossumd0648992007-08-20 19:25:41 +0000883 if not self._initialized:
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000884 raise RuntimeError("thread.__init__() not called")
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000885
Benjamin Peterson672b8032008-06-11 19:14:14 +0000886 if self._started.is_set():
Senthil Kumaranfdd4d0f2010-04-06 03:30:18 +0000887 raise RuntimeError("threads can only be started once")
Victor Stinner066e5b12019-06-14 18:55:22 +0200888
Benjamin Petersond23f8222009-04-05 19:13:16 +0000889 with _active_limbo_lock:
890 _limbo[self] = self
Gregory P. Smith3fdd9642010-02-28 18:57:46 +0000891 try:
892 _start_new_thread(self._bootstrap, ())
893 except Exception:
894 with _active_limbo_lock:
895 del _limbo[self]
896 raise
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000897 self._started.wait()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000898
899 def run(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +0200900 """Method representing the thread's activity.
901
902 You may override this method in a subclass. The standard run() method
903 invokes the callable object passed to the object's constructor as the
904 target argument, if any, with sequential and keyword arguments taken
905 from the args and kwargs arguments, respectively.
906
907 """
Christian Heimesd3eb5a152008-02-24 00:38:49 +0000908 try:
909 if self._target:
910 self._target(*self._args, **self._kwargs)
911 finally:
912 # Avoid a refcycle if the thread is running a function with
913 # an argument that has a member that points to the thread.
914 del self._target, self._args, self._kwargs
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000915
Guido van Rossumd0648992007-08-20 19:25:41 +0000916 def _bootstrap(self):
Guido van Rossum61e21b52007-08-20 19:06:03 +0000917 # Wrapper around the real bootstrap code that ignores
918 # exceptions during interpreter cleanup. Those typically
919 # happen when a daemon thread wakes up at an unfortunate
920 # moment, finds the world around it destroyed, and raises some
921 # random exception *** while trying to report the exception in
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000922 # _bootstrap_inner() below ***. Those random exceptions
Guido van Rossum61e21b52007-08-20 19:06:03 +0000923 # don't help anybody, and they confuse users, so we suppress
924 # them. We suppress them only when it appears that the world
925 # indeed has already been destroyed, so that exceptions in
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000926 # _bootstrap_inner() during normal business hours are properly
Guido van Rossum61e21b52007-08-20 19:06:03 +0000927 # reported. Also, we only suppress them for daemonic threads;
928 # if a non-daemonic encounters this, something else is wrong.
929 try:
Guido van Rossumd0648992007-08-20 19:25:41 +0000930 self._bootstrap_inner()
Guido van Rossum61e21b52007-08-20 19:06:03 +0000931 except:
Guido van Rossumd0648992007-08-20 19:25:41 +0000932 if self._daemonic and _sys is None:
Guido van Rossum61e21b52007-08-20 19:06:03 +0000933 return
934 raise
935
Benjamin Petersond23f8222009-04-05 19:13:16 +0000936 def _set_ident(self):
Victor Stinner2a129742011-05-30 23:02:52 +0200937 self._ident = get_ident()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000938
Jake Teslerb121f632019-05-22 08:43:17 -0700939 if _HAVE_THREAD_NATIVE_ID:
940 def _set_native_id(self):
941 self._native_id = get_native_id()
942
Antoine Pitrou7b476992013-09-07 23:38:37 +0200943 def _set_tstate_lock(self):
944 """
945 Set a lock object which will be released by the interpreter when
946 the underlying thread state (see pystate.h) gets deleted.
947 """
948 self._tstate_lock = _set_sentinel()
949 self._tstate_lock.acquire()
950
Victor Stinner468e5fe2019-06-13 01:30:17 +0200951 if not self.daemon:
952 with _shutdown_locks_lock:
953 _shutdown_locks.add(self._tstate_lock)
954
Guido van Rossumd0648992007-08-20 19:25:41 +0000955 def _bootstrap_inner(self):
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000956 try:
Benjamin Petersond23f8222009-04-05 19:13:16 +0000957 self._set_ident()
Antoine Pitrou7b476992013-09-07 23:38:37 +0200958 self._set_tstate_lock()
Jake Teslerb121f632019-05-22 08:43:17 -0700959 if _HAVE_THREAD_NATIVE_ID:
960 self._set_native_id()
Christian Heimes9e7f1d22008-02-28 12:27:11 +0000961 self._started.set()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000962 with _active_limbo_lock:
963 _active[self._ident] = self
964 del _limbo[self]
Jeremy Hyltonbfccb352003-06-29 16:58:41 +0000965
966 if _trace_hook:
Jeremy Hyltonbfccb352003-06-29 16:58:41 +0000967 _sys.settrace(_trace_hook)
968 if _profile_hook:
Jeremy Hyltonbfccb352003-06-29 16:58:41 +0000969 _sys.setprofile(_profile_hook)
Tim Petersd1b108b2003-06-29 17:24:17 +0000970
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000971 try:
972 self.run()
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000973 except:
Victor Stinnercd590a72019-05-28 00:39:52 +0200974 self._invoke_excepthook(self)
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000975 finally:
Christian Heimes1af737c2008-01-23 08:24:23 +0000976 with _active_limbo_lock:
Christian Heimes1af737c2008-01-23 08:24:23 +0000977 try:
Georg Brandl0c77a822008-06-10 16:37:50 +0000978 # We don't call self._delete() because it also
Christian Heimes1af737c2008-01-23 08:24:23 +0000979 # grabs _active_limbo_lock.
Victor Stinner2a129742011-05-30 23:02:52 +0200980 del _active[get_ident()]
Christian Heimes1af737c2008-01-23 08:24:23 +0000981 except:
982 pass
Guido van Rossum7f5013a1998-04-09 22:01:42 +0000983
Guido van Rossumd0648992007-08-20 19:25:41 +0000984 def _stop(self):
Tim Petersb5e9ac92013-09-09 14:41:50 -0500985 # After calling ._stop(), .is_alive() returns False and .join() returns
986 # immediately. ._tstate_lock must be released before calling ._stop().
987 #
988 # Normal case: C code at the end of the thread's life
989 # (release_sentinel in _threadmodule.c) releases ._tstate_lock, and
990 # that's detected by our ._wait_for_tstate_lock(), called by .join()
991 # and .is_alive(). Any number of threads _may_ call ._stop()
992 # simultaneously (for example, if multiple threads are blocked in
993 # .join() calls), and they're not serialized. That's harmless -
994 # they'll just make redundant rebindings of ._is_stopped and
995 # ._tstate_lock. Obscure: we rebind ._tstate_lock last so that the
996 # "assert self._is_stopped" in ._wait_for_tstate_lock() always works
997 # (the assert is executed only if ._tstate_lock is None).
998 #
999 # Special case: _main_thread releases ._tstate_lock via this
1000 # module's _shutdown() function.
1001 lock = self._tstate_lock
1002 if lock is not None:
1003 assert not lock.locked()
Tim Peters78755232013-09-09 13:47:16 -05001004 self._is_stopped = True
1005 self._tstate_lock = None
Victor Stinner468e5fe2019-06-13 01:30:17 +02001006 if not self.daemon:
1007 with _shutdown_locks_lock:
Victor Stinner6f75c872019-06-13 12:06:24 +02001008 _shutdown_locks.discard(lock)
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001009
Guido van Rossumd0648992007-08-20 19:25:41 +00001010 def _delete(self):
Tim Peters21429932004-07-21 03:36:52 +00001011 "Remove current thread from the dict of currently running threads."
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001012 with _active_limbo_lock:
1013 del _active[get_ident()]
1014 # There must not be any python code between the previous line
1015 # and after the lock is released. Otherwise a tracing function
1016 # could try to acquire the lock again in the same thread, (in
1017 # current_thread()), and would block.
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001018
1019 def join(self, timeout=None):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001020 """Wait until the thread terminates.
1021
1022 This blocks the calling thread until the thread whose join() method is
1023 called terminates -- either normally or through an unhandled exception
1024 or until the optional timeout occurs.
1025
1026 When the timeout argument is present and not None, it should be a
1027 floating point number specifying a timeout for the operation in seconds
1028 (or fractions thereof). As join() always returns None, you must call
Dong-hee Na36d9e9a2019-01-18 18:50:47 +09001029 is_alive() after join() to decide whether a timeout happened -- if the
Georg Brandlc30b59f2013-10-13 10:43:59 +02001030 thread is still alive, the join() call timed out.
1031
1032 When the timeout argument is not present or None, the operation will
1033 block until the thread terminates.
1034
1035 A thread can be join()ed many times.
1036
1037 join() raises a RuntimeError if an attempt is made to join the current
1038 thread as that would cause a deadlock. It is also an error to join() a
1039 thread before it has been started and attempts to do so raises the same
1040 exception.
1041
1042 """
Guido van Rossumd0648992007-08-20 19:25:41 +00001043 if not self._initialized:
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001044 raise RuntimeError("Thread.__init__() not called")
Benjamin Peterson672b8032008-06-11 19:14:14 +00001045 if not self._started.is_set():
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001046 raise RuntimeError("cannot join thread before it is started")
Benjamin Peterson672b8032008-06-11 19:14:14 +00001047 if self is current_thread():
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001048 raise RuntimeError("cannot join current thread")
Tim Peterse5bb0bf2013-10-25 20:46:51 -05001049
Tim Petersc363a232013-09-08 18:44:40 -05001050 if timeout is None:
1051 self._wait_for_tstate_lock()
Tim Peters7bad39f2013-10-25 22:33:52 -05001052 else:
1053 # the behavior of a negative timeout isn't documented, but
Tim Petersa577f1e2013-10-26 11:56:16 -05001054 # historically .join(timeout=x) for x<0 has acted as if timeout=0
Tim Peters7bad39f2013-10-25 22:33:52 -05001055 self._wait_for_tstate_lock(timeout=max(timeout, 0))
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001056
Tim Petersc363a232013-09-08 18:44:40 -05001057 def _wait_for_tstate_lock(self, block=True, timeout=-1):
Antoine Pitrou7b476992013-09-07 23:38:37 +02001058 # Issue #18808: wait for the thread state to be gone.
Tim Petersc363a232013-09-08 18:44:40 -05001059 # At the end of the thread's life, after all knowledge of the thread
1060 # is removed from C data structures, C code releases our _tstate_lock.
Martin Panter46f50722016-05-26 05:35:26 +00001061 # This method passes its arguments to _tstate_lock.acquire().
Tim Petersc363a232013-09-08 18:44:40 -05001062 # If the lock is acquired, the C code is done, and self._stop() is
1063 # called. That sets ._is_stopped to True, and ._tstate_lock to None.
Antoine Pitrou7b476992013-09-07 23:38:37 +02001064 lock = self._tstate_lock
Tim Petersc363a232013-09-08 18:44:40 -05001065 if lock is None: # already determined that the C code is done
1066 assert self._is_stopped
1067 elif lock.acquire(block, timeout):
Antoine Pitrou7b476992013-09-07 23:38:37 +02001068 lock.release()
Tim Petersc363a232013-09-08 18:44:40 -05001069 self._stop()
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001070
Benjamin Petersonfdbea962008-08-18 17:33:47 +00001071 @property
1072 def name(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001073 """A string used for identification purposes only.
1074
1075 It has no semantics. Multiple threads may be given the same name. The
1076 initial name is set by the constructor.
1077
1078 """
Guido van Rossumd0648992007-08-20 19:25:41 +00001079 assert self._initialized, "Thread.__init__() not called"
1080 return self._name
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001081
Benjamin Petersonfdbea962008-08-18 17:33:47 +00001082 @name.setter
1083 def name(self, name):
Guido van Rossumd0648992007-08-20 19:25:41 +00001084 assert self._initialized, "Thread.__init__() not called"
1085 self._name = str(name)
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001086
Benjamin Peterson773c17b2008-08-18 16:45:31 +00001087 @property
1088 def ident(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001089 """Thread identifier of this thread or None if it has not been started.
1090
Skip Montanaro56343312018-05-18 13:38:36 -05001091 This is a nonzero integer. See the get_ident() function. Thread
Georg Brandlc30b59f2013-10-13 10:43:59 +02001092 identifiers may be recycled when a thread exits and another thread is
1093 created. The identifier is available even after the thread has exited.
1094
1095 """
Georg Brandl0c77a822008-06-10 16:37:50 +00001096 assert self._initialized, "Thread.__init__() not called"
1097 return self._ident
1098
Jake Teslerb121f632019-05-22 08:43:17 -07001099 if _HAVE_THREAD_NATIVE_ID:
1100 @property
1101 def native_id(self):
1102 """Native integral thread ID of this thread, or None if it has not been started.
1103
1104 This is a non-negative integer. See the get_native_id() function.
1105 This represents the Thread ID as reported by the kernel.
1106
1107 """
1108 assert self._initialized, "Thread.__init__() not called"
1109 return self._native_id
1110
Benjamin Peterson672b8032008-06-11 19:14:14 +00001111 def is_alive(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001112 """Return whether the thread is alive.
1113
1114 This method returns True just before the run() method starts until just
1115 after the run() method terminates. The module function enumerate()
1116 returns a list of all alive threads.
1117
1118 """
Guido van Rossumd0648992007-08-20 19:25:41 +00001119 assert self._initialized, "Thread.__init__() not called"
Tim Petersc363a232013-09-08 18:44:40 -05001120 if self._is_stopped or not self._started.is_set():
Antoine Pitrou7b476992013-09-07 23:38:37 +02001121 return False
Antoine Pitrou7b476992013-09-07 23:38:37 +02001122 self._wait_for_tstate_lock(False)
Tim Petersc363a232013-09-08 18:44:40 -05001123 return not self._is_stopped
Tim Petersb90f89a2001-01-15 03:26:36 +00001124
Benjamin Petersonfdbea962008-08-18 17:33:47 +00001125 @property
1126 def daemon(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001127 """A boolean value indicating whether this thread is a daemon thread.
1128
1129 This must be set before start() is called, otherwise RuntimeError is
1130 raised. Its initial value is inherited from the creating thread; the
1131 main thread is not a daemon thread and therefore all threads created in
1132 the main thread default to daemon = False.
1133
mbarkhaubb110cc2019-06-22 14:51:06 +02001134 The entire Python program exits when only daemon threads are left.
Georg Brandlc30b59f2013-10-13 10:43:59 +02001135
1136 """
Guido van Rossumd0648992007-08-20 19:25:41 +00001137 assert self._initialized, "Thread.__init__() not called"
1138 return self._daemonic
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001139
Benjamin Petersonfdbea962008-08-18 17:33:47 +00001140 @daemon.setter
1141 def daemon(self, daemonic):
Guido van Rossumd0648992007-08-20 19:25:41 +00001142 if not self._initialized:
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001143 raise RuntimeError("Thread.__init__() not called")
Benjamin Peterson672b8032008-06-11 19:14:14 +00001144 if self._started.is_set():
Antoine Pitrou10959072014-03-17 18:22:41 +01001145 raise RuntimeError("cannot set daemon status of active thread")
Guido van Rossumd0648992007-08-20 19:25:41 +00001146 self._daemonic = daemonic
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001147
Benjamin Peterson6640d722008-08-18 18:16:46 +00001148 def isDaemon(self):
1149 return self.daemon
1150
1151 def setDaemon(self, daemonic):
1152 self.daemon = daemonic
1153
1154 def getName(self):
1155 return self.name
1156
1157 def setName(self, name):
1158 self.name = name
1159
Victor Stinnercd590a72019-05-28 00:39:52 +02001160
1161try:
1162 from _thread import (_excepthook as excepthook,
1163 _ExceptHookArgs as ExceptHookArgs)
1164except ImportError:
1165 # Simple Python implementation if _thread._excepthook() is not available
1166 from traceback import print_exception as _print_exception
1167 from collections import namedtuple
1168
1169 _ExceptHookArgs = namedtuple(
1170 'ExceptHookArgs',
1171 'exc_type exc_value exc_traceback thread')
1172
1173 def ExceptHookArgs(args):
1174 return _ExceptHookArgs(*args)
1175
1176 def excepthook(args, /):
1177 """
1178 Handle uncaught Thread.run() exception.
1179 """
1180 if args.exc_type == SystemExit:
1181 # silently ignore SystemExit
1182 return
1183
1184 if _sys is not None and _sys.stderr is not None:
1185 stderr = _sys.stderr
1186 elif args.thread is not None:
1187 stderr = args.thread._stderr
1188 if stderr is None:
1189 # do nothing if sys.stderr is None and sys.stderr was None
1190 # when the thread was created
1191 return
1192 else:
1193 # do nothing if sys.stderr is None and args.thread is None
1194 return
1195
1196 if args.thread is not None:
1197 name = args.thread.name
1198 else:
1199 name = get_ident()
1200 print(f"Exception in thread {name}:",
1201 file=stderr, flush=True)
1202 _print_exception(args.exc_type, args.exc_value, args.exc_traceback,
1203 file=stderr)
1204 stderr.flush()
1205
1206
Mario Corchero750c5ab2020-11-12 18:27:44 +01001207# Original value of threading.excepthook
1208__excepthook__ = excepthook
1209
1210
Victor Stinnercd590a72019-05-28 00:39:52 +02001211def _make_invoke_excepthook():
1212 # Create a local namespace to ensure that variables remain alive
1213 # when _invoke_excepthook() is called, even if it is called late during
1214 # Python shutdown. It is mostly needed for daemon threads.
1215
1216 old_excepthook = excepthook
1217 old_sys_excepthook = _sys.excepthook
1218 if old_excepthook is None:
1219 raise RuntimeError("threading.excepthook is None")
1220 if old_sys_excepthook is None:
1221 raise RuntimeError("sys.excepthook is None")
1222
1223 sys_exc_info = _sys.exc_info
1224 local_print = print
1225 local_sys = _sys
1226
1227 def invoke_excepthook(thread):
1228 global excepthook
1229 try:
1230 hook = excepthook
1231 if hook is None:
1232 hook = old_excepthook
1233
1234 args = ExceptHookArgs([*sys_exc_info(), thread])
1235
1236 hook(args)
1237 except Exception as exc:
1238 exc.__suppress_context__ = True
1239 del exc
1240
1241 if local_sys is not None and local_sys.stderr is not None:
1242 stderr = local_sys.stderr
1243 else:
1244 stderr = thread._stderr
1245
1246 local_print("Exception in threading.excepthook:",
1247 file=stderr, flush=True)
1248
1249 if local_sys is not None and local_sys.excepthook is not None:
1250 sys_excepthook = local_sys.excepthook
1251 else:
1252 sys_excepthook = old_sys_excepthook
1253
1254 sys_excepthook(*sys_exc_info())
1255 finally:
1256 # Break reference cycle (exception stored in a variable)
1257 args = None
1258
1259 return invoke_excepthook
1260
1261
Martin v. Löwis44f86962001-09-05 13:44:54 +00001262# The timer class was contributed by Itamar Shtull-Trauring
1263
Éric Araujo0cdd4452011-07-28 00:28:28 +02001264class Timer(Thread):
Martin v. Löwis44f86962001-09-05 13:44:54 +00001265 """Call a function after a specified number of seconds:
Tim Petersb64bec32001-09-18 02:26:39 +00001266
Georg Brandlc30b59f2013-10-13 10:43:59 +02001267 t = Timer(30.0, f, args=None, kwargs=None)
1268 t.start()
1269 t.cancel() # stop the timer's action if it's still waiting
1270
Martin v. Löwis44f86962001-09-05 13:44:54 +00001271 """
Tim Petersb64bec32001-09-18 02:26:39 +00001272
R David Murray19aeb432013-03-30 17:19:38 -04001273 def __init__(self, interval, function, args=None, kwargs=None):
Martin v. Löwis44f86962001-09-05 13:44:54 +00001274 Thread.__init__(self)
1275 self.interval = interval
1276 self.function = function
R David Murray19aeb432013-03-30 17:19:38 -04001277 self.args = args if args is not None else []
1278 self.kwargs = kwargs if kwargs is not None else {}
Martin v. Löwis44f86962001-09-05 13:44:54 +00001279 self.finished = Event()
Tim Petersb64bec32001-09-18 02:26:39 +00001280
Martin v. Löwis44f86962001-09-05 13:44:54 +00001281 def cancel(self):
Georg Brandlc30b59f2013-10-13 10:43:59 +02001282 """Stop the timer if it hasn't finished yet."""
Martin v. Löwis44f86962001-09-05 13:44:54 +00001283 self.finished.set()
Tim Petersb64bec32001-09-18 02:26:39 +00001284
Martin v. Löwis44f86962001-09-05 13:44:54 +00001285 def run(self):
1286 self.finished.wait(self.interval)
Benjamin Peterson672b8032008-06-11 19:14:14 +00001287 if not self.finished.is_set():
Martin v. Löwis44f86962001-09-05 13:44:54 +00001288 self.function(*self.args, **self.kwargs)
1289 self.finished.set()
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001290
Antoine Pitrou1023dbb2017-10-02 16:42:15 +02001291
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001292# Special thread class to represent the main thread
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001293
1294class _MainThread(Thread):
1295
1296 def __init__(self):
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +00001297 Thread.__init__(self, name="MainThread", daemon=False)
Tim Petersc363a232013-09-08 18:44:40 -05001298 self._set_tstate_lock()
Christian Heimes9e7f1d22008-02-28 12:27:11 +00001299 self._started.set()
Benjamin Petersond23f8222009-04-05 19:13:16 +00001300 self._set_ident()
Jake Teslerb121f632019-05-22 08:43:17 -07001301 if _HAVE_THREAD_NATIVE_ID:
1302 self._set_native_id()
Benjamin Petersond23f8222009-04-05 19:13:16 +00001303 with _active_limbo_lock:
1304 _active[self._ident] = self
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001305
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001306
1307# Dummy thread class to represent threads not started here.
Tim Peters711906e2005-01-08 07:30:42 +00001308# These aren't garbage collected when they die, nor can they be waited for.
Benjamin Peterson672b8032008-06-11 19:14:14 +00001309# If they invoke anything in threading.py that calls current_thread(), they
Tim Peters711906e2005-01-08 07:30:42 +00001310# leave an entry in the _active dict forever after.
Benjamin Peterson672b8032008-06-11 19:14:14 +00001311# Their purpose is to return *something* from current_thread().
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001312# They are marked as daemon threads so we won't wait for them
1313# when we exit (conform previous semantics).
1314
1315class _DummyThread(Thread):
Tim Petersb90f89a2001-01-15 03:26:36 +00001316
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001317 def __init__(self):
Antoine Pitrou0bd4deb2011-02-25 22:07:43 +00001318 Thread.__init__(self, name=_newname("Dummy-%d"), daemon=True)
Tim Peters711906e2005-01-08 07:30:42 +00001319
Christian Heimes9e7f1d22008-02-28 12:27:11 +00001320 self._started.set()
Benjamin Petersond23f8222009-04-05 19:13:16 +00001321 self._set_ident()
Jake Teslerb121f632019-05-22 08:43:17 -07001322 if _HAVE_THREAD_NATIVE_ID:
1323 self._set_native_id()
Benjamin Petersond23f8222009-04-05 19:13:16 +00001324 with _active_limbo_lock:
1325 _active[self._ident] = self
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001326
Antoine Pitrou8e6e0fd2012-04-19 23:55:01 +02001327 def _stop(self):
1328 pass
1329
Xiang Zhangf3a9fab2017-02-27 11:01:30 +08001330 def is_alive(self):
1331 assert not self._is_stopped and self._started.is_set()
1332 return True
1333
Neal Norwitz45bec8c2002-02-19 03:01:36 +00001334 def join(self, timeout=None):
Guido van Rossum8ca162f2002-04-07 06:36:23 +00001335 assert False, "cannot join a dummy thread"
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001336
1337
1338# Global API functions
1339
Benjamin Peterson672b8032008-06-11 19:14:14 +00001340def current_thread():
Georg Brandlc30b59f2013-10-13 10:43:59 +02001341 """Return the current Thread object, corresponding to the caller's thread of control.
1342
1343 If the caller's thread of control was not created through the threading
1344 module, a dummy thread object with limited functionality is returned.
1345
1346 """
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001347 try:
Victor Stinner2a129742011-05-30 23:02:52 +02001348 return _active[get_ident()]
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001349 except KeyError:
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001350 return _DummyThread()
1351
Benjamin Petersonb3085c92008-09-01 23:09:31 +00001352currentThread = current_thread
Benjamin Petersonf0923f52008-08-18 22:10:13 +00001353
Benjamin Peterson672b8032008-06-11 19:14:14 +00001354def active_count():
Georg Brandlc30b59f2013-10-13 10:43:59 +02001355 """Return the number of Thread objects currently alive.
1356
1357 The returned count is equal to the length of the list returned by
1358 enumerate().
1359
1360 """
Benjamin Petersond23f8222009-04-05 19:13:16 +00001361 with _active_limbo_lock:
1362 return len(_active) + len(_limbo)
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001363
Benjamin Petersonb3085c92008-09-01 23:09:31 +00001364activeCount = active_count
Benjamin Petersonf0923f52008-08-18 22:10:13 +00001365
Antoine Pitroubdec11f2009-11-05 13:49:14 +00001366def _enumerate():
1367 # Same as enumerate(), but without the lock. Internal use only.
1368 return list(_active.values()) + list(_limbo.values())
1369
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001370def enumerate():
Georg Brandlc30b59f2013-10-13 10:43:59 +02001371 """Return a list of all Thread objects currently alive.
1372
1373 The list includes daemonic threads, dummy thread objects created by
1374 current_thread(), and the main thread. It excludes terminated threads and
1375 threads that have not yet been started.
1376
1377 """
Benjamin Petersond23f8222009-04-05 19:13:16 +00001378 with _active_limbo_lock:
1379 return list(_active.values()) + list(_limbo.values())
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001380
Kyle Stanleyb61b8182020-03-27 15:31:22 -04001381
1382_threading_atexits = []
1383_SHUTTING_DOWN = False
1384
1385def _register_atexit(func, *arg, **kwargs):
1386 """CPython internal: register *func* to be called before joining threads.
1387
1388 The registered *func* is called with its arguments just before all
1389 non-daemon threads are joined in `_shutdown()`. It provides a similar
1390 purpose to `atexit.register()`, but its functions are called prior to
1391 threading shutdown instead of interpreter shutdown.
1392
1393 For similarity to atexit, the registered functions are called in reverse.
1394 """
1395 if _SHUTTING_DOWN:
1396 raise RuntimeError("can't register atexit after shutdown")
1397
1398 call = functools.partial(func, *arg, **kwargs)
1399 _threading_atexits.append(call)
1400
1401
Georg Brandl2067bfd2008-05-25 13:05:15 +00001402from _thread import stack_size
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001403
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001404# Create the main thread object,
1405# and make it available for the interpreter
1406# (Py_Main) as threading._shutdown.
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001407
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001408_main_thread = _MainThread()
1409
1410def _shutdown():
Victor Stinner468e5fe2019-06-13 01:30:17 +02001411 """
1412 Wait until the Python thread state of all non-daemon threads get deleted.
1413 """
Tim Petersc363a232013-09-08 18:44:40 -05001414 # Obscure: other threads may be waiting to join _main_thread. That's
1415 # dubious, but some code does it. We can't wait for C code to release
1416 # the main thread's tstate_lock - that won't happen until the interpreter
1417 # is nearly dead. So we release it here. Note that just calling _stop()
1418 # isn't enough: other threads may already be waiting on _tstate_lock.
Antoine Pitrouee84a602017-08-16 20:53:28 +02001419 if _main_thread._is_stopped:
1420 # _shutdown() was already called
1421 return
Victor Stinner468e5fe2019-06-13 01:30:17 +02001422
Kyle Stanleyb61b8182020-03-27 15:31:22 -04001423 global _SHUTTING_DOWN
1424 _SHUTTING_DOWN = True
Victor Stinner468e5fe2019-06-13 01:30:17 +02001425 # Main thread
Tim Petersb5e9ac92013-09-09 14:41:50 -05001426 tlock = _main_thread._tstate_lock
1427 # The main thread isn't finished yet, so its thread state lock can't have
1428 # been released.
1429 assert tlock is not None
1430 assert tlock.locked()
1431 tlock.release()
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001432 _main_thread._stop()
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001433
Kyle Stanleyb61b8182020-03-27 15:31:22 -04001434 # Call registered threading atexit functions before threads are joined.
1435 # Order is reversed, similar to atexit.
1436 for atexit_call in reversed(_threading_atexits):
1437 atexit_call()
1438
Victor Stinner468e5fe2019-06-13 01:30:17 +02001439 # Join all non-deamon threads
1440 while True:
1441 with _shutdown_locks_lock:
1442 locks = list(_shutdown_locks)
1443 _shutdown_locks.clear()
1444
1445 if not locks:
1446 break
1447
1448 for lock in locks:
1449 # mimick Thread.join()
1450 lock.acquire()
1451 lock.release()
1452
1453 # new threads can be spawned while we were waiting for the other
1454 # threads to complete
1455
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001456
1457def main_thread():
Andrew Svetlovb1dd5572013-09-04 10:33:11 +03001458 """Return the main thread object.
1459
1460 In normal conditions, the main thread is the thread from which the
1461 Python interpreter was started.
1462 """
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001463 return _main_thread
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001464
Jim Fultond15dc062004-07-14 19:11:50 +00001465# get thread-local implementation, either from the thread
1466# module, or from the python fallback
1467
1468try:
Georg Brandl2067bfd2008-05-25 13:05:15 +00001469 from _thread import _local as local
Brett Cannoncd171c82013-07-04 17:43:24 -04001470except ImportError:
Jim Fultond15dc062004-07-14 19:11:50 +00001471 from _threading_local import local
1472
Guido van Rossum7f5013a1998-04-09 22:01:42 +00001473
Jesse Nollera8513972008-07-17 16:49:17 +00001474def _after_fork():
Antoine Pitrou4a8bcdf2017-05-28 14:02:26 +02001475 """
1476 Cleanup threading module state that should not exist after a fork.
1477 """
Jesse Nollera8513972008-07-17 16:49:17 +00001478 # Reset _active_limbo_lock, in case we forked while the lock was held
1479 # by another (non-forked) thread. http://bugs.python.org/issue874900
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001480 global _active_limbo_lock, _main_thread
Victor Stinner468e5fe2019-06-13 01:30:17 +02001481 global _shutdown_locks_lock, _shutdown_locks
Jesse Nollera8513972008-07-17 16:49:17 +00001482 _active_limbo_lock = _allocate_lock()
1483
1484 # fork() only copied the current thread; clear references to others.
1485 new_active = {}
Victor Stinnerd8ff44c2020-03-27 17:50:42 +01001486
1487 try:
1488 current = _active[get_ident()]
1489 except KeyError:
1490 # fork() was called in a thread which was not spawned
1491 # by threading.Thread. For example, a thread spawned
1492 # by thread.start_new_thread().
1493 current = _MainThread()
1494
Andrew Svetlov58b5c5a2013-09-04 07:01:07 +03001495 _main_thread = current
Victor Stinner468e5fe2019-06-13 01:30:17 +02001496
1497 # reset _shutdown() locks: threads re-register their _tstate_lock below
1498 _shutdown_locks_lock = _allocate_lock()
1499 _shutdown_locks = set()
1500
Jesse Nollera8513972008-07-17 16:49:17 +00001501 with _active_limbo_lock:
Antoine Pitrou5da7e792013-09-08 13:19:06 +02001502 # Dangling thread instances must still have their locks reset,
1503 # because someone may join() them.
1504 threads = set(_enumerate())
1505 threads.update(_dangling)
1506 for thread in threads:
Charles-François Natalib055bf62011-12-18 18:45:16 +01001507 # Any lock/condition variable may be currently locked or in an
1508 # invalid state, so we reinitialize them.
Jesse Nollera8513972008-07-17 16:49:17 +00001509 if thread is current:
Antoine Pitrou5fe291f2008-09-06 23:00:03 +00001510 # There is only one active thread. We reset the ident to
1511 # its new value since it can have changed.
Antoine Pitrou7b476992013-09-07 23:38:37 +02001512 thread._reset_internal_locks(True)
Victor Stinner2a129742011-05-30 23:02:52 +02001513 ident = get_ident()
Antoine Pitrou5fe291f2008-09-06 23:00:03 +00001514 thread._ident = ident
Jesse Nollera8513972008-07-17 16:49:17 +00001515 new_active[ident] = thread
1516 else:
1517 # All the others are already stopped.
Antoine Pitrou7b476992013-09-07 23:38:37 +02001518 thread._reset_internal_locks(False)
Charles-François Natalib055bf62011-12-18 18:45:16 +01001519 thread._stop()
Jesse Nollera8513972008-07-17 16:49:17 +00001520
1521 _limbo.clear()
1522 _active.clear()
1523 _active.update(new_active)
1524 assert len(_active) == 1
Antoine Pitrou4a8bcdf2017-05-28 14:02:26 +02001525
1526
Gregory P. Smith163468a2017-05-29 10:03:41 -07001527if hasattr(_os, "register_at_fork"):
1528 _os.register_at_fork(after_in_child=_after_fork)