a
XC?hC" � @ s� d Z dZddlmZ ddlZddlZddlZddlZddlZddl Z e�
� Z da e�
� Zdd� Ze�e� ee d�r�e jejejejd � G d
d � d e�Zd d
� ZG dd� dej�ZG dd� dej�Z dS )z Implements ThreadPoolExecutor.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 ZdZdS )�BrokenThreadPoolzR
Raised when a worker thread in a ThreadPoolExecutor failed initializing.
N)r r r �__doc__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� |du r t dt�� pdd �}|dkr.td��|durFt|�sFtd��|| _t�� | _ t
� d�| _ t
� | _d| _d| _t
�� | _|p�d | �� | _|| _|| _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)�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) rD r rB 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> �lenrA r< rF r? ZThreadr. �weakref�refrG rH �start�addr )r rO Z num_threadsZ thread_namer r
r
r rJ � 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) rD rB 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 ) rD r r> rX r= rY r Zcancelr rA r
)r �waitrZ r- r r
r
r �shutdown� s
*
zThreadPoolExecutor.shutdown)Nr2 Nr
)T)r r r � itertools�count�__next__rE r rM r �Executorr0 rJ r' r\ r
r
r
r r1 v s
�
&
r1 ) r0 �
__author__Zconcurrent.futuresr r] r= r? r" rT r7 �WeakKeyDictionaryr r rC r r Z_register_atexit�hasattrr rR �_at_fork_reinitr* �objectr r. ZBrokenExecutorr/ r` r1 r
r
r
r � s,
�+