]g
@ s= d d d d d d d d d d d
d d g
Z d
d l Z d
d l Z d
d l Z d
d l Z d
d l Z d
d l Z d
d l Z d d l m
Z
d d l m Z d d l m Z d d l m
Z
d d l m Z Gd d d e
j Z e j
j Z e j
j Z e j
j Z e d d d d d e d d Z d d Z e d d d d Z e d d Z d d d d d d Z e d d d d d Z d d d! d" Z e e d C s z!Task.all_tasks..)r r
_all_tasks)r r r )r r all_tasks; s zTask.all_tasksr c sk t j d | | j r& | j d =| | _ d | _ d | _ | j j | j | j j
j | d S)Nr r F) super__init___source_traceback_coro _fut_waiter _must_cancelr call_soon_step __class__r! add)selfcoror )r, r r r% E s
z
Task.__init__c C sg | j t j k rS | j rS d | d d i } | j rC | j | d <| j j | t j j | d S)Ntaskmessagez%Task was destroyed but it is pending!Zsource_traceback) _stater Z_PENDING_log_destroy_pendingr& r Zcall_exception_handlerFuture__del__)r. contextr r r r5 T s
z Task.__del__c sr t j } | j r" d | d z wait_for=%r)r$
_repr_infor) r Z_format_coroutiner' insertr( )r. infor/ )r, r r r8 _ s
zTask._repr_infolimitc C s
g } y | j j } Wn t k
r6 | j j } Yn X| d k r xI | d k r | d k ru | d k rk P| d 8} | j | | j } qF W| j nj | j d k r| j j } xL | d k r| d k r | d k r P| d 8} | j | j | j
} q W| S)Nr r ) r' cr_frameAttributeErrorgi_frameappendf_backreverse
_exception
__traceback__tb_frametb_next)r. r; Zframesftbr r r get_stackm s0
zTask.get_stackfilec C s] g } t } x | j d | D] } | j } | j } | j } | j } | | k rr | j | t j | t j | | | j
}
| j | | | |
f q" W| j } | s t
d | d | n7 | d k r t
d | d | n t
d | d | t j | d | | d k rYx3 t j | j | D] }
t
|
d | d d q9Wd S)Nr; zNo stack for %rrI z)Traceback for %r (most recent call last):z%Stack for %r (most recent call last):end )setrH f_linenof_code co_filenameco_namer- linecache
checkcachegetline f_globalsr? rB print traceback
print_listformat_exception_onlyr, ) r. r; rI extracted_listZcheckedrF linenocofilenamenamelineexcr r r print_stack s0
zTask.print_stackc C sH d | _ | j r d S| j d k r; | j j r; d Sd | _ d S)NFT)Z_log_tracebackdoner( cancelr) )r. r r r rb s z Task.cancelc s | j r0 t | t j s' t j } d | _ | j } d | _ | | j j | j .)r isfuturer iscoroutine TypeErrortyperx
ValueErrorr r r rp r r rL _wait)fsr r~ r r )r r r L s c G s | j s | j d d S)N)ra rj )waiterargsr r r _release_waiterl s r c c s
| d k r t j } | d k r- | Ed HS| j } | j | t | } t j t | } t | d | } | j | z| y
| Ed HWn/ t j
k
r | j | | j Yn X| j
r | j S| j | | j t j Wd | j Xd S)Nr )r r
create_future
call_laterr functoolspartialr rq r re remove_done_callbackrb ra ru TimeoutError)futr~ r r timeout_handlecbr r r r q s,
c # s | j d | d k r3 | j | t t | f d d } x | D] } | j | qa Wz
Ed HWd d k r j Xt t } } xD | D]< } | j | | j r | j | q | j | q W| | f S)Nc s d 8 d k sM t k sM t k r| | j r| | j d k r| d k rc j j s| j d d S)Nr r )r r cancelled exceptionrb ra rj )rF )counterr r r r r _on_completion s
z _wait.._on_completion)
r r r lenrq rb rL r ra r- )r r~ r r r rF ra pendingr )r r r r r r s&
r c # sL t j | s t j | r7 t d t | j d k rI n t j f d d t | D d d l
m } | d d f d d } f d d
t f d d } x D] } | j
q W r#| d k r# j | | x" t t D] } | Vq6Wd S)
Nz expect a list of futures, not %sc s" h | ] } t | d q S)r )r )r rF )r r r r s z as_completed..r )Queuer c s9 x( D] } | j j d q W j d S)N)r
put_nowaitclear)rF )r ra todor r _on_timeout s
z!as_completed.._on_timeoutc sE s
d S j | j | rA d k rA j d S)N)remover rb )rF )ra r r r r r s
z$as_completed.._on_completionc 3 s0 j Ed H} | d k r& t j | j S)N)r r r ru )rF )ra r r
_wait_for_one s z#as_completed.._wait_for_one)r r r r r r rx r r rL queuesr r rq r ranger )r r r~ r r r rF _r )r ra r r r r r s
c
c sv | d k r d V| S| d k r- t j } | j } | j j | t j | | } z
| Ed HSWd | j Xd S)Nr )r r r r r r Z _set_result_unless_cancelledrb )delayru r rw hr r r r s
c C s& t j d t d d t | d | S)Nz;asyncio.async() function is deprecated, use ensure_future()
stacklevelr7 r )warningswarnDeprecationWarningr )coro_or_futurer r r r async_ s r c C s t j | r: | d k r6 | | j k r6 t d | St j | r | d k ra t j } | j | } | j r | j d =| St
j r t j
| r t t | d | St d d S)Nz$loop argument must agree with Futurer r z:An asyncio.Future, a coroutine or an awaitable is requiredr# )r r r r r r r r Z create_taskr& r ZPY35rs Z isawaitabler _wrap_awaitabler )r r r0 r r r r + s
c c s | j Ed HS)N) __await__)Z awaitabler r r r B s r c s4 e Z d Z d d f d d Z d d Z S)_GatheringFuturer Nc s t j d | | | _ d S)Nr )r$ r% _children)r. childrenr )r, r r r% T s z_GatheringFuture.__init__c C s@ | j r d Sd } x# | j D] } | j r d } q W| S)NFT)ra r rb )r. retZchildr r r rb X s
z_GatheringFuture.cancel)rx ry rz r% rb r r )r, r r L s r return_exceptionsFc s | s; | d k r t j } | j j g Si x t | D] } t j | s t | d | } | d k r | j } d | _ n9 | } | d k r | j } n | j | k r t
d | | s zgather..r c s j r& | j s" | j d S| j rX t j } s j | d Sn> | j d k r | j } s j | d Sn | j } | | < d 7 k r j d S)Nr ) ra r r r re ri rB _resultrj )ir res) nchildren nfinishedouterresultsr r r _done_callback s&
z gather.._done_callback)r r r rj rL r r r r r3 r r r enumeraterq r r )r r Zcoros_or_futuresr r r r r r )r r r r r r r r
b s8
c sZ t | d | } | j r" | S| j } | j f d d } | j | S)Nr c s~ j r& | j s" | j d S| j r? j n; | j } | d k rg j | n j | j d S)N)r r rb ri rj ru )innerr_ )r r r r s
z shield.._done_callback)r ra r r rq )r r r r r )r r r s
c sS t j s t d t j j f d d } j | S)Nz A coroutine object is requiredc sh y t j t d WnA t k
rc } z! j rN j | WYd d } ~ Xn Xd S)Nr )r Z
_chain_futurer rl set_running_or_notify_cancelri )r_ )r/ rw r r r callback s
z*run_coroutine_threadsafe..callback)r r r
concurrentr r4 Zcall_soon_threadsafe)r/ r r r )r/ rw r r r
s
)#__all__concurrent.futuresr r rs rQ rV r r{ rK r r r r r r4 r r r r r r r r r r r globalsrx r r r r
r r
r r r r sN 0 --8
T5