
    ՟f++                     <   d Z ddlZddlZddlZddlmZmZmZmZm	Z	m
Z
mZmZ 	 ddlmZ dZddlmZ ddlmZ dd	lmZ dd
lmZ ej2                  dk\  rej4                  Znej8                  ded   fd       Z eed      Z G d dee         Z y# e$ r dZY nw xY w)zeThis module contains an implementation of the BaseRateLimiter class based on the aiolimiter
library.
    N)AnyAsyncIteratorCallable	CoroutineDictListOptionalUnion)AsyncLimiterTF)
get_logger)JSONDict)
RetryAfter)BaseRateLimiter)   
   returnc                    K   d  y wN r       W/var/www/cvtools/html/venv/lib/python3.12/site-packages/telegram/ext/_aioratelimiter.pynull_contextr   .   s     
s   	AIORateLimiter)
class_namec                   T   e Zd ZdZdZ	 	 	 	 	 ddedededededd	fd
ZddZddZ	de
eeef   ddfdZdede
eeef   dedeeee
eeee   f   f   f   dedeeef   de
eeee   f   fdZdedeeee
eeee   f   f   f   dedeeef   dedeeef   dee   de
eeee   f   fdZy	)r   a  
    Implementation of :class:`~telegram.ext.BaseRateLimiter` using the library
    `aiolimiter <https://aiolimiter.readthedocs.io/en/stable>`_.

    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 <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overall_max_rateoverall_time_periodgroup_max_rategroup_time_periodmax_retriesr   Nc                 &   t         st        d      |r|rt        ||      | _        nd | _        |r|r|| _        || _        nd| _        d| _        i | _        || _        t        j                         | _
        | j                  j                          y )NzeTo use `AIORateLimiter`, PTB must be installed via `pip install "python-telegram-bot[rate-limiter]"`.max_ratetime_periodr   )AIO_LIMITER_AVAILABLERuntimeErrorr   r   r   r   r   r    asyncioEventr!   set)selfr"   r#   r$   r%   r&   s         r   __init__zAIORateLimiter.__init__   s     %8   39E)7J:D "&D/*8D ->D##$D &'D#DF!,")--/##%r   c                    K   ywzDoes nothing.Nr   r0   s    r   
initializezAIORateLimiter.initialize           c                    K   ywr3   r   r4   s    r   shutdownzAIORateLimiter.shutdown   r6   r7   group_idr   c                    t        | j                        dkD  r_| j                  j                         j                         D ]4  \  }}||k(  r|j	                  |j
                        s(| j                  |= 6 || j                  vr.t        | j                  | j                        | j                  |<   | j                  |   S )Ni   r(   )	lenr   copyitemshas_capacityr)   r   r   r   )r0   r:   keylimiters       r   _get_group_limiterz!AIORateLimiter._get_group_limiter   s    
 t##$s* $ 4 4 9 9 ; A A C 2W(?''(8(89,,S1	2 4///-9-- 33.D  * ##H--r   chatgroupcallback.argskwargsc                   K   |r| j                   r| j                   n	t               }|r| j                  r| j                  |      n	t               }|4 d {    |4 d {    | j                  j                          d {     ||i | d {   cd d d       d {    cd d d       d {    S 7 c7 Z7 :7 ,7 7 # 1 d {  7  sw Y   nxY wd d d       d {  7   y # 1 d {  7  sw Y   y xY wwr   )r   r   r   rB   r!   wait)r0   rC   rD   rE   rF   rG   base_contextgroup_contexts           r   _run_requestzAIORateLimiter._run_request   s      /3t7I7It))P\P^.38L8LD##E*R^R` 	 ! 	3 	3, 	3 	3))..000!42622		3 	3 	3 	3 	3 	302		3 	3 	3 	3 	3 	3 	3 	3 	3 	3s   ADB:DC-!B<"C-%CB>CC CC-#C$C-(D4C5D<C->C CC-DC	CC	C-D&C)'D-C?3C64C?;Dendpointdatarate_limit_argsc                 ^  K   |xs | j                   }d}d}	|j                  d      }
|
d}	t        j                  t        t
              5  t        |
      }
ddd       t        |
t              r|
dk  st        |
t              r|
}t        |dz         D ]<  }	 | j                  |	||||       d{   | j                  j                          c S  y# 1 sw Y   ~xY w7 .# t        $ r}||k(  rt        j                  d||	        |j                   d
z   }t        j#                  d|       | j                  j%                          t'        j(                  |       d{  7   Y d}~nd}~ww xY w	 | j                  j                          # | j                  j                          w xY ww)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`.
        Fchat_idNTr      )rC   rD   rE   rF   rG   z*Rate limit hit after maximum of %d retries)exc_infog?z)Rate limit hit. Retrying after %f seconds)r    get
contextlibsuppress
ValueError	TypeErrorint
isinstancestrrangerL   r!   r/   r   _LOGGER	exceptionretry_afterinfoclearr-   sleep)r0   rE   rF   rG   rM   rN   rO   r&   rD   rC   rQ   iexcrb   s                 r   process_requestzAIORateLimiter.process_request   s    ( &:):):',((9%D   Y7 	#'lG	# w$1GS9Q E{Q' 	.A.!..UXDQW /  " ''++-'	.( 9	# 	#  +#%%Dk\_ &  #-H%P''--/mmE***+ + ''++-''++-sz   AF-
CA F-C!0C1C!4F-CF-C!!	E/*A5E*E" E*%F*E//F3F-F**F-)   rR      <   r   )r   N)__name__
__module____qualname____doc__	__slots__floatrY   r1   r5   r9   r
   r[   boolrB   r   r   r   r   r   r   rL   r	   re   r   r   r   r   r   6   s   >@I #%%& "#%&& #& 	&
 !& & 
&@.5c4+@ .^ .(33 S#t^$3 3	#sE$$x.:X4Y*Y ZZ[	3
 3 S#X3 
tXtH~-	.3(93	#sE$$x.:X4Y*Y ZZ[9 9 S#X	9
 9 38n9 "#9 
tXtH~-	.9r   )!rl   r-   rU   systypingr   r   r   r   r   r   r	   r
   
aiolimiterr   r+   ImportErrortelegram._utils.loggingr   telegram._utils.typesr   telegram.errorr   telegram.ext._baseratelimiterr   version_infonullcontextr   asynccontextmanagerri   r]   rY   r   r   r   r   <module>r{      s   &   
 W W W"'  / * % 9
 w))L ##d 3  $ X*:
;Q_S) Q/  "!"s   B BB