!A?h d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z G d d ej Z
G d
d Zdd d
Zd ZdS ))Runnerrun N )
coroutines)events)
exceptions)tasksc e Zd ZdZdZdZdS )_Statecreated initializedclosedN)__name__
__module__ __qualname__CREATED INITIALIZEDCLOSED n )
! $ r c . | | S N)
_lazy_initr# s r __enter__zRunner.__enter__: s r c . | d S r& )close)r# exc_typeexc_valexc_tbs r __exit__zRunner.__exit__> s
r c | j t j urdS | j }t | | | | | | j rt j
d | d| _ t j | _ dS # | j rt j
d | d| _ t j | _ w xY w)z Shutdown and close event loop.N)
r r r r _cancel_all_tasksrun_until_completeshutdown_asyncgensshutdown_default_executorr" r set_event_loopr+ r )r# loops r r+ z Runner.closeA s ;f 0 0 0 F
(:D d # # # # #D$;$;$=$= > > > # #D$B$B$D$D E E E#
,%d+++ JJLLL DJ -DKKK #
,%d+++ JJLLL DJ -DK ' ' ' 's
A$C A D
c 8 | | j S )zReturn embedded event loop.)r' r r( s r get_loopzRunner.get_loopQ s zr contextc t j | s"t d | t j t
d | || j }| j
|| }t j t j
u rxt j t j t j u rNt% j | j | } t j t j | n# t $ r d}Y nw xY wd}d| _ | j | |Jt j t j |u r+t j t j t j S S S # t. j $ r< | j dk r/t3 |dd }| | dk rt5 w xY w# |Jt j t j |u r+t j t j t j w w w xY w)z/Run a coroutine inside the embedded event loop.z"a coroutine was expected, got {!r}Nz7Runner.run() cannot be called from a running event loopr9 ) main_taskr uncancel)r iscoroutine
ValueErrorformatr _get_running_loop RuntimeErrorr' r r create_task threadingcurrent_thread main_threadsignal getsignalSIGINTdefault_int_handler functoolspartial
_on_sigintr! r2 r CancelledErrorgetattrKeyboardInterrupt)r# coror: tasksigint_handlerr= s r r z
Runner.runV s5 %d++ P A H H N NOO O # % % 1 IK K
K
?mGz%%dG%<< $ & &)*?*A*A A A //63MMM & .t$ O O ON
&
fm^<<<<
&
&
& "&
& "N ! I:0066 *$V]33~EE
fmV-GHHHH +E ( $q(("4T::'HHJJ!OO+--- *$V]33~EE
fmV-GHHHH +Es, > D D-,D-:F! !A G,,G/ /AH>c | j t j u rt d | j t j u rd S | j @t
j | _ | j s t
j
| j d| _ n| | _ | j | j | j t j | _ t j | _ d S )NzRunner is closedT)r r r rB r r r new_event_loopr r" r5 r set_debug contextvars copy_contextr r( s r r' zRunner._lazy_init s ;&- ' ' 122 2 ;&, , , F % .00DJ'
, %dj111'+$ ++--DJ ; " J - - -#022
( r c | xj dz
c_ | j dk rE| s1| | j d d S t )Nr c d S r& r r r r