idZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z ee Zdddddded efd ee gdfdzd ed ed ejdzdegefdzdededdfdZdS)aThis module contains a network retry loop implementation. Its specifically tailored to handling the Telegram API and its errors. .. versionadded:: 21.11 Hint: It was originally part of the `Updater` class, but as part of #4657 it was extracted into its own module to be used by other parts of the library. Warning: Contents of this module are intended to be used internally by the library and *not* by the user. Changes to this module are not considered breaking changes and may not be documented in the changelog. N)Callable Coroutine) get_logger) InvalidToken RetryAfter TelegramErrorTimedOutF) on_err_cb stop_event is_runningrepeat_on_success action_cb.r descriptionintervalr r max_retriesr returncK|rdkrtdd|d|pd}ddtdtd tffd } dfd } td |} d} |rO | d {V|s#td | dz } d S |} n#t $r=} d}|| jz} | d|d}| | |rYd } ~ nd } ~ wt$r}d} d|d}| | |rYd }~nsd }~wt$rt dt$r>}|r ||| | r| dkrdntdd| z} Yd }~nd }~wwxYw| dz } n #| dz } wxYw| rtj| d {V|Md Sd S)aPerform a loop calling `action_cb`, retrying after network errors. Stop condition for loop in case of ``max_retries < 0``: * `is_running()` evaluates :obj:`False` * `stop_event` is set. * calling `action_cb` succeeds and `repeat_on_success` is :obj:`False`. Additional stop condition for loop in case of `max_retries >= 0``: * a call to `action_cb` succeeds * or `max_retries` is reached. Args: action_cb (:term:`coroutine function`): Network oriented callback function to call. on_err_cb (:obj:`callable`): Optional. Callback to call when TelegramError is caught. Receives the exception object as a parameter. Hint: Only required if you want to handle the error in a special way. Logging about the error is already handled by the loop. Important: Must not raise exceptions! If it does, the loop will be aborted. description (:obj:`str`): Description text to use for logs and exception raised. interval (:obj:`float` | :obj:`int`): Interval to sleep between each call to `action_cb`. stop_event (:class:`asyncio.Event` | :obj:`None`): Event to wait on for stopping the loop. Setting the event will make the loop exit even if `action_cb` is currently running. Defaults to :obj:`None`. is_running (:obj:`callable`): Function to check if the loop should continue running. Must return a boolean value. Defaults to `lambda: True`. max_retries (:obj:`int`): Maximum number of retries before stopping the loop. * < 0: Retry indefinitely. * 0: No retries. * > 0: Number of retries. repeat_on_success (:obj:`bool`): Whether to repeat the action after a successful call. Defaults to :obj:`False`. Raises: ValueError: When passing `repeat_on_success=True` and `max_retries >= 0`. This case is currently not supported. rz7Cannot use repeat_on_success=True with max_retries >= 0zNetwork Retry Loop (z):cdS)NTr_/Users/shanyulin/my-agent/.venv/lib/python3.11/site-packages/telegram/ext/_utils/networkloop.pyz$network_retry_loop..hs$rcurrent_retriesexception_inforc|rd|n}dks|krtd||dStd||dS)aLCheck if max retries reached and log accordingly. Args: current_retries: The current retry count. exception_info: Additional context about the exception (e.g., "Timed out: ..."). Returns: bool: True if max retries reached (should abort), False otherwise (should retry).  rz(%s Failed run number %s of %s. Retrying.Fz(%s Failed run number %s of %s. Aborting.T)_LOGGERdebug exception)rrprefix_with_info log_prefixrs rcheck_max_retries_and_logz5network_retry_loop..check_max_retries_and_logjs@N]j;;>;;;S] ??o ;; MM:      5 6       trNcKsd{VdStj}tj}tj||ftjd{V\}}t jtj5|D]}| dddn #1swxYwY||vrt ddS| dS)N) return_whenz %s Cancelled) asyncio create_taskwaitFIRST_COMPLETED contextlibsuppressCancelledErrorcancelrrresult)action_cb_task stop_taskdonependingtaskrr"r s r do_actionz%network_retry_loop..do_actionsu )++        F ,YY[[99' (9(9:: %l Y 'W5L         g !7 8 8                         MM.* 5 5 5 F s#C  CCz %s Startingz#%s Action succeeded. Stopping loop.g?z . Adding z seconds to the specified time.z Timed out: .z&%s Invalid token. Aborting retry loop.g?)r)rN) ValueErrorintstrboolrrr _retry_after total_secondsr rr rminr&sleep)rr rrr r rr effective_is_runningr#r4 cur_intervalretriesexc slack_timertoe telegram_excr"s` ` ` @rnetwork_retry_looprG,s>nT[A--RSSS7 777J%7,,3UY8        , MM-,,,LG   ).% )++       $  CZPPPB qLGGGG B$LL=   J%(8(F(F(H(HHL #YYjYYYN)(.AA         L13111N)(.AA            F S S S  S S S ( ,'''((11 !- 1 111s2s\?Q7R7RLLLLLL S qLGGGqLGGGGG  .- -- - - - - - - -S   ).).).).).sT9-B2/F2 F <3C4/F4 F DF2F  4FFF  FF)__doc__r&r*collections.abcrrtelegram._utils.loggingrtelegram.errorrrrr __name__rr:floatEventr;r9rGrrrrOsS&  ////////......LLLLLLLLLLLL *X   9=(,,0#\.\.\.Y'\.-.5\. \.  \.  $ \.T"T)\.\.\. \.\.\.\.\.\.r