idZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z ddl mZddlmZdd lmZmZerdd l mZed ZGd d eeeefZdS)z-This module contains the PrefixHandler class.N) TYPE_CHECKINGAnyTypeVar)Update) DEFAULT_TRUE)SCTDVType)filters) BaseHandler)CCTHandlerCallback) ApplicationRTc$eZdZdZdZdefdddeedeedee e e fd e j dzd eef fd Zd ed eeeeeeeefzdzfzdzfdZde d e dddeeeeefzdzd df dZxZS) PrefixHandleraHandler 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 ` 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 NselfzPrefixHandler[CCT, RT]prefixcommandcallbackr blockct||t|tr|hn d|D}t|tr|hn d|D}t dt j||D|_||ntj j |_ dS)N)rrc6h|]}|Slower.0xs d/Users/shanyulin/my-agent/.venv/lib/python3.11/site-packages/telegram/ext/_handlers/prefixhandler.py z)PrefixHandler.__init__..s!D_D_D_STQWWYYD_D_D_c6h|]}|Srrrs r r!z)PrefixHandler.__init__..s!FbFbFbUVqwwyyFbFbFbr"c3&K|] \}}||zV dS)Nr)rpcs r z)PrefixHandler.__init__..s;2 2 aAE2 2 2 2 2 2 r") super__init__ isinstancestrr frozenset itertoolsproductrfilters_module UpdateTypeMESSAGESr ) rrrrr rprefixesr __class__s r r)zPrefixHandler.__init__~s (%888'1&#'>'>_FLLNN##D_D_X^D_D_D_(27C(@(@bGMMOO$$FbFbZaFbFbFb(12 2 '/(CC2 2 2 ) )  *GG0I0R r"updatereturnc&t|tr{|jrt|j}|jrf|j}|d|jvrdS|j|}|r |dd|fSdSdS)aDetermines 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. rNF) r*reffective_messagetextsplitrrr check_update)rr4message text_list filter_results r r;zPrefixHandler.check_updates ff % % &*B .G| #L..00 Q<%%''t}<<4 $ 9 9& A A  8$QRR=-77utr"context applicationz)Application[Any, CCT, Any, Any, Any, Any] check_resultct|trE|d|_t|dtr||ddSdSdS)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. rr7N)r*tupleargsdictr4)rr?r4r@rAs r collect_additional_contextz(PrefixHandler.collect_additional_contextsg lE * * 0'?GL,q/400 0|A///// 0 0 0 0r")__name__ __module__ __qualname____doc__ __slots__rrr+r rr rr/ BaseFilterr boolr)objectrClistrErr;rF __classcell__)r3s@r rr%s_SSl(I59*   & C S "&#r/2   *T1  d|       * d3iS#X!6!==> > E2 0 0 0A 0 U49d?33d: 0  0 0 0 0 0 0 0 0r"r)rJr-typingrrrtelegramrtelegram._utils.defaultvaluertelegram._utils.typesrr telegram.extr r/"telegram.ext._handlers.basehandlerr telegram.ext._utils.typesr r rrrrr"r rXs&43..........555555--------222222::::::::::::::)(((((( WT]]T0T0T0T0T0KR0T0T0T0T0T0r"