U
    g                     @   sn   d Z ddlmZmZ ddlmZ ddlmZ ddlm	Z	m
Z
 ddlmZmZmZ G dd deeeef Zd	S )
z0This module contains the ChatBoostHandler class.    )FinalOptional)Update)BaseHandler)parse_chat_idparse_username)CCTRTHandlerCallbackc                       s   e Zd ZU dZdZdZee ed< dZ	ee ed< dZ
ee ed< ed	d	d
fdeeeef eee ee ed fddZeedddZ  ZS )ChatBoostHandlera~  
    Handler class to handle Telegram updates that contain a chat boost.

    Warning:
        When setting :paramref:`block` to :obj:`False`, you cannot rely on adding custom
        attributes to :class:`telegram.ext.CallbackContext`. See its docs for more info.

    .. versionadded:: 20.8

    Args:
        callback (:term:`coroutine function`): The callback function for this handler. Will be
            called when :meth:`check_update` has determined that an update should be processed by
            this handler. Callback signature::

                async def callback(update: Update, context: CallbackContext)

            The return value of the callback is usually ignored except for the special case of
            :class:`telegram.ext.ConversationHandler`.
        chat_boost_types (:obj:`int`, optional): Pass one of
            :attr:`CHAT_BOOST`, :attr:`REMOVED_CHAT_BOOST` or
            :attr:`ANY_CHAT_BOOST` to specify if this handler should handle only updates with
            :attr:`telegram.Update.chat_boost`,
            :attr:`telegram.Update.removed_chat_boost` or both. Defaults to
            :attr:`CHAT_BOOST`.
        chat_id (:obj:`int` | Collection[:obj:`int`], optional): Filters reactions to allow
            only those which happen in the specified chat ID(s).
        chat_username (:obj:`str` | Collection[:obj:`str`], optional): Filters reactions to allow
            only those which happen in the specified username(s).
        block (:obj:`bool`, optional): Determines whether the return value of the callback should
            be awaited before processing the next handler in
            :meth:`telegram.ext.Application.process_update`. Defaults to :obj:`True`.

            .. seealso:: :wiki:`Concurrency`

    Attributes:
        callback (:term:`coroutine function`): The callback function for this handler.
        chat_boost_types (:obj:`int`): Optional. Specifies if this handler should handle only
            updates with :attr:`telegram.Update.chat_boost`,
            :attr:`telegram.Update.removed_chat_boost` or both.
        block (:obj:`bool`): Determines whether the callback will run in a blocking way.
    )	_chat_ids_chat_usernameschat_boost_types
CHAT_BOOSTr   REMOVED_CHAT_BOOST   ANY_CHAT_BOOSTNTzChatBoostHandler[CCT, RT]selfcallbackr   chat_idchat_usernameblockc                    s.   t  j||d || _t|| _t|| _d S )N)r   )super__init__r   r   r   r   r   r   	__class__ K/tmp/pip-unpacked-wheel-swnnwir2/telegram/ext/_handlers/chatboosthandler.pyr   V   s    
zChatBoostHandler.__init__)updatereturnc                 C   s   t |tsdS |js|jsdS | j| jkr4|js4dS | j| jkrJ|jsJdS t| j| j	fs^dS |j
 }rn|jnd}|r||jnd}t| jo|| jkpt| j	o|| j	kS )zDetermines whether an update should be passed to this handler's :attr:`callback`.

        Args:
            update (:class:`telegram.Update` | :obj:`object`): Incoming update.

        Returns:
            :obj:`bool`

        FTN)
isinstancer   Z
chat_boostZremoved_chat_boostr   r   r   anyr   r   Zeffective_chatidusernamebool)r   r    Zchatr   r   r   r   r   check_updatec   s    

zChatBoostHandler.check_update)__name__
__module____qualname____doc__	__slots__r   r   int__annotations__r   r   r
   r   r   r	   r   strr&   r   objectr'   __classcell__r   r   r   r   r      s$   
*r   N)r+   typingr   r   Ztelegramr   Z"telegram.ext._handlers.basehandlerr   Z#telegram.ext._utils._update_parsingr   r   Ztelegram.ext._utils.typesr   r	   r
   r   r   r   r   r   <module>   s   