o ;i @sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddlm Z mZmZmZmZddlmZddlmZejdkrIejZnejjZer]dd lmZdd lmZmZGd d d e dZGd ddeZ ddde!fddZ"dddedddfddZ#d'ddZ$de%dedde e&ddfddZ'edej(ej)fddedd eej*d!ee%de dfd"d#Z+d$egefdegeffd%d&Z,dS)(N) TracebackType) TYPE_CHECKINGOptionalSetTypeContextManagerList)Iterator CollectionCallableAnycast)wraps)contextmanager))Deadline) IServable IClosablec@seZdZUdZdZeeed<dZee ed<dZ ee ed<dddZ dd d Z d ee ed eed eeddfddZdeddfddZdS)WrapperaDSpecial wrapper for coroutines to wake them up in case of some error. Example: .. code-block:: python3 w = Wrapper() async def blocking_call(): with w: await asyncio.sleep(10) # and somewhere else: w.cancel(NoNeedToWaitError('With explanation')) N_error cancelled cancel_failedreturncCs t|_dSN)set_tasksselfr A/home/ubuntu/.local/lib/python3.10/site-packages/grpclib/utils.py__init__. zWrapper.__init__cCs6|jdur|jt}|durtd|j|dS)NzCalled not inside a task)r _current_task RuntimeErrorradd)rtaskr r r! __enter__1s zWrapper.__enter__exc_typeexc_valexc_tbcCs:t}|sJ|j||jdur|tju|_|jdSr)r$rdiscardrasyncioCancelledErrorr)rr)r*r+r'r r r!__exit__;s   zWrapper.__exit__errorcCs$||_|jD]}|qd|_dS)NT)rrcancelr)rr0r'r r r!r1Hs   zWrapper.cancelrN)__name__ __module__ __qualname____doc__rr Exception__annotations__rboolrr"r(r BaseExceptionrr/r1r r r r!rs"      rc@s*eZdZdZedddedfddZdS)DeadlineWrappera{Deadline wrapper to specify deadline once for any number of awaiting method calls. Example: .. code-block:: python3 dw = DeadlineWrapper() with dw.start(deadline): await handle_request() # somewhere during request handling: async def blocking_call(): with dw: await asyncio.sleep(10) deadlinerrNc#sZ|}|s tddfdd }t}|||}z dVW|dS|w)NDeadline exceededrcstddS)Nr=)r1r- TimeoutErrorr rr r!callbackisz'DeadlineWrapper.start..callbackr2)time_remainingr-r>get_event_loop call_laterr1)rr<timeoutr?looptimerr rr!startcs  zDeadlineWrapper.start)r3r4r5r6rr rFr r r r!r;Osr;servicerrcCs6|}tt|d}|dusJ|d\}}}|S)N/) __mapping__nextitersplit)rGmethods method_name_ service_namer r r! _service_namets  rQsig_numsignal.Signalsserversrc CsDd}|D]}z|Wqtyd}Yqw|r t|dSdS)NFT)closer% _second_stage)rRrTfailserverr r r! _first_stage|s   rYcCs td|)N) SystemExit)rRr r r!rVr#rVflagcCs4|r ttd|dSttd|||ddS)NrST)rVr rYappend)rRrTr\r r r! _exit_handlersr^)rDsignalsrDr_c cs|r tjdtdd|pt}t|}g}|D] }||t|||qzdVW|D]}||q,dS|D]}||q8w)aUtility context-manager to help properly shutdown server in response to the OS signals By default this context-manager handles ``SIGINT`` and ``SIGTERM`` signals. There are two stages: 1. first received signal closes servers 2. subsequent signals raise ``SystemExit`` exception Example: .. code-block:: python3 async def main(...): ... with graceful_exit([server]): await server.start(host, port) print('Serving on {}:{}'.format(host, port)) await server.wait_closed() print('Server closed') First stage calls ``server.close()`` and ``await server.wait_closed()`` should complete successfully without errors. If server wasn't started yet, second stage runs to prevent server start. Second stage raises ``SystemExit`` exception, but you will receive ``asyncio.CancelledError`` in your ``async def main()`` coroutine. You can use ``try..finally`` constructs and context-managers to properly handle this error. This context-manager is designed to work in cooperation with :py:func:`python:asyncio.run` function: .. code-block:: python3 if __name__ == '__main__': asyncio.run(main()) :param servers: list of servers :param loop: (deprecated) asyncio-compatible event loop :param signals: set of the OS signals to handle .. note:: Not supported in Windows zHThe loop argument is deprecated and scheduled for removal in grpclib 0.5) stacklevelN) warningswarnDeprecationWarningr-rAradd_signal_handlerr^remove_signal_handler)rTrDr_r\rRr r r! graceful_exits"4   rgfunccstdtffdd }|S)Nrcs*zjWSty_jYSwr) __result__AttributeErrorr rhr r!wrappers   z_cached..wrapper)rr )rhrlr rkr!_cachedsrm)rRrSrN)-syssignalr-rbtypesrtypingrrrrrrr r r r r functoolsr contextlibr version_info current_taskr$Taskmetadatar_typingrrrr;strrQrYrVintr9r^SIGINTSIGTERMAbstractEventLooprgrmr r r r!s`      7%     &D