jgSUddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZej<dk\rddl mZnddl mZerdd l!m"Z"ed ed ef Z#ed Z$edZ%dejLddfdZ'e(edr$ejRZ)ejTZ*ee#ge#fe+d<nejRZ)de#de#fdZ*GddZ,Gddee$e%fZ-Gddee$e%fZ.eddde/deeee$eeee%ffee$ee%ffgee$e%fffdZ0edddeee$eeee%ffee$ee%ffde/dee$e%ffd Z0 d*dddeeee$eeee%ffee$ee%ffde/deeeee$eeee%ffee$ee%ffgee$e%ffee$e%fffd!Z0ed"dd#d$e/d%ed&deee$e%fgee$eeee%ffffd'Z1ed"dd#dee$e%fd$e/d%ed&dee$eeee%fffd(Z1 d*d"dd#deee$e%fd$e/d%ed&deeee$e%fgee$eeee%fffee$eeee%ffffd)Z1y)+N)FutureThreadPoolExecutor) TYPE_CHECKINGAny AwaitableCallable CoroutineDictGenericListOptionalTypeVarUnionoverload)CurrentThreadExecutor)Local) ) ParamSpec) OptExcInfo_F.)bound_P_Rcontextreturnc|D]8}|j|} |j|k7r|j|:y#t$r|j|YXwxYwN)getset LookupError)rcvarcvalues J/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/asgiref/sync.py_restore_contextr&,s^T" xxzV#    HHV  s$?AAmarkcoroutinefunctionfunccDtjj|_|Sr)asyncio coroutines _is_coroutine)r(s r%r'r'Cs$//== c"eZdZdZdZdZdZy)ThreadSensitiveContextaYAsync context manager to manage context for thread sensitive mode This context manager controls which thread pool executor is used when in thread sensitive mode. By default, a single thread pool executor is shared within a process. The ThreadSensitiveContext() context manager may be used to specify a thread pool per context. This context manager is re-entrant, so only the outer-most call to ThreadSensitiveContext will set the context. Usage: >>> import time >>> async with ThreadSensitiveContext(): ... await sync_to_async(time.sleep, 1)() cd|_yr)tokenselfs r%__init__zThreadSensitiveContext.__init__\s  r-cK tjj|S#t$r(tjj ||_Y|SwxYwwr) SyncToAsyncthread_sensitive_contextr r"r!r1r2s r% __aenter__z!ThreadSensitiveContext.__aenter___sT H  0 0 4 4 6  H$==AA$GDJ  Hs%A$A-AAAAcK|jsytjj|d}|r|j tj j |jywr)r1r6context_to_thread_executorpopshutdownr7reset)r3excvaluetbexecutors r% __aexit__z ThreadSensitiveContext.__aexit__gsOzz 99==dDI     ,,224::>sA*A,N)__name__ __module__ __qualname____doc__r4r8rBr-r%r/r/Hs&?r-r/cBeZdZUdZeZded<iZded< ddee e e e e e ffe e ee ffdefdZd e j"d e j$d e fd Zd Zde de d e e e ffdZdddddddeej0d e j"d e j$d dfdZy) AsyncToSynca Utility class which turns an awaitable that only works on the thread with the event loop into a synchronous callable that works in a subthread. If the call stack contains an async loop, the code runs there. Otherwise, the code runs in a new loop in a new thread. Either way, this thread then pauses and waits to run any thread_sensitive code called from further down the call stack using SyncToAsync, before finally exiting once the async task returns. r executorsz6Dict[asyncio.AbstractEventLoop, CurrentThreadExecutor]loop_thread_executors awaitableforce_new_loopc\t|r!t|s-tt|d|stjdd||_ |j j |_||_d|_ tj|_ y#t$rY4wxYw#t$rYywxYw)N__call__z4async_to_sync was passed a non-async-marked callable) stacklevel) callableiscoroutinefunctiongetattrwarningswarnrL__self__AttributeErrorrMmain_event_loopr*get_running_loop RuntimeError)r3rLrMs r%r4zAsyncToSync.__init__s "#I.' :y(QR MMFST #  NN33DM-# #*#;#;#=D        s$ B6B BB B+*B+argskwargsrc&d}|js`|jsTttjdd}|r7|t j k(r ttjdd|_ tj}|jr tdt}t|jdd}t}||j_tj g} ttjdd} d} |j"|t%j&| | g|i|} |jr|jjsptj(} ||j*| <t-d} | j/|j0| | }|r|j3||j5nC|jj7|jj8| |r|j3|| |j*| =t;| d ||j_|j5S#t$rYwxYw#| |j*| =t;| d ||j_wxYw) NTmain_event_loop_pidrYznYou cannot use AsyncToSync in the same thread as an async event loop - just await the async function directly.current task_contextr max_workersr)rMrYrTr6 threadlocalosgetpidr*rZ is_runningr[rrJrr` contextvars copy_context main_wrapsysexc_infonew_event_looprKrsubmit_run_event_looprun_until_futureresultcall_soon_threadsafe create_taskr&)r3r\r]__traceback_hide__r_ event_loop call_result old_executorcurrent_executorrralooprL loop_executor loop_futures r%rOzAsyncToSync.__call__sn!""4+?+?#*'')>#  #':biik'I'.++->($  113J$$&"> %+H t~~y$? 02!1++-.{66M % 2&      I((T-A-A-L-L-N--/3C**40 2q A +22(($  $$55kB""$$$99((44i$$55kB..t4 WQZ (%1DNN "!!##O   @..t4 WQZ (%1DNN "s0IDI  II 0Jc tj| |j| tj|D]}|j fd}|j|D]G}|j r|j %|jd|j |dIt|dr|j|j|jtj|jy#|jtj|jwxYw# tj|D]}|j fd}|j|D]G}|j r|j %|jd|j |dIt|dr|j|j|jtj|jw#|jtj|jwxYwxYw)zP Runs the given event loop (designed to be called in a thread). cLKtjddid{y7w)Nreturn_exceptionsT)r*gather)taskssr%rz+AsyncToSync._run_event_loop..gathers!..%H4HHHs $"$Nz(unhandled exception during loop shutdown)message exceptiontaskshutdown_asyncgens) r*set_event_looprun_until_complete all_taskscancel cancelledrcall_exception_handlerhasattrrcloserY)r3rycororrrs @r%rozAsyncToSync._run_event_loops t$ =  # #D ) = ))$/!"DKKM"I''1! D~~' ~~'333+U-1^^-=(, 4!56++D,C,C,EF &&t';';< &&t';';<5 = ))$/!"DKKM"I''1! D~~' ~~'333+U-1^^-=(, 4!56++D,C,C,EF &&t';';< &&t';';E A-DAD1E  I0A-H<v>>  " "6 *'L,D##L1$113GAJ%C#14>>4#B6#BBBFC? )  % %a ( ( ) 'L,D##L1$113GAJs|c>C>HRV4D>(4!(4(4: (4 k))* (4 ww (4))(4 (4r-rIc heZdZUdZej ZedZe jdZ de d<e jdZ de d<ejZde d < dd eeefd ed eddd fdZdej.dej0defdZdededeeeeeefffdZdZy )r6a Utility class which turns a synchronous callable into an awaitable that runs in a threadpool. It also sets a threadlocal inside the thread so calls to AsyncToSync can escape it. If thread_sensitive is passed, the code will run in the same thread as any outer code. This is needed for underlying Python code that is not threadsafe (for example, code which handles SQLite database connections). If the outermost program is async (i.e. SyncToAsync is outermost), then this will be a dedicated single sub-thread that all sync code runs in, one after the other. If the outermost program is sync (i.e. AsyncToSync is outermost), this will just be the main thread. This is achieved by idling with a CurrentThreadExecutor while AsyncToSync is blocking its sync parent, rather than just blocking. If executor is passed in, that will be used instead of the loop's default executor. In order to pass in an executor, thread_sensitive must be set to False, otherwise a TypeError will be raised. rrbr7z.contextvars.ContextVar[ThreadSensitiveContext]deadlock_contextzcontextvars.ContextVar[bool]zEweakref.WeakKeyDictionary[ThreadSensitiveContext, ThreadPoolExecutor]r:Nr(thread_sensitiverArrc>t|r!t|stt|d|r td||_t j ||||_t||r | td||_ |j|_ y#t$rYywxYw)NrOz4sync_to_async can only be applied to sync functions.z6executor must not be set when thread_sensitive is True) rRrSrT TypeErrorr(rr_thread_sensitiver' _executorrWrX)r3r(rrAs r%r4zSyncToAsync.__init__s"4("74T#BCRS S   t,!1d#  4TU U!  MMDM   s>B BBr\r]c Kd}tj}|jrttj dd}|r|}n|j jdrT|j j}||jvr|j|}ntd}||j|<n|tjvrtj|}nZ|jjdr td|j}|jjdn |j}t!j"}t%j&|j(g|i|} |j*} g} |j-|t%j&|j.|t1j2| | | } tj4| d{} t?||jjd| S7,#tj6$rd} | d}|j9 |d{7d}n#tj6$rYnwxYwn#t:$rYnwxYw| j=r|r| j9| d{7} YwxYw#t?||jjdwxYww)NTr`rrbFz9Single thread executor already being used, would deadlockr) r*rZrrTrIrJr7r r:rrKrr[single_thread_executorr!rrhrirrr(runrun_in_executorthread_handlerrkrlshieldCancelledErrorr IndexErrordoner&)r3r\r]rtrycurrent_thread_executorrAr7rchildr(ra exec_cororet cancel_parentrs r%rOzSyncToAsync.__call__s!'')  ! !&-k.C.CYPT&U #&2..2248,0+H+H+L+L+N(+t/N/NN#>>?WXH 2a@HPXD334LM:::&<K HHH(K HJI4I9H<:IIIIIIJ I(%J'I((+JJJJJJ(KK rrctj|j|}tj||jSr)rrrOrr(rs r%rzSyncToAsync.__get__s1   7''dii88r-cd}||j_tj|j_||j_|dr |d||i|S#t $r ||i|cYSwxYw)zE Wraps the sync application with exception handling. Tr)rdrYrerfr_rar)r3ryrlrar(r\r]rts r%rzSyncToAsync.thread_handlers ",0(/1yy{,(4% A; -qk!(( (! -T,V,, -sAA/.A/)TN)rCrDrErF threadinglocalrdrrrh ContextVarr7rrweakrefWeakKeyDictionaryr:rrrrr r4r\r]rOrr rrrGr-r%r6r6Vs",")//#K0A>  9:N 8N{7M7M84 "!!# g"&37 r2v/0   0NBGGNryyNRN`99$'9 "iS" -- .9)r-r6FrMrMcyrrGrs r% async_to_syncr sr-rLcyrrGrLrMs r%rrsr-c,|fdSt|S)Nct|S)NrrI)frMs r%zasync_to_sync..5s ) r-rrrs `r%rr$s*    % r-TrrArrArcyrrGrs r% sync_to_asyncr?s r-cyrrGr(rrAs r%rrHsr-c2|fdSt|S)Nc t|S)Nrr6)rrArs r%rzsync_to_async..\s - r-rrrs ``r%rrRs, |   ) r-r)2r*asyncio.coroutinesrhrinspectrerkrrUrconcurrent.futuresrrtypingrrrrr r r r r rrrrrrr version_infortyping_extensions _typeshedrrrrrr&rrSr'rr/rIr6rrrrGr-r%rsM 9    ;w +$ T#s(+,t_ T] k11 d " 7+,!55070M0M8RD"H-M!55B2 &?&?Rb4'"b&/b4Jt)'"b&/t)n ! 8B #sB,// 0(2y};L2M MNO RV  ! YsC|,,-Yr]"# %  b"f " ! R3R<00 1 R2& ' )   xIc3l334hr9R=?P6QQ RSR  RV 6 "/3+,xB (2yc2/F+F"GGH   "/3  2r6 +,  b)CbL))*  (,"/3  8BF# $+,    hr2v "iS" .E*E!F FG R3R<( ()+ r-