B 0*Щ\∙vу@s<dZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl mZddl mZddl m Z ddlmZd=d d ДZd>ddДZd?d dДZGddДde jГZeZyddlZWnek r╠YnXejZZddДZej jZej jZej jZddedЬddДZddДZddЬddДZddДZddДZ dddЬd d!ДZ!ejd"d#ДГZ"d@ddЬd$d%ДZ#ddЬd&d'ДZ$ed(d)ДГZ%Gd*d+Дd+e j&ГZ'dd,d-Ьd.d/ДZ(ddЬd0d1ДZ)d2d3ДZ*eа+бZ,iZ-d4d5ДZ.d6d7ДZ/d8d9ДZ0d:d;ДZ1e.Z2e1Z3e/Z4e0Z5y$dd(szall_tasks..)rr┌list┌ _all_tasks)rr)rr r"srcs&ИdkrtабЙЗfddДttГDГS)Ncsh|]}tа|бИkr|ТqSr)rr!)r#r$)rrr r%4sz$_all_tasks_compat..)r┌get_event_loopr&r')rr)rr ┌_all_tasks_compat,sr)csвeZdZdZedddДГZedddДГZddЬЗfdd Д ZЗfd dДZdd ДZ ddДZ ddДZddЬddДZdddЬddДZ ddДZd ЗfddД ZddДZЗZS)!rTNcCs(tjdtddН|dkr tаб}t|ГS)NzETask.current_task() is deprecated, use asyncio.current_task() insteadщ)┌ stacklevel)┌warnings┌warn┌PendingDeprecationWarningrr(r)┌clsrrrr rIszTask.current_taskcCstjdtddНt|ГS)Nz?Task.all_tasks() is deprecated, use asyncio.all_tasks() insteadr*)r+)r,r-r.r))r/rrrr rYszTask.all_tasks)rcsvtГj|dН|jr|jd=tа|бs:d|_td|ЫЭГВd|_d|_||_ t аб|_|j j|j|jdНt|ГdS)N)rщ Fza coroutine was expected, got )┌context)┌super┌__init__┌_source_tracebackr┌iscoroutine┌_log_destroy_pending┌ TypeError┌_must_cancel┌_fut_waiter┌_coro┌contextvarsZcopy_context┌_context┌_loop┌ call_soon┌_Task__stepr)┌self┌coror)┌ __class__rr r3es z Task.__init__csF|jtjkr8|jr8|ddЬ}|jr,|j|d<|jа|бtГабdS)Nz%Task was destroyed but it is pending!)┌task┌messageZsource_traceback) Z_staterZ_PENDINGr6r4r=Zcall_exception_handlerr2┌__del__)r@r1)rBrr rEws zTask.__del__cCs tа|бS)N)rZ_task_repr_info)r@rrr ┌ _repr_infoВszTask._repr_infocCstdГВdS)Nz*Task does not support set_result operation)┌RuntimeError)r@┌resultrrr ┌ set_resultЕszTask.set_resultcCstdГВdS)Nz-Task does not support set_exception operation)rG)r@┌ exceptionrrr ┌ set_exceptionИszTask.set_exception)┌limitcCstа||бS)N)rZ_task_get_stack)r@rLrrr ┌ get_stackЛszTask.get_stack)rL┌filecCstа|||бS)N)rZ_task_print_stack)r@rLrNrrr ┌print_stackвs zTask.print_stackcCs4d|_|абrdS|jdk r*|jабr*dSd|_dS)NFT)Z_log_tracebackr"r9┌cancelr8)r@rrr rPнs zTask.cancelc s╪|абrtаd|Ыd|ЫЭбВ|jr>t|tjГs8tаб}d|_|j}d|_t|j |ГРzhy"|dkrp|а dб}n |а|б}Wn╥tk r╠}z0|jrмd|_t Гаtаббnt Га|jбWdd}~XYРnЄtjk rьt ГабYРn╥tk Рr}zt Га|бWdd}~XYРnвtk РrN}zt Га|бВWdd}~XYРnpXt|ddГ}|dk РrBtа|б|j k Рrжtd|Ыd|ЫdЭГ}|j j|j||jdНnЪ|Рr||kРr▄td |ЫЭГ}|j j|j||jdНn8d|_|j|j|jdН||_|jРr@|jабРr@d|_n*td |Ыd|ЫЭГ}|j j|j||jdНn||dkРrb|j j|j|jdНn\tа|бРrЪtd|Ыd|ЫЭГ}|j j|j||jdНn$td |ЫЭГ}|j j|j||jdНWdt|j |Гd}XdS)Nz_step(): already done: z, F┌_asyncio_future_blockingzTask z got Future z attached to a different loop)r1zTask cannot await on itself: z-yield was used instead of yield from in task z with z;yield was used instead of yield from for generator in task zTask got bad yield: ) r"rZInvalidStateErrorr8┌ isinstance┌CancelledErrorr:r9rr=┌send┌throw┌ StopIterationr2rKrI┌valuerP┌ Exception┌ BaseException┌getattrr!rGr>r?r<rQ┌add_done_callback┌ _Task__wakeup┌inspectZisgeneratorr)r@┌excrArHZblocking┌new_exc)rBrr Z__step╬s| zTask.__stepc CsJy|абWn,tk r8}z|а|бWdd}~XYn X|абd}dS)N)rHrXr?)r@┌futurer^rrr Z__wakeup"sz Task.__wakeup)N)N)N)┌__name__┌ __module__┌__qualname__r6┌classmethodrrr3rErFrIrKrMrOrPr?r\┌ __classcell__rr)rBr r7s!TrcCstаб}|а|бS)N)rrr)rArrrr r?sr)r┌timeout┌return_whencГsКtа|бstа|бr(tdt|ГjЫЭГВ|s4tdГВ|tt t fkrPtd|ЫЭГВИdkr`tабЙЗfddДt |ГDГ}t|||ИГIdHS)Nzexpect a list of futures, not z#Set of coroutines/Futures is empty.zInvalid return_when value: csh|]}t|ИdНТqS))r)r)r#┌f)rrr r%iszwait..)r┌isfuturerr5r7┌typera┌ ValueErrorrrrrr(┌set┌_wait)┌fsrrfrgr)rr rOsrcGs|абs|аdбdS)N)r"rI)┌waiter┌argsrrr ┌_release_waiternsrq)rc├s|dkrtаб}|dkr"|IdHS|dkrVt||dН}|абrF|абS|абtабВ|аб}|а |t |б}tаt |б}t||dН}|а |бzry|IdHWn*tjk r╚|а|б|абВYnX|абr┌|абS|а|бt||dНIdHtабВWd|абXdS)Nr)r)rr(rr"rHrPr┌TimeoutError┌ create_future┌ call_laterrq┌ functools┌partialr[rS┌remove_done_callback┌_cancel_and_wait)┌futrfrro┌timeout_handle┌cbrrr rss8 rcГs┬|абЙdЙ|dk r"|а|tИбЙt|ГЙЗЗЗЗfddД}x|D]}|а|бqBWzИIdHWdИdk rtИабXtГtГ}}x4|D],}|а|б|абrм|а |бqК|а |бqКW||fS)NcsZИd8ЙИdks4Иtks4ИtkrV|абsV|абdk rVИdk rDИабИабsVИаdбdS)Nrr)rr┌ cancelledrJrPr"rI)rh)┌counterrgrzrorr ┌_on_completion╕sz_wait.._on_completion) rsrtrq┌lenr[rPrlrwr"┌add)rnrfrgrr~rhr"Zpendingr)r}rgrzror rmмs& rmc ├sF|аб}tаt|б}|а|бz|аб|IdHWd|а|бXdS)N)rsrurvrqr[rPrw)ryrror{rrr rx╫s rx)rrfc#sъtа|бstа|бr(tdt|ГjЫЭГВИdk r4ИntабЙЗfddДt |ГDГЙddl m}|ИdНЙdЙЗЗЗfddД}ЗЗЗfd d ДЙЗfddД}xИD]}|аИбqЮWИr╚|dk r╚Иа ||бЙxttИГГD]}|ГVq╓WdS) Nzexpect a list of futures, not csh|]}t|ИdНТqS))r)r)r#rh)rrr r%¤szas_completed..r)┌Queue)rcs.x ИD]}|аИбИаdбqWИабdS)N)rw┌ put_nowait┌clear)rh)r~r"┌todorr ┌_on_timeouts z!as_completed.._on_timeoutcs4ИsdSИа|бИа|бИs0Иdk r0ИабdS)N)┌removerВrP)rh)r"rzrДrr r~s z$as_completed.._on_completioncУs$ИабIdH}|dkrtjВ|абS)N)rrrrrH)rh)r"rr ┌ _wait_for_onesz#as_completed.._wait_for_one)rrirr5r7rjrarr(rlZqueuesrБr[rt┌ranger)rnrrfrБrЕrЗrh┌_r)r~r"rrzrДr rшs rccs dVdS)Nrrrrr ┌__sleep0s rКc├s\|dkrtГIdH|S|dkr(tаб}|аб}|а|tj||б}z |IdHS|абXdS)Nr)rКrr(rsrtrZ_set_result_unless_cancelledrP)ZdelayrHrr`┌hrrr r +s r cCsИtа|бr6|dkrtаб}|а|б}|jr2|jd=|Stа|бrb|dk r^|tа|бk r^t dГВ|St а|бr|tt |Г|dНStdГВdS)Nr0z$loop argument must agree with Future)rz:An asyncio.Future, a coroutine or an awaitable is required)rr5rr(rr4rrir!rkr]Zisawaitabler┌_wrap_awaitabler7)Zcoro_or_futurerrCrrr r=s rccs|абEdHS)N)┌ __await__)Z awaitablerrr rМTsrМcs*eZdZddЬЗfddД ZddДZЗZS)┌_GatheringFutureN)rcstГj|dН||_d|_dS)N)rF)r2r3┌ _children┌_cancel_requested)r@┌childrenr)rBrr r3fsz_GatheringFuture.__init__cCs:|абrdSd}x|jD]}|абrd}qW|r6d|_|S)NFT)r"rПrPrР)r@┌retZchildrrr rPksz_GatheringFuture.cancel)rarbrcr3rPrerr)rBr rО^srОF)r┌return_exceptionscs╬|s*|dkrtаб}|абЙИаgбИSЗЗЗЗЗfddД}i}gЙdЙdЙxn|D]f}||krиt||dН}|dkr~tа|б}||k rМd|_Иd7Й|||<|а|бn||}Иа |бqTWt И|dНЙИS)Ncs╓Иd7ЙИабr$|абs |абdSИsd|абrFtаб}Иа|бdS|аб}|dk rdИа|бdSИИkr╥g}x@ИD]8}|абrМtаб}n|аб}|dkrд|аб}|а|бqvWИjr╚Иаtаббn Иа |бdS)Nr) r"r|rJrrSrKrH┌appendrРrI)ryr^┌results┌res)rС┌ nfinished┌nfuts┌outerrУrr ┌_done_callbackШs4 zgather.._done_callbackr)rFr)rr(rsrIrrr!r6r[rФrО)rrУZcoros_or_futuresrЪZ arg_to_fut┌argryr)rСrЧrШrЩrУr r zs2 1 r csDt||dН}|абr|Stа|б}|абЙЗfddД}|а|бИS)N)rcs\Иабr|абs|абdS|абr.Иабn*|аб}|dk rJИа|бnИа|аббdS)N)r|rJrPrKrIrH)┌innerr^)rЩrr rЪs zshield.._done_callback)rr"rr!rsr[)rЫrrЬrЪr)rЩr rчs rcs:tаИбstdГВtjабЙЗЗЗfddД}Иа|бИS)NzA coroutine object is requiredc sTytаtИИdНИбWn6tk rN}zИабr<Иа|бВWdd}~XYnXdS)N)r)rZ _chain_futurerrXZset_running_or_notify_cancelrK)r^)rAr`rrr ┌callback%s z*run_coroutine_threadsafe..callback)rr5r7┌ concurrentr┌FutureZcall_soon_threadsafe)rArrЭr)rAr`rr r s r cCstа|бdS)N)r'rА)rCrrr r9srcCs4tа|б}|dk r(td|Ыd|ЫdЭГВ|t|<dS)NzCannot enter into task z while another task z is being executed.)rrrG)rrCrrrr r>s rcCs2tа|б}||k r(td|Ыd|ЫdЭГВt|=dS)Nz Leaving task z! does not match the current task ┌.)rrrG)rrCrrrr rFs rcCstа|бdS)N)r'┌discard)rCrrr rNsr)rrrrr'r)N)N)N)N):┌__all__┌concurrent.futuresrЮr;rur]┌typesr,┌weakref┌rrrrrrrr)Z _PyFuturerZ_PyTaskZ_asyncio┌ImportErrorZ_CTaskrrrrrrqrrmrxrrКr rrМrЯrОr rr ZWeakSetr'rrrrrZ_py_register_taskZ_py_unregister_taskZ_py_enter_taskZ_py_leave_taskZ_c_register_taskZ_c_unregister_taskZ _c_enter_taskZ _c_leave_taskrrrr ┌sv } 9+7 m5$