o
    "h                     @   s   d Z ddlZddl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mZ ddlmZ ddlmZ dd	lmZmZ erDdd
lmZ edZG dd dee	eef ZdS )z-This module contains the PrefixHandler class.    N)TYPE_CHECKINGAnyOptionalTypeVarUnion)Update)DEFAULT_TRUE)SCTDVType)filters)BaseHandler)CCTHandlerCallback)ApplicationRTc                       s   e Zd ZdZdZdefdddee dee dee	e
ef d	eej d
ee f fddZdedeeeeee eeeeeef f  f f  fddZde
de	dddeeeeee ee f f  ddf
ddZ  ZS )PrefixHandlera  Handler class to handle custom prefix commands.

    This is an intermediate handler between :class:`MessageHandler` and :class:`CommandHandler`.
    It supports configurable commands with the same options as :class:`CommandHandler`. It will
    respond to every combination of :paramref:`prefix` and :paramref:`command`.
    It will add a :obj:`list` to the :class:`CallbackContext` named :attr:`CallbackContext.args`,
    containing a list of strings, which is the text following the command split on single or
    consecutive whitespace characters.

    Examples:

        Single prefix and command:

        .. code:: python

            PrefixHandler("!", "test", callback)  # will respond to '!test'.

        Multiple prefixes, single command:

        .. code:: python

            PrefixHandler(["!", "#"], "test", callback)  # will respond to '!test' and '#test'.

        Multiple prefixes and commands:

        .. code:: python

            PrefixHandler(
                ["!", "#"], ["test", "help"], callback
            )  # will respond to '!test', '#test', '!help' and '#help'.


    By default, the handler listens to messages as well as edited messages. To change this behavior
    use :attr:`~filters.UpdateType.EDITED_MESSAGE <telegram.ext.filters.UpdateType.EDITED_MESSAGE>`

    Note:
        * :class:`PrefixHandler` does *not* handle (edited) channel posts.

    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.

    .. versionchanged:: 20.0

        * :class:`PrefixHandler` is no longer a subclass of :class:`CommandHandler`.
        * Removed the attributes ``command`` and ``prefix``. Instead, the new :attr:`commands`
          contains all commands that this handler listens to as a :class:`frozenset`, which
          includes the prefixes.
        * Updating the prefixes and commands this handler listens to is no longer possible.

    Args:
        prefix (:obj:`str` | Collection[:obj:`str`]):
            The prefix(es) that will precede :paramref:`command`.
        command (:obj:`str` | Collection[:obj:`str`]):
            The command or list of commands this handler should listen for. Case-insensitive.
        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`.
        filters (:class:`telegram.ext.filters.BaseFilter`, optional): A filter inheriting from
            :class:`telegram.ext.filters.BaseFilter`. Standard filters can be found in
            :mod:`telegram.ext.filters`. Filters can be combined using bitwise
            operators (``&`` for :keyword:`and`, ``|`` for :keyword:`or`, ``~`` for :keyword:`not`)
        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:
        commands (frozenset[:obj:`str`]): The commands that this handler will listen for, i.e. the
            combinations of :paramref:`prefix` and :paramref:`command`.
        callback (:term:`coroutine function`): The callback function for this handler.
        filters (:class:`telegram.ext.filters.BaseFilter`): Optional. Only allow updates with these
            Filters.
        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`.

    )commandsr   NselfzPrefixHandler[CCT, RT]prefixcommandcallbackr   blockc                    s   t  j||d t|tr| hndd |D }t|tr#| hndd |D }tdd t||D | _|d urA|| _d S t	j
j| _d S )N)r   r   c                 S      h | ]}|  qS  lower.0xr   r   W/var/www/html/venv/lib/python3.10/site-packages/telegram/ext/_handlers/prefixhandler.py	<setcomp>       z)PrefixHandler.__init__.<locals>.<setcomp>c                 S   r   r   r   r   r   r   r   r       r!   c                 s   s    | ]	\}}|| V  qd S )Nr   )r   pcr   r   r   	<genexpr>   s    
z)PrefixHandler.__init__.<locals>.<genexpr>)super__init__
isinstancestrr   	frozenset	itertoolsproductr   filters_module
UpdateTypeMESSAGESr   )r   r   r   r   r   r   prefixesr   	__class__r   r   r&   }   s   ""


zPrefixHandler.__init__updatereturnc                 C   sd   t |tr0|jr0|j}|jr0|j }|d  | jvrdS | j|}|r.|dd |fS dS dS )a  Determines whether an update should be passed to this handler's :attr:`callback`.

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

        Returns:
            :obj:`list`: The list of args for the handler.

        r   N   F)	r'   r   effective_messagetextsplitr   r   r   check_update)r   r2   message	text_listfilter_resultr   r   r   r8      s   
zPrefixHandler.check_updatecontextapplicationz)Application[Any, CCT, Any, Any, Any, Any]check_resultc                 C   s<   t |tr|d |_t |d tr||d  dS dS dS )zAdd text after the command to :attr:`CallbackContext.args` as list, split on single
        whitespaces and add output of data filters to :attr:`CallbackContext` as well.
        r   r4   N)r'   tupleargsdictr2   )r   r<   r2   r=   r>   r   r   r   collect_additional_context   s   


z(PrefixHandler.collect_additional_context)__name__
__module____qualname____doc__	__slots__r   r	   r(   r   r   r   r   r   r,   
BaseFilterr
   boolr&   objectr   r?   listrA   r   r8   rB   __classcell__r   r   r0   r   r   $   sD    V.
r   )rF   r*   typingr   r   r   r   r   telegramr   telegram._utils.defaultvaluer   telegram._utils.typesr	   r
   telegram.extr   r,   "telegram.ext._handlers.basehandlerr   telegram.ext._utils.typesr   r   r   r   r   r   r   r   r   <module>   s   