U
e5d@" @ s d Z ddlZddlmZ ddlZddlZddlZddlZddlZe Z
da dd Z e
e G dd deZd d
ZG d d d ejZG d
d dejZdS )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 ZdS )BrokenThreadPoolN)r r r 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 |d kr t dt pdd }|dkr.td|d k rFt|sFtd|| _t | _ t
d| _ t
| _d| _d| _t
| _|pd| | _|| _|| _d S ) 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)lenr4 popwarningswarnDeprecationWarningr< r; r) r RuntimeErrorr ZFuturer r7 r _adjust_thread_count)r r r r rE 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# acquirer7 rC r: r5 r> r8 ZThreadr( weakrefrefr? r@ Zdaemonstartaddr )r rN Z num_threadsZ thread_namer
r r r
rI 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; r7 Z
get_nowaitr6 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 r7 r r: r )r waitr
r r r
shutdown s
zThreadPoolExecutor.shutdown)Nr+ Nr )T)r r r itertoolscount__next__r= r rL r Executor__text_signature____doc__rI r! rW r r r r
r* s s
&
r* )
__author__atexitZconcurrent.futuresr rX r6 r8 rR r0 WeakKeyDictionaryr r r registerobjectr r( ZBrokenExecutorr) r[ r* r r r r
s
+