M/edZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZej&eZej,d k7rrej.gZej2ej4ej6ej8fD]3Zej<eej>k7s#ejAe5ngZGddZ!Gdde!Z"y)zBRegisters functions to be called if an exception or signal occurs.N) TracebackType)Any)Callable)Dict)List)Optional)Type)Union)errors)osntceZdZdZdedefdededdfdZdd Zd ee e d ee d ee de fd Z dedefdededdfdZddZddZddZdededdfdZddZy) ErrorHandleraGContext manager for running code that must be cleaned up on failure. The context manager allows you to register functions that will be called when an exception (excluding SystemExit) or signal is encountered. Usage:: handler = ErrorHandler(cleanup1_func, *cleanup1_args, **cleanup1_kwargs) handler.register(cleanup2_func, *cleanup2_args, **cleanup2_kwargs) with handler: do_something() Or for one cleanup function:: with ErrorHandler(func, args, kwargs): do_something() If an exception is raised out of do_something, the cleanup functions will be called in last in first out order. Then the exception is raised. Similarly, if a signal is encountered, the cleanup functions are called followed by the previously received signal handler. Each registered cleanup function is called exactly once. If a registered function raises an exception, it is logged and the next function is called. Signals received while the registered functions are executing are deferred until they finish. func.argskwargsreturnNc|d|_d|_g|_i|_g|_||j |g|i|yyNF)call_on_regular_exit body_executedfuncs prev_handlersreceived_signalsregisterselfrrrs A/usr/lib/python3/dist-packages/certbot/_internal/error_handler.py__init__zErrorHandler.__init__PsL$)!".0 DF+-   DMM$ 0 0 0 c2d|_|jyr)r_set_signal_handlers)rs r __enter__zErrorHandler.__enter__Ys" !!#r exec_type exec_valuetracec d|_d}|tur|S||jsq|S|tjur#t j d|jd}n:t j ddjtj||||j|j|j|S)NTFzEncountered signals: %szEncountered exception: %s)r SystemExitrr SignalExitloggerdebugrjoin tracebackformat_exception_call_registered_reset_signal_handlers _call_signals)rr$r%r&retvals r__exit__zErrorHandler.__exit__]s"  "M  ,, &++ + LL2D4I4I JF LL5rww**9j%H8J K  ##%  r ch|jjtj|g|i|y)zSets func to be run with the given arguments during cleanup. :param function func: function to be called in case of an error N)rappend functoolspartialrs rrzErrorHandler.registerts+ )++DB4B6BCr ctjd|jr= |jd|jj|jr"?%m4 /48.DXc3h/DDsDtD <# $c $ $ $)r rc>eZdZdZdedefdededdffd ZxZS) ExitHandlerzContext manager for running code that must be cleaned up. Subclass of ErrorHandler, with the same usage and parameters. In addition to cleaning up on all signals, also cleans up on regular exit. r.rrrNc:t||g|i|d|_y)NT)superrr)rrrr __class__s rrzExitHandler.__init__s" ///$(!r )rSrTrUrVrrr __classcell__)r_s@rr\r\s7 )Xc3h/))s)t))r r\)#rVr7loggingrEr.typesrtypingrrrrrr r certbotr certbot.compatr getLoggerrSr+nameSIGTERMrDSIGHUPSIGQUITSIGXCPUSIGXFSZ signal_coderFSIG_IGNr6rr\rZr rrosH    8 $77d?H v~~8)  6  K (FNN : OOK ( )(Hu)u)p ), )r