U
e5d@" @ s d Z dZddlZddlmZ ddlZddlZddlZddlZddl Z e
Z da dd Z
ee
G dd d eZd
d ZG d d
d
ejZG dd dejZdS )z Implements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com) N)_baseFc C sB da tt } | D ]\}}|d q| D ]\}}| q,d S NT) _shutdownlist_threads_queuesitemsputjoin)r tq r 1/usr/lib64/python3.8/concurrent/futures/thread.py _python_exit! s r c @ s e Zd Zdd Zdd ZdS ) _WorkItemc C s || _ || _|| _|| _d S N)futurefnargskwargs)selfr r r r r r r
__init__. s z_WorkItem.__init__c
C sf | j sd S z| j| j| j}W n2 tk
rT } z| j | d } W 5 d }~X Y nX | j | d S r )r Z set_running_or_notify_cancelr r r
BaseException
set_exceptionZ
set_result)r resultexcr r r
run4 s
z
_WorkItem.runN)__name__
__module__ __qualname__r r r r r r
r - s r c C s |d k rRz || W n< t k
rP tjjddd | }|d k rJ| Y d S X zx|jdd}|d k r| ~| }|d k r|j ~qT| }t s|d ks|j r|d k rd|_ |
d W d S ~qTW n$ t k
r tjjddd Y nX d S )NzException in initializer:T)exc_info)blockzException in worker) r r ZLOGGERZcritical_initializer_failedgetr _idle_semaphorereleaser r )Zexecutor_referenceZ
work_queue initializerinitargsZexecutor work_itemr r r
_workerB s8
r( c @ s e Zd ZdZdS )BrokenThreadPoolzR
Raised when a worker thread in a ThreadPoolExecutor failed initializing.
N)r r r __doc__r r r r
r) m s r) c @ sf e Zd Ze jZdddZdd Ze j
jj e_ e j
jj e_ dd Z
d
d Zdd
dZe j
jj e_ dS )ThreadPoolExecutorN r c C s |dkr t dt pdd }|dkr.td|dk rFt|sFtd|| _t | _ t
d| _ t
| _d| _d| _t
| _|pd | | _|| _|| _dS )
a Initializes a new ThreadPoolExecutor instance.
Args:
max_workers: The maximum number of threads that can be used to
execute the given calls.
thread_name_prefix: An optional name prefix to give our threads.
initializer: A callable used to initialize worker threads.
initargs: A tuple of arguments to pass to the initializer.
N r z"max_workers must be greater than 0z initializer must be a callableFzThreadPoolExecutor-%d)minos cpu_count
ValueErrorcallable TypeError _max_workersqueueZ SimpleQueue _work_queue threadingZ Semaphorer# set_threads_brokenr ZLock_shutdown_lock_counter_thread_name_prefix _initializer _initargs)r Z max_workersZthread_name_prefixr% r& r r r
r x s$
zThreadPoolExecutor.__init__c
O s t | dkr| ^}}} nV| s&tdnHd|krZ|d}| ^}} dd l}|jdtdd ntdt | d |jf |jrt|j|j rt
d t rt
d
t }t
||| |}|j| | |W 5 Q R S Q R X d S ) N zDdescriptor 'submit' of 'ThreadPoolExecutor' object needs an argumentr r z.Passing 'fn' as keyword argument is deprecated)
stacklevelz6submit expected at least 1 positional argument, got %dr. z*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown)lenr5 popwarningswarnDeprecationWarningr= r< r) r RuntimeErrorr ZFuturer r8 r _adjust_thread_count)r r r r rF fwr r r
submit s6
zThreadPoolExecutor.submitc C s | j jddrd S | jfdd}t| j}|| jk rd| jp>| |f }tj|t t
| || j| j | j
fd}d|_| | j| | jt|< d S )Nr )Ztimeoutc S s | d d S r )r )_r r r r
weakref_cb s z;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)nametargetr T)r# acquirer8 rD r; r6 r? r9 ZThreadr( weakrefrefr@ rA Zdaemonstartaddr )r rO Z num_threadsZ thread_namer
r r r
rJ s&
z'ThreadPoolExecutor._adjust_thread_countc C sb | j R d| _z| j }W n tjk
r6 Y qTY nX |d k r|jt| j qW 5 Q R X d S )NzBA thread initializer failed, the thread pool is not usable anymore) r= r< r8 Z
get_nowaitr7 ZEmptyr r r) )r r' r r r
r! s
z&ThreadPoolExecutor._initializer_failedTc C s@ | j d| _| jd W 5 Q R X |r<| jD ] }| q.d S r )r= r r8 r r; r )r waitr
r r r
shutdown s
zThreadPoolExecutor.shutdown)Nr, Nr )T)r r r itertoolscount__next__r> r rM r Executor__text_signature__r* rJ r! rX r r r r
r+ s s
&
r+ )r*
__author__atexitZconcurrent.futuresr rY r7 r9 rS r1 WeakKeyDictionaryr r r registerobjectr r( ZBrokenExecutorr) r\ r+ r r r r
s
+