o 6Wj@sUddlmZdZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZdd lmZdd lmZdd lmZejd krSdd l mZmZndd lmZmZejdkruddlmZmZd=ddZGdddZnDejdkrddl Z ddl!Z!dZ"de#d<dZ$de#d<dZ%de#d<e%e"fZ&de#d <e$e"fZ'de#d!<e(d"d#d$Z)Gd%ddZnGd&ddZd'Z*de#d(<d)Z+e d*Z,ed+Z-eeed,Z.eed-Z/d>d1d2Z0dd3d?d8d9Z1d@d;d<Z2dS)A) annotations)run_sync#current_default_interpreter_limiterN)deque)Callable)AnyFinalTypeVar) current_time to_thread)BrokenWorkerInterpreter)CapacityLimiter)RunVar) ) TypeVarTupleUnpack)r)ExecutionFailedcreatefuncCallable[..., Any]argstuple[Any, ...]returntuple[Any, bool]c Cs@z ||}W|dfSty}z |dfWYd}~Sd}~ww)NTF) BaseException)rrretvalexcr c/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/anyio/to_interpreter.py _interp_calls r"c@8eZdZUdZded<dddZddd ZdddZdS)_Workerrfloat last_usedrNonecCs t|_dSN)r _interpreterselfr r r!__init__)s _Worker.__init__cCs|jdSr()r)closer*r r r!destroy,s_Worker.destroyrCallable[..., T_Retval]rrT_Retvalc CsJz |jt||\}}Wnty}zt|j|d}~ww|r#||Sr()r)callr"rr excinfo)r+rrres is_exceptionrr r r!r3/s  _Worker.callNrr'rr1rrrr2__name__ __module__ __qualname__r&__annotations__r,r/r3r r r r!r$&s   r$)r rUNBOUND FMT_UNPICKLED FMT_PICKLEDQUEUE_PICKLE_ARGSQUEUE_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) zexecc@r#)r$rr%r&rr'cCs6t|_tjdgtR|_t|jd|jidS)Nr queue_id) _interpretersr_interpreter_id _interpqueuesrE _queue_idset___main___attrsr*r r r!r,gs  r-cCst|jt|jdSr()rJr/rKrHrIr*r r r!r/ns r0rr1rrr2c Csddl}|||f|j}tj|j|gtRt|j t }|r%t |t |j}|dd\\}}}|t kr>||}|rB||S)Nrr@)pickledumpsHIGHEST_PROTOCOLrJputrKrDrHrFrI _run_funcr getrCloads) r+rrrMitemexc_infor5r6fmtr r r!r3rs  r7Nr8r9r:r r r r!r$ds   c@s8eZdZUdZded<dddZdd dZdddZdS)r$rr%r&rr'cCstd)Nz,subinterpreters require at least Python 3.13) RuntimeErrorr*r r r!r,sr-rr1rrr2cCstr()NotImplementedError)r+rrr r r!r3sr7cCsdSr(r r*r r r!r/sr0Nr8r9)r;r<r=r&r>r,r3r/r r r r!r$s   DEFAULT_CPU_COUNTr2PosArgsT_available_workers_default_interpreter_limiterworkersdeque[_Worker]r'cCs|D]}|q|dSr()r/clear)r_workerr r r! _stop_workerss  rclimiter&Callable[[Unpack[PosArgsT]], T_Retval]Unpack[PosArgsT]reCapacityLimiter | Nonec sf|durt}zt}Wnty%t}t|tt|Ynw|4IdHz| }Wn t y?t }YnwWdIdHn 1IdHsPwYz5t j |j|||dIdHWt}|r||djtkrrnt j |j|dIdH|sht|_||St}|r||djtkrnt j |j|dIdH|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 Nrdr)r _idle_workersrR LookupErrorrsetatexitregisterrcpop IndexErrorr$r rr3r r&MAX_WORKER_IDLE_TIMEpopleftr/append)rrer idle_workersrbnowr r r!rsR      (   rrcCs<ztWStyttpt}t||YSw)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^rRrjros cpu_countrZrkrdr r r!rs   r)rrrrrr)r_r`rr')rrfrrgrerhrr2)rr)3 __future__r__all__rlrusys collectionsrcollections.abcrtypingrrr r r _core._exceptionsr _core._synchronizationrlowlevelr version_inforrtyping_extensionsconcurrent.interpretersrrr"r$rJrHrAr>rBrCrDrEcompilerQrZrpr2r\rir^rcrrr r r r!sZ            &    9