U
    gM$                     @   s   d 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mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZmZ erdd	lmZ e
d
ZG dd deeeef ZdS )z4This module contains the CallbackQueryHandler class.    N)	TYPE_CHECKINGAnyCallableMatchOptionalPatternTypeVarUnioncast)Update)DEFAULT_TRUE)DVType)BaseHandler)CCTHandlerCallback)ApplicationRTc                       s   e Zd ZdZdZddefdeeee	f e
eeee eeege
e f f  e
eeee f  ee d fddZee
eeef  dd	d
Zeedeeee f ddddZ  ZS )CallbackQueryHandlera  Handler class to handle Telegram
    :attr:`callback queries <telegram.Update.callback_query>`. Optionally based on a regex.

    Read the documentation of the :mod:`re` module for more information.

    Note:
        * If your bot allows arbitrary objects as
          :paramref:`~telegram.InlineKeyboardButton.callback_data`, it may happen that the
          original :attr:`~telegram.InlineKeyboardButton.callback_data` for the incoming
          :class:`telegram.CallbackQuery` can not be found. This is the case when either a
          malicious client tempered with the :attr:`telegram.CallbackQuery.data` or the data was
          simply dropped from cache or not persisted. In these
          cases, an instance of :class:`telegram.ext.InvalidCallbackData` will be set as
          :attr:`telegram.CallbackQuery.data`.

          .. versionadded:: 13.6

        * If neither :paramref:`pattern` nor :paramref:`game_pattern` is set, `any`
          ``CallbackQuery`` will be handled. If only :paramref:`pattern` is set, queries with
          :attr:`~telegram.CallbackQuery.game_short_name` will `not` be considered and vice versa.
          If both patterns are set, queries with either :attr:
          `~telegram.CallbackQuery.game_short_name` or :attr:`~telegram.CallbackQuery.data`
          matching the defined pattern will be handled

          .. versionadded:: 21.5

    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.

    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`.
        pattern (:obj:`str` | :func:`re.Pattern <re.compile>` | :obj:`callable` | :obj:`type`,             optional):
            Pattern to test :attr:`telegram.CallbackQuery.data` against. If a string or a regex
            pattern is passed, :func:`re.match` is used on :attr:`telegram.CallbackQuery.data` to
            determine if an update should be handled by this handler. If your bot allows arbitrary
            objects as :paramref:`~telegram.InlineKeyboardButton.callback_data`, non-strings will
            be accepted. To filter arbitrary objects you may pass:

            - a callable, accepting exactly one argument, namely the
              :attr:`telegram.CallbackQuery.data`. It must return :obj:`True` or
              :obj:`False`/:obj:`None` to indicate, whether the update should be handled.
            - a :obj:`type`. If :attr:`telegram.CallbackQuery.data` is an instance of that type
              (or a subclass), the update will be handled.

            If :attr:`telegram.CallbackQuery.data` is :obj:`None`, the
            :class:`telegram.CallbackQuery` update will not be handled.

            .. seealso:: :wiki:`Arbitrary callback_data <Arbitrary-callback_data>`

            .. versionchanged:: 13.6
               Added support for arbitrary callback data.
        game_pattern (:obj:`str` | :func:`re.Pattern <re.compile>` | optional)
            Pattern to test :attr:`telegram.CallbackQuery.game_short_name` against. If a string or
            a regex pattern is passed, :func:`re.match` is used on
            :attr:`telegram.CallbackQuery.game_short_name` to determine if an update should be
            handled by this handler.

            .. versionadded:: 21.5
        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.
        pattern (:func:`re.Pattern <re.compile>` | :obj:`callable` | :obj:`type`): Optional.
            Regex pattern, callback or type to test :attr:`telegram.CallbackQuery.data` against.

            .. versionchanged:: 13.6
               Added support for arbitrary callback data.
        game_pattern (:func:`re.Pattern <re.compile>`): Optional.
            Regex pattern to test :attr:`telegram.CallbackQuery.game_short_name`
        block (:obj:`bool`): Determines whether the return value of the callback should be
            awaited before processing the next handler in
            :meth:`telegram.ext.Application.process_update`.

    )game_patternpatternNzCallbackQueryHandler[CCT, RT]selfcallbackr   r   blockc                    sb   t  j||d t|r*t|r*tdt|tr>t	|}t|trRt	|}|| _
|| _d S )N)r   zQThe `pattern` must not be a coroutine function! Use an ordinary function instead.)super__init__callableasyncioiscoroutinefunction	TypeError
isinstancestrrecompiler   r   r   	__class__ O/tmp/pip-unpacked-wheel-swnnwir2/telegram/ext/_handlers/callbackqueryhandler.pyr      s    	



zCallbackQueryHandler.__init__)updatereturnc                 C   s   t |tr|jsdS |jj}|jj}t| j| jgs8dS |r| jsFdS t | jtr^t || jS t	| jrr| |S t |t
sdS t| j| }r|S n*|r| jsdS t| j| }r|S ndS d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`

        NTF)r    r   Zcallback_querydatagame_short_nameanyr   r   typer   r!   r"   match)r   r(   Zcallback_datar+   r.   r&   r&   r'   check_update   s0    


z!CallbackQueryHandler.check_updatez)Application[Any, CCT, Any, Any, Any, Any])contextr(   applicationcheck_resultr)   c                 C   s   | j rtt|}|g|_dS )zAdd the result of ``re.match(pattern, update.callback_query.data)`` to
        :attr:`CallbackContext.matches` as list with one element.
        N)r   r
   r   matches)r   r0   r(   r1   r2   r&   r&   r'   collect_additional_context   s    

z/CallbackQueryHandler.collect_additional_context)__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r	   r!   r   r-   r   objectboolr   r   r/   r   r4   __classcell__r&   r&   r$   r'   r   $   s*   X -r   )r8   r   r"   typingr   r   r   r   r   r   r   r	   r
   Ztelegramr   Ztelegram._utils.defaultvaluer   Ztelegram._utils.typesr   Z"telegram.ext._handlers.basehandlerr   Ztelegram.ext._utils.typesr   r   Ztelegram.extr   r   r   r&   r&   r&   r'   <module>   s   ,