File: //usr/lib/python3.9/concurrent/futures/__pycache__/thread.cpython-39.pyc
a
kx�g� � @ s� d Z dZddlmZ ddlZddlZddlZddlZddlZddl Z e�
� Z da e�
� Zdd� Ze�e� G dd � d e�Zd
d � ZG d d
� d
ej�ZG dd� dej�ZdS )z Implements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com)� )�_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
�//usr/lib/python3.9/concurrent/futures/thread.py� _python_exit s " r c @ 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 �run/ 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 �_worker? 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. j 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 u 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.<locals>.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 �Executorr/ rI r& r[ r
r
r
r r0 p s
�
&
r0 )r/ �
__author__Zconcurrent.futuresr r\ r< r> r! rS r6 �WeakKeyDictionaryr r rB r r Z_register_atexit�objectr r- ZBrokenExecutorr. r_ r0 r
r
r
r �<module> s"
+