o Zzj@sdZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z ee Zdddddded efd ee gdfdBd ed ed ejdBdegefdBdededdfddZdS)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 returnc s|r dkr tdd|d|pdd}ddtd td tffd d } dfdd } td|} d} |rzz| Id H|sTtdWW| d7} d SWn~ty~} zd}|| j} | d|d}| | |rtWYd } ~ n\d } ~ wt y}zd} d|d}| | |rWYd }~n=d }~wt yt dt y}z|r||| | r| dkrdnt dd| } WYd }~nd }~ww|} W| d7} n| d7} w| rt| Id H|s= 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):cSsdS)NTrrrZ/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telegram/ext/_utils/networkloop.pyhsz$network_retry_loop..current_retriesexception_inforcsN|r d|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_prefixrrrcheck_max_retries_and_logjs  z5network_retry_loop..check_max_retries_and_logNcss IdHdSt}t}tj||ftjdIdH\}}ttj|D]}|q0Wdn1sAwY||vrRt ddS| dS)N) return_whenz %s Cancelled) asyncio create_taskwaitFIRST_COMPLETED contextlibsuppressCancelledErrorcancelrrresult)action_cb_task stop_taskdonependingtask)r rr rr do_actions$      z%network_retry_loop..do_actionz %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_secondsrrrrminr sleep)r r rrr r rr effective_is_runningrr. cur_intervalretriesexc slack_timertoe telegram_excr)r rrr rnetwork_retry_loop,sd 7      !       & rA)__doc__r r$collections.abcrrtelegram._utils.loggingrtelegram.errorrrrr__name__rr4floatEventr5r3rArrrrs<