o
    "hJ                     @  s  U d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl	m
Z
mZmZ ddlmZmZ ddlmZ dd	lmZ dd
lmZ ejdkrQd dl	mZmZ nd dlmZmZ ejdkrsd dlmZmZ d;ddZG dd dZnDejdkrd dlZd dl Z dZ!de"d< d Z#de"d< dZ$de"d< e$e!fZ%de"d< e#e!fZ&de"d< e'dd d!Z(G d"d dZnG d#d dZd$Z)de"d%< d&Z*ed'Z+ed(Z,eee  d)Z-ee d*Z.d<d/d0Z/dd1d=d6d7Z0d>d9d:Z1dS )?    )annotationsN)deque)Callable)AnyFinalTypeVar   )current_time	to_thread)BrokenWorkerInterpreter)CapacityLimiter)RunVar)      )TypeVarTupleUnpack)r      )ExecutionFailedcreatefuncCallable[..., Any]argstuple[Any, ...]c              
   C  s@   z	| | }W |dfS  t y } z
|dfW  Y d }~S d }~ww )NTF)BaseException)r   r   retvalexc r   G/var/www/html/venv/lib/python3.10/site-packages/anyio/to_interpreter.py_interp_call   s   
r   c                   @  8   e Zd ZU dZded< dddZddd	ZdddZdS )Workerr   float	last_usedreturnNonec                 C  s   t  | _d S N)r   _interpreterselfr   r   r   __init__"   s   Worker.__init__c                 C  s   | j   d S r%   )r&   closer'   r   r   r   destroy%   s   Worker.destroyr   Callable[..., T_Retval]r   r   T_Retvalc              
   C  sJ   z| j t||\}}W n ty } zt|j|d }~ww |r#||S r%   )r&   callr   r   r   excinfo)r(   r   r   resis_exceptionr   r   r   r   r0   (   s   Worker.callNr#   r$   r   r.   r   r   r#   r/   __name__
__module____qualname__r"   __annotations__r)   r,   r0   r   r   r   r   r       s
   
 

r    )r         r   UNBOUNDFMT_UNPICKLEDFMT_PICKLEDQUEUE_PICKLE_ARGSQUEUE_UNPICKLE_ARGSa_  
import _interpqueues
from _interpreters import NotShareableError
from pickle import loads, dumps, HIGHEST_PROTOCOL

QUEUE_PICKLE_ARGS = (1, 2)
QUEUE_UNPICKLE_ARGS = (0, 2)

item = _interpqueues.get(queue_id)[0]
try:
    func, args = loads(item)
    retval = func(*args)
except BaseException as exc:
    is_exception = True
    retval = exc
else:
    is_exception = False

try:
    _interpqueues.put(queue_id, (retval, is_exception), *QUEUE_UNPICKLE_ARGS)
except NotShareableError:
    retval = dumps(retval, HIGHEST_PROTOCOL)
    _interpqueues.put(queue_id, (retval, is_exception), *QUEUE_PICKLE_ARGS)
    z<string>execc                   @  r   )r    r   r!   r"   r#   r$   c                 C  s6   t  | _tjdgtR  | _t | jd| ji d S )Nr   queue_id)_interpretersr   _interpreter_id_interpqueuesrB   	_queue_idset___main___attrsr'   r   r   r   r)   `   s
   
r*   c                 C  s   t | j t| j d S r%   )rG   r,   rH   rE   rF   r'   r   r   r   r,   g   s   r-   r   r.   r   r   r/   c           	      C  s   dd l }|||f|j}tj| j|gtR   t| j	t
}|r%t|t| j}|d d \\}}}|tkr>||}|rB||S )Nr   r=   )pickledumpsHIGHEST_PROTOCOLrG   putrH   rA   rE   rC   rF   	_run_funcr   getr@   loads)	r(   r   r   rJ   itemexc_infor2   r3   fmtr   r   r   r0   k   s   
r4   Nr5   r6   r7   r   r   r   r   r    ]   s
   
 

c                   @  s8   e Zd ZU dZded< dddZdddZdddZdS )r    r   r!   r"   r#   r$   c                 C  s   t d)Nz,subinterpreters require at least Python 3.13)RuntimeErrorr'   r   r   r   r)      s   r*   r   r.   r   r   r/   c                 C  s   t r%   )NotImplementedError)r(   r   r   r   r   r   r0      s   r4   c                 C  s   d S r%   r   r'   r   r   r   r,      s   r-   Nr5   r6   )r8   r9   r:   r"   r;   r)   r0   r,   r   r   r   r   r       s
   
 

   DEFAULT_CPU_COUNT   r/   PosArgsT_available_workers_default_interpreter_limiterworkersdeque[Worker]r#   r$   c                 C  s   | D ]}|   q|   d S r%   )r,   clear)r\   workerr   r   r   _stop_workers   s   
r`   limiter&Callable[[Unpack[PosArgsT]], T_Retval]Unpack[PosArgsT]rb   CapacityLimiter | Nonec             
     sf  |du rt  }zt }W n ty%   t }t| tt| Y nw |4 I dH  z|	 }W n t
y?   t }Y nw W d  I dH  n1 I dH sPw   Y  z5tj|j| ||dI dH W t }|r||d j tkrrntj| j|dI dH  |sht |_|| S t }|r||d j tkrntj| j|dI dH  |st |_|| w )a  
    Call the given function with the given arguments in a subinterpreter.

    .. warning:: On Python 3.13, the :mod:`concurrent.interpreters` module was not yet
        available, so the code path for that Python version relies on an undocumented,
        private API. As such, it is recommended to not rely on this function for anything
        mission-critical on Python 3.13.

    :param func: a callable
    :param args: the positional arguments for the callable
    :param limiter: capacity limiter to use to limit the total number of subinterpreters
        running (if omitted, the default limiter is used)
    :return: the result of the call
    :raises BrokenWorkerInterpreter: if there's an internal error in a subinterpreter

    Nra   r   )#current_default_interpreter_limiter_idle_workersrO   LookupErrorr   setatexitregisterr`   pop
IndexErrorr    r
   run_syncr0   r	   r"   MAX_WORKER_IDLE_TIMEpopleftr,   append)r   rb   r   idle_workersr_   nowr   r   r   rn      sR   

(rn   r   c                  C  s<   zt  W S  ty   tt pt} t |  |  Y S w )z
    Return the capacity limiter used by default to limit the number of concurrently
    running subinterpreters.

    Defaults to the number of CPU cores.

    :return: a capacity limiter object

    )r[   rO   rh   r   os	cpu_countrW   ri   ra   r   r   r   rf      s   


rf   )r   r   r   r   )r\   r]   r#   r$   )r   rc   r   rd   rb   re   r#   r/   )r#   r   )2
__future__r   rj   rt   syscollectionsr   collections.abcr   typingr   r   r    r	   r
   _core._exceptionsr   _core._synchronizationr   lowlevelr   version_infor   r   typing_extensionsconcurrent.interpretersr   r   r   r    rG   rE   r>   r;   r?   r@   rA   rB   compilerN   rW   ro   r/   rY   rg   r[   r`   rn   rf   r   r   r   r   <module>   sX    



&

9