#tjUddlmZdZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZdd lmZdd lmZdd lmZejd kr dd l mZmZndd lmZmZejdkrddlmZmZd9dZGddZnbejdkrJddl 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$dZn Gd%dZd&Z*de#d'<d(Z+e d)Z,ed*Z-eeed+Z.eed,Z/d:d0Z0dd1d;d6Z1dexecc6eZdZUdZded<ddZddZdd ZdS)r%rr&r'rr(ctj|_tjdgtR|_tj|jd|jidS)Nr queue_id) _interpretersr_interpreter_id _interpqueuesrH _queue_idset___main___attrsr,s r"r.z_Worker.__init__gs\#0#7#9#9D *1!J6IJJJDN  ,$z4>&B     r/cjtj|jtj|jdSr*)rOr2rPrMrNr,s r"r2z_Worker.destroyns/  !$. 1 1 1  !$"6 7 7 7 7 7r/rr3rrr4c|ddl}|||f|j}tj|j|gt Rtj|j t}|rt|tj |j}|dd\\}}}|tkr||}|r||S)NrrC)pickledumpsHIGHEST_PROTOCOLrOputrPrGrMrIrN _run_funcrgetrFloads) r-rrrTitemexc_infor8r9fmts r"r6z _Worker.callrs MMM<<t f.EFFD  dnd G5F G G G G$)$*> JJH 8-h777#DN33C'*2A2w $ S,k!!ll3''  Jr/Nr:r;r<rAr/r"r%r%dsc      8 8 8 8      r/c6eZdZUdZded<ddZdd Zdd ZdS)r%rr&r'rr(c td)Nz,subinterpreters require at least Python 3.13) RuntimeErrorr,s r"r.z_Worker.__init__sMNN Nr/rr3rrr4ctr*)NotImplementedError)r-rrs r"r6z _Worker.calls & %r/cdSr*rAr,s r"r2z_Worker.destroys Dr/Nr:r;)r=r>r?r'r@r.r6r2rAr/r"r%r%sg  O O O O & & & &      r/DEFAULT_CPU_COUNTr4PosArgsT_available_workers_default_interpreter_limiterworkersdeque[_Worker]r(c`|D]}||dSr*)r2clear)rjworkers r" _stop_workersros5 MMOOOOOr/limiter&Callable[[Unpack[PosArgsT]], T_Retval]Unpack[PosArgsT]rqCapacityLimiter | Nonec0K|t} t}nR#t$rEt }t|t jt|YnwxYw|4d{V | }n#t$rt}YnwxYwdddd{Vn#1d{VswxYwY tj |j|||d{V t}|rO||djz t"krn5tj |j|d{V|Ot|_||S#t}|rO||djz t"krn5tj |j|d{V|Ot|_||wxYw)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 Nrpr)r _idle_workersrY LookupErrorrsetatexitregisterropop IndexErrorr%r rr6r r'MAX_WORKER_IDLE_TIMEpopleftr2append)rrqr idle_workersrnnows r"rrs*5775$((** 555ww ,''' |444445  !%%''FF   YYFFF  $' K             nn V\!_..2FFF$\%9%9%;%;%CWUUU U U U U U U U  V (>>F####nn V\!_..2FFF$\%9%9%;%;%CWUUU U U U U U U U  V (>>F####sS.A A=<A=C BCB:7C9B::C CC!"F B Hrc tS#t$rFtt jpt }t||cYSwxYw)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 )rirYrwros cpu_countrerxrps r"rrsj+//111 !",.."E4EFF$((111sA A+*A+)rrrrrr)rjrkrr()rrrrrsrqrtrr4)rr)3 __future__r__all__ryrsys collectionsrcollections.abcrtypingrr r r r _core._exceptionsr_core._synchronizationrlowlevelr version_inforrtyping_extensionsconcurrent.interpretersrrr#r%rOrMrDr@rErFrGrHcompilerXrer}r4rgrvrirorrrAr/r"rs4"""""""   $$$$$$&&&&&&&&&&%%%%%%%%666666333333w+++++++++66666666w????????!!!!.   GMK +W55555"/!99999 0 5I:#########L" 7:   < # #&uW~&';<< 6vo67UVV'+6$6$6$6$6$6$rr/