a
XC?hC" � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZe� � Z
da e� � Z
dd� Ze�e� eed�r�eje
je
je
jd� G d d
� d
e�Zd d � ZG d
d� dej�ZG dd� dej�ZdS )z"Brian Quinlan ([email protected])� )�_baseNFc C sf t � daW d � n1 s 0 Y tt�� �} | D ]\}}|�d � q8| D ]\}}|�� qPd S �NT)�_global_shutdown_lock� _shutdown�list�_threads_queues�items�put�join)r �t�q� r
�>/opt/alt/python39/lib64/python3.9/concurrent/futures/thread.py� _python_exit s " r �register_at_fork)ZbeforeZafter_in_childZafter_in_parentc @ s& e Zd Zdd� Zdd� Zeej�ZdS )� _WorkItemc C s || _ || _|| _|| _d S �N)�future�fn�args�kwargs)�selfr r r r r
r
r �__init__/ s z_WorkItem.__init__c
C sl | j �� sd S z| j| ji | j��}W n4 tyZ } z | j �|� d } W Y d }~nd }~0 0 | j �|� d S r )r Z set_running_or_notify_cancelr r r �
BaseException�
set_exceptionZ
set_result)r �result�excr
r
r �run5 s
z
_WorkItem.runN) �__name__�
__module__� __qualname__r r � classmethod�types� GenericAlias�__class_getitem__r
r
r
r r . s
r c C s� |d urPz ||� W n: t yN tjjddd� | � }|d urH|�� Y d S 0 zx|jdd�}|d ur�|�� ~| � }|d ur�|j�� ~qR| � }t s�|d u s�|j r�|d ur�d|_ |�
d � W d S ~qRW n" t y� tjjddd� Y n0 d S )NzException in initializer:T)�exc_info)�blockzException in worker) r r ZLOGGERZcritical�_initializer_failed�getr �_idle_semaphore�releaser r )Zexecutor_referenceZ
work_queue� initializer�initargsZexecutor� work_itemr
r
r �_workerE s8
r. c @ s e Zd ZdS )�BrokenThreadPoolN)r r r r
r
r
r r/ p s r/ c @ s` e Zd Ze�� jZddd�Zdd� Ze j
jj e_ dd � Z d
d � Z
dd
d�dd�Ze j
jj e_ dS )�ThreadPoolExecutorN� r
c C s� |d u r t dt�� pdd �}|dkr.td��|d urFt|�sFtd��|| _t�� | _ t
� d�| _ t
� | _d| _d| _t
�� | _|p�d| �� | _|| _|| _d S ) N� � � r z"max_workers must be greater than 0z initializer must be a callableFzThreadPoolExecutor-%d)�min�os� cpu_count�
ValueError�callable� TypeError� _max_workers�queueZ SimpleQueue� _work_queue� threadingZ Semaphorer) �set�_threads�_brokenr �Lock�_shutdown_lock�_counter�_thread_name_prefix� _initializer� _initargs)r Z max_workersZthread_name_prefixr+ r, r
r
r r { s$
�zThreadPoolExecutor.__init__c O s� | j �� t�v | jr t| j��| jr,td��tr8td��t�� }t||||�}| j �
|� | � � |W d � W d � S 1 s�0 Y W d � n1 s�0 Y d S )Nz*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown) rC r rA r/ r � RuntimeErrorr ZFuturer r= r �_adjust_thread_count)r r r r �f�wr
r
r �submit� s
zThreadPoolExecutor.submitc C s� | j jdd�rd S | jfdd�}t| j�}|| jk r�d| jp>| |f }tj|t t
� | |�| j| j | j
fd�}|�� | 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)�name�targetr )r) �acquirer= �lenr@ r; rE r> ZThreadr. �weakref�refrF rG �start�addr )r rN Z num_threadsZ thread_namer r
r
r rI � s$
�
�� z'ThreadPoolExecutor._adjust_thread_countc C st | j �Z d| _z| j�� }W n tjy4 Y qRY n0 |d ur|j�t| j�� qW d � n1 sf0 Y d S )NzBA thread initializer failed, the thread pool is not usable anymore) rC rA r= �
get_nowaitr< �Emptyr r r/ )r r- r
r
r r' � s
z&ThreadPoolExecutor._initializer_failedTF)�cancel_futuresc C s� | j �b d| _|rNz| j�� }W n tjy8 Y qNY n0 |d ur|j�� q| j�d � W d � n1 sn0 Y |r�| j D ] }|�
� q�d S r ) rC r r= rW r< rX r Zcancelr r@ r
)r �waitrY r- r r
r
r �shutdown� s
*
zThreadPoolExecutor.shutdown)Nr1 Nr
)T)r r r � itertools�count�__next__rD r rL r �Executor�__doc__rI r' r[ r
r
r
r r0 v s
�
&
r0 ) �
__author__Zconcurrent.futuresr r\ r< r> r" rS r6 �WeakKeyDictionaryr r rB r r Z_register_atexit�hasattrr rQ �_at_fork_reinitr* �objectr r. ZBrokenExecutorr/ r_ r0 r
r
r
r � s*
�+