B
    v9aA                 @   s   d Z ddlZddlmZ ddlZddlZddlZddlZddlZe	 Z
dadd Zee G dd deZd	d
 ZG dd dejZG dd dejZdS )z"Brian Quinlan (brian@sweetapp.com)    N)_baseFc              C   sJ   da tt } x| D ]\}}|d  qW x| D ]\}}|  q2W d S )NT)	_shutdownlist_threads_queuesitemsputjoin)r   tq r   /usr/lib/python3.7/thread.py_python_exit!   s    r   c               @   s   e Zd Zdd Zdd ZdS )	_WorkItemc             C   s   || _ || _|| _|| _d S )N)futurefnargskwargs)selfr   r   r   r   r   r   r   __init__.   s    z_WorkItem.__init__c          
   C   sf   | j  sd S y| j| j| j}W n2 tk
rT } z| j | d } W d d }~X Y nX | j | d S )N)r   Zset_running_or_notify_cancelr   r   r   BaseExceptionset_exceptionZ
set_result)r   resultexcr   r   r   run4   s    
z_WorkItem.runN)__name__
__module____qualname__r   r   r   r   r   r   r   -   s   r   c             C   s   |d k	rPy||  W n: t k
rN   tjjddd |  }|d k	rJ|  d S X y`xZ|jdd}|d k	rt|  ~qT|  }ts|d ks|jr|d k	rd|_|d  d S ~qTW W n$ t k
r   tjjddd Y nX d S )NzException in initializer:T)exc_info)blockzException in worker)	r   r   ZLOGGERZcritical_initializer_failedgetr   r   r   )Zexecutor_referenceZ
work_queueinitializerinitargsexecutor	work_itemr   r   r   _workerB   s2    

r%   c               @   s   e Zd ZdS )BrokenThreadPoolN)r   r   r   r   r   r   r   r&   f   s   r&   c               @   sZ   e Zd Ze jZdddZdd Ze	j
jje_dd	 Zd
d ZdddZe	j
jje_dS )ThreadPoolExecutorN r   c             C   s   |d krt  pdd }|dkr(td|d k	r@t|s@td|| _t | _t	 | _
d| _d| _t | _|p|d|   | _|| _|| _d S )N      r   z"max_workers must be greater than 0zinitializer must be a callableFzThreadPoolExecutor-%d)os	cpu_count
ValueErrorcallable	TypeError_max_workersqueueZSimpleQueue_work_queueset_threads_brokenr   	threadingZLock_shutdown_lock_counter_thread_name_prefix_initializer	_initargs)r   max_workersZthread_name_prefixr!   r"   r   r   r   r   q   s     

zThreadPoolExecutor.__init__c           	   O   s   t | dkr| ^}}} n>| s&tdn0d|krB|d}| ^}} ntdt | d  |jZ |jrnt|j|jr|tdtrtdt	 }t
||| |}|j| |  |S Q R X d S )N   zDdescriptor 'submit' of 'ThreadPoolExecutor' object needs an argumentr   z6submit expected at least 1 positional argument, got %dr)   z*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown)lenr/   popr7   r5   r&   r   RuntimeErrorr   ZFuturer   r2   r   _adjust_thread_count)r   r   r   r   fwr   r   r   submit   s*    



zThreadPoolExecutor.submitc             C   s   | j fdd}t| j}|| jk r~d| jp,| |f }tj|tt	| || j | j
| jfd}d|_|  | j| | j t|< d S )Nc             S   s   | d  d S )N)r   )_r
   r   r   r   
weakref_cb   s    z;ThreadPoolExecutor._adjust_thread_count.<locals>.weakref_cbz%s_%d)nametargetr   T)r2   r>   r4   r0   r9   r6   ZThreadr%   weakrefrefr:   r;   Zdaemonstartaddr   )r   rF   Znum_threadsZthread_namer	   r   r   r   rA      s    



z'ThreadPoolExecutor._adjust_thread_countc          	   C   sd   | j T d| _xFy| j }W n tjk
r6   P Y nX |d k	r|jt| j qW W d Q R X d S )NzBA thread initializer failed, the thread pool is not usable anymore)	r7   r5   r2   Z
get_nowaitr1   ZEmptyr   r   r&   )r   r$   r   r   r   r      s    z&ThreadPoolExecutor._initializer_failedTc          	   C   sD   | j  d| _| jd  W d Q R X |r@x| jD ]}|  q0W d S )NT)r7   r   r2   r   r4   r   )r   waitr	   r   r   r   shutdown   s    zThreadPoolExecutor.shutdown)Nr(   Nr   )T)r   r   r   	itertoolscount__next__r8   r   rD   r   Executor__doc__rA   r   rN   r   r   r   r   r'   l   s   
 

r'   )
__author__atexitconcurrent.futuresr   rO   r1   r6   rI   r+   WeakKeyDictionaryr   r   r   registerobjectr   r%   ZBrokenExecutorr&   rR   r'   r   r   r   r   <module>   s   	
$