Dh %dZddlmZddlZddlZddlZddlmZejr&ddl m Z ddl m Z ddl mZdd l mZdd lmZed Zd ZGd dej(j*ZGddej,Zej0ddd ddZej0dd ddZej0 dddd ddZ dddd ddZy)z3ThreadPooled implementation. Asyncio is supported ) annotationsN)_base_threaded)AbstractEventLoop)Task) Awaitable)Callable) ParamSpecSpec) ThreadPooled threadpooledc8eZdZdZdZeddZeddZy)ThreadPoolExecutorzhProvide readers for protected attributes. Simply extend concurrent.futures.ThreadPoolExecutor. c|jS)z)MaxWorkers. :rtype: int ) _max_workersselfs x/var/www/fastuser/data/www/generator.snapmosaic.io/flask_app/venv/lib/python3.12/site-packages/threaded/_threadpooled.py max_workerszThreadPoolExecutor.max_workers3s    c|jS)z7Executor shutdown state. :rtype: bool ) _shutdownrs r is_shutdownzThreadPoolExecutor.is_shutdown;s ~~rN)returnintrbool)__name__ __module__ __qualname____doc__ __slots__propertyrrrrrrr+s4 I !!rrceZdZUdZdZdZded<edddZeddZ e ddZ ddd d  dfd Z e dd Z e dd ZddZ ddZ dfd ZddZxZS)r z$Post function to ThreadPoolExecutor.) __loop_getter__loop_getter_need_contextNzThreadPoolExecutor | None_ThreadPooled__executorct|jtr4|jj|k(ry|jj t||_y)zPool executor create and configure. :param max_workers: Maximum workers :type max_workers: typing.Optional[int] N)r) isinstancer(rrshutdown)clsrs r configurezThreadPooled.configureKsD cnn&8 9~~))[8 NN # # %+ DrcR|j|jjyy)zShutdown executor.N)r(r+)r,s rr+zThreadPooled.shutdownYs# >> % NN # # % &rct|jtr|jjr|j |jS)z?Executor instance. :rtype: ThreadPoolExecutor )r*r(rrr-rs rexecutorzThreadPooled.executor_s3 $//+=>$//B]B] NN rF loop_getterloop_getter_need_contextcBt||||_||_y)a(Wrap function in future and return. :param func: function to wrap :type func: typing.Optional[Callable] :param loop_getter: Method to get event loop, if wrap in asyncio task :type loop_getter: typing.Union[ None, Callable[..., AbstractEventLoop], AbstractEventLoop ] :param loop_getter_need_context: Loop getter requires function context :type loop_getter_need_context: bool )funcN)super__init___ThreadPooled__loop_getter'_ThreadPooled__loop_getter_need_context)rr5r2r3 __class__s rr7zThreadPooled.__init__is%( d#\g0H'rc|jS)znLoop getter. :rtype: typing.Union[None, Callable[..., AbstractEventLoop], AbstractEventLoop] )r8rs rr2zThreadPooled.loop_getters!!!rc|jS)zBLoop getter need execution context. :rtype: bool )r9rs rr3z%ThreadPooled.loop_getter_need_contexts ...rct|jr.|jr|j|i|S|jS|jS)zGet event loop in decorator class. :return: event loop if available or getter available :rtype: Optional[AbstractEventLoop] )callabler2r3)rargskwargss r _get_loopzThreadPooled._get_loopsN D$$ %,,'t''888##% %rcvj|tj dfd }|S)zHere should be constructed and returned real decorator. :param func: Wrapped function :type func: Callable :return: wrapped coroutine or function :rtype: Callable[..., Union[Awaitable, concurrent.futures.Future]] cj|i|}| jjg|i|S|jjt j g|i|S)zMain function wrapper. :return: coroutine or function :rtype: Union[Awaitable, concurrent.futures.Future] )rAr0submitrun_in_executor functoolspartial)r?r@looppreparedrs rwrapperz3ThreadPooled._get_function_wrapper..wrappersn.)r:r_funcr2r3idrs r__repr__zThreadPooled.__repr__s`''(zzn++./((,(E(E'HIhq\  $ rN)r,type[ThreadPooled]rz int | NonerNone)r,rYrrZ)rr)r58Callable[..., Awaitable[typing.Any] | typing.Any] | Noner2z=None | (Callable[..., AbstractEventLoop] | AbstractEventLoop)r3rrrZ)rz;Callable[..., AbstractEventLoop] | AbstractEventLoop | Noner)r? typing.Anyr@r\rzAbstractEventLoop | None)r5z2Callable[Spec, Awaitable[typing.Any] | typing.Any]rzMCallable[Spec, concurrent.futures.Future[typing.Any] | Awaitable[typing.Any]])r?z>Callable[..., Awaitable[typing.Any] | typing.Any] | typing.Anyr@r\rzconcurrent.futures.Future[typing.Any] | Awaitable[typing.Any] | Callable[..., concurrent.futures.Future[typing.Any] | Awaitable[typing.Any]])rstr)rr r!r"r#r(__annotations__ classmethodr-r+r$r0r7r2r3rArMrOrW __classcell__)r:s@rr r Ds .?I,0J)0 E E&& JNIVZ). IFIS I #' I  I0" D""//  F V>1M11  W 1  rr Fr1cy)z,Overload: function callable, no loop getter.Nrr5r2r3s rr r r)r3cy)z3Overload: function callable, loop getter available.Nrrbs rr r rcrcy)zOverload: No function.Nrrbs rr r rcrcJ|t|||Std|||S)aPost function to ThreadPoolExecutor. :param func: function to wrap :type func: typing.Optional[Callable[..., typing.Union[Awaitable, typing.Any]]] :param loop_getter: Method to get event loop, if wrap in asyncio task :type loop_getter: typing.Union[ None, Callable[..., AbstractEventLoop], AbstractEventLoop ] :param loop_getter_need_context: Loop getter requires function context :type loop_getter_need_context: bool :return: ThreadPooled instance, if called as function or argumented decorator, else callable wrapper :rtype: typing.Union[ThreadPooled, Callable[..., typing.Union[concurrent.futures.Future, Awaitable]]] Nrb)r rbs rr r sC* |#%=   < !9    r)r51Callable[..., Awaitable[typing.Any] | typing.Any]r2rZr3rrz4Callable[..., concurrent.futures.Future[typing.Any]])r5rgr2z4Callable[..., AbstractEventLoop] | AbstractEventLoopr3rrzCallable[..., Task[typing.Any]]rX)r5rZr2;None | Callable[..., AbstractEventLoop] | AbstractEventLoopr3rrr )r5r[r2rhr3rrz[ThreadPooled | Callable[..., concurrent.futures.Future[typing.Any] | Awaitable[typing.Any]])r" __future__rconcurrent.futures concurrentrFtypingr TYPE_CHECKINGasynciorrcollections.abcrr typing_extensionsr r __all__futuresr APIPooledr overloadr rrrrvs #  ))(+ V D *++>>2V >++V r%* 7 ;77# 7 : 77 &+ > ;>F># > % >>!PT%* ! !M!# !  !!FJ PT%*  B M #  a  r