o /h++@sdZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z z ddl mZdZWn ey7dZYnwddlmZddlmZdd lmZdd lmZejd krYejZn ejd edfd dZeeddZGdddeeZ dS)zeThis module contains an implementation of the BaseRateLimiter class based on the aiolimiter library. N)Any AsyncIteratorCallable CoroutineDictListOptionalUnion) AsyncLimiterTF) get_logger)JSONDict) RetryAfter)BaseRateLimiter) returncCs dVdSNrrrU/var/www/html/govbot/env/lib/python3.10/site-packages/telegram/ext/_aioratelimiter.py null_context.s rAIORateLimiter) class_namec@s<eZdZdZdZ     d&ded ed ed ed ed df ddZd'ddZd'ddZ de e ee fd dfddZ de de e ee fdedeeee e eeefffdedee efd e e eeeff dd Zdedeeee e eeefffdedee efd!e d"ee efd#eed e e eeeffd$d%ZdS)(ra Implementation of :class:`~telegram.ext.BaseRateLimiter` using the library `aiolimiter `_. Important: If you want to use this class, you must install PTB with the optional requirement ``rate-limiter``, i.e. .. code-block:: bash pip install "python-telegram-bot[rate-limiter]" The rate limiting is applied by combining two levels of throttling and :meth:`process_request` roughly boils down to:: async with group_limiter(group_id): async with overall_limiter: await callback(*args, **kwargs) Here, ``group_id`` is determined by checking if there is a ``chat_id`` parameter in the :paramref:`~telegram.ext.BaseRateLimiter.process_request.data`. The ``overall_limiter`` is applied only if a ``chat_id`` argument is present at all. Attention: * Some bot methods accept a ``chat_id`` parameter in form of a ``@username`` for supergroups and channels. As we can't know which ``@username`` corresponds to which integer ``chat_id``, these will be treated as different groups, which may lead to exceeding the rate limit. * As channels can't be differentiated from supergroups by the ``@username`` or integer ``chat_id``, this also applies the group related rate limits to channels. * A :exc:`~telegram.error.RetryAfter` exception will halt *all* requests for :attr:`~telegram.error.RetryAfter.retry_after` + 0.1 seconds. This may be stricter than necessary in some cases, e.g. the bot may hit a rate limit in one group but might still be allowed to send messages in another group. Note: This class is to be understood as minimal effort reference implementation. If you would like to handle rate limiting in a more sophisticated, fine-tuned way, we welcome you to implement your own subclass of :class:`~telegram.ext.BaseRateLimiter`. Feel free to check out the source code of this class for inspiration. .. seealso:: :wiki:`Avoiding Flood Limits ` .. versionadded:: 20.0 Args: overall_max_rate (:obj:`float`): The maximum number of requests allowed for the entire bot per :paramref:`overall_time_period`. When set to 0, no rate limiting will be applied. Defaults to ``30``. overall_time_period (:obj:`float`): The time period (in seconds) during which the :paramref:`overall_max_rate` is enforced. When set to 0, no rate limiting will be applied. Defaults to 1. group_max_rate (:obj:`float`): The maximum number of requests allowed for requests related to groups and channels per :paramref:`group_time_period`. When set to 0, no rate limiting will be applied. Defaults to 20. group_time_period (:obj:`float`): The time period (in seconds) during which the :paramref:`group_max_rate` is enforced. When set to 0, no rate limiting will be applied. Defaults to 60. max_retries (:obj:`int`): The maximum number of retries to be made in case of a :exc:`~telegram.error.RetryAfter` exception. If set to 0, no retries will be made. Defaults to ``0``. ) _base_limiter_group_limiters_group_max_rate_group_time_period _max_retries_retry_after_event<roverall_max_rateoverall_time_periodgroup_max_rategroup_time_period max_retriesrNcCsptstd|r|rt||d|_nd|_|r |r ||_||_nd|_d|_i|_||_t |_ |j dS)NzeTo use `AIORateLimiter`, PTB must be installed via `pip install "python-telegram-bot[rate-limiter]"`.max_rate time_periodr) AIO_LIMITER_AVAILABLE RuntimeErrorr rrrrrasyncioEventrset)selfr"r#r$r%r&rrr__init__s$  zAIORateLimiter.__init__cdSz Does nothing.Nrr/rrr initializezAIORateLimiter.initializecr1r2rr3rrrshutdownr5zAIORateLimiter.shutdowngroup_idr cCsnt|jdkr"|jD]\}}||krq||jr!|j|=q||jvr2t|j|jd|j|<|j|S)Nir') lenrcopyitems has_capacityr(r rr)r/r7keylimiterrrr_get_group_limiters   z!AIORateLimiter._get_group_limiterchatgroupcallback.argskwargsc s|r |jr |jnt}|r|jr||nt}|4IdHD|4IdH'|jIdH||i|IdHWdIdHWdIdHS1IdHsSwYWdIdHdS1IdHsiwYdSr)rrrr>rwait)r/r?r@rArBrC base_context group_contextrrr _run_requestspzAIORateLimiter._run_requestendpointdatarate_limit_argsc sJ|p|j}d}d} |d} | durd} ttt t| } Wdn1s*wYt| tr8| dks=t| tr?| }t |dD]]} zUz|j | ||||dIdHWW|j St y} z+| |krstjd|| d | jd } td | |j t| IdHWYd} ~ nd} ~ wwW|j qE|j wdS) a Processes a request by applying rate limiting. See :meth:`telegram.ext.BaseRateLimiter.process_request` for detailed information on the arguments. Args: rate_limit_args (:obj:`None` | :obj:`int`): If set, specifies the maximum number of retries to be made in case of a :exc:`~telegram.error.RetryAfter` exception. Defaults to :paramref:`AIORateLimiter.max_retries`. Fchat_idNTrr)r?r@rArBrCz*Rate limit hit after maximum of %d retries)exc_infog?z)Rate limit hit. Retrying after %f seconds)rget contextlibsuppress ValueError TypeErrorint isinstancestrrangerGrr.r _LOGGER exception retry_afterinfoclearr,sleep)r/rArBrCrHrIrJr&r@r?rKiexcr[rrrprocess_requestsB        zAIORateLimiter.process_request)rrr r!r)rN)__name__ __module__ __qualname____doc__ __slots__floatrRr0r4r6r rTboolr>rrrr rrrGrr^rrrrr6sh@   "  "  )!rbr,rNsystypingrrrrrrrr aiolimiterr r* ImportErrortelegram._utils.loggingr telegram._utils.typesr telegram.errorr telegram.ext._baseratelimiterr version_info nullcontextrasynccontextmanagerr_rVrRrrrrrs*(