o -j@sdddlmZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z Gdd d e ZdS) )iscoroutinefunction)CallableTupleN) PyromodConfig) CallbackQuery IdentifierListener)HandlercsxeZdZdZddeffdd ZdefddZd d ded ee e ffd d Z d d defddZ d d defddZ ZS)CallbackQueryHandleraThe CallbackQuery handler class. Used to handle callback queries coming from inline buttons. It is intended to be used with :meth:`~pyrogram.Client.add_handler` For a nicer way to register this handler, have a look at the :meth:`~pyrogram.Client.on_callback_query` decorator. Parameters: callback (``Callable``): Pass a function that will be called when a new CallbackQuery arrives. It takes *(client, callback_query)* as positional arguments (look at the section below for a detailed description). filters (:obj:`Filters`): Pass one or more filters to allow only a subset of callback queries to be passed in your callback function. Other parameters: client (:obj:`~pyrogram.Client`): The Client itself, useful when you want to call other API methods inside the message handler. callback_query (:obj:`~pyrogram.types.CallbackQuery`): The received callback query. Ncallbackcs||_t|j|dSN)original_callbacksuper__init__resolve_future_or_callback)selfr filters __class__e/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/pyrogram/handlers/callback_query_handler.pyr7szCallbackQueryHandler.__init__querycCsz|j}|r|jnd}|r|jnd}d}d}|jr2t|jdt|jdd}|jjr2|jjj|jjjg}t||||g|jdS)z Composes an Identifier object from a CallbackQuery object. :param query: The CallbackQuery object to compose of. :return: An Identifier object. Nid message_id)rchat_id from_user_idinline_message_id) from_userrusernamemessagegetattrchatrr)rrrrfrom_user_usernamerrrrrcompose_data_identifier;s"z,CallbackQueryHandler.compose_data_identifierclientzpyrogram.Clientreturncs||}||tjjj}d}|r=|j}t|r;t|j r+|||IdH}||fS|j d|||IdH}||fSd}||fS)a Checks if the CallbackQuery object has a matching listener. :param client: The Client object to check with. :param query: The CallbackQuery object to check with. :return: A tuple of a boolean and a Listener object. The boolean indicates whether the found listener has filters and its filters matches with the CallbackQuery object. The Listener object is the matching listener. FNT) r$get_listener_matching_with_datapyrogramenums ListenerTypesCALLBACK_QUERYrcallabler__call__looprun_in_executor)rr%rdatalistenerlistener_does_matchrrrrcheck_if_has_matching_listenerXs$    z3CallbackQueryHandler.check_if_has_matching_listenerc s|||IdH\}}t|jr/t|jjr!|||IdH}n|jd|j||IdH}nd}||}tj rht |j |j |j dd}||}|rh|rh|sh|j rht|j tr[|j ntj} || IdHdS|pk|S)aU Checks if the CallbackQuery object has a matching listener or handler. :param client: The Client object to check with. :param query: The CallbackQuery object to check with. :return: A boolean indicating whether the CallbackQuery object has a matching listener or the handler filter matches. NT)rrrrF)r3r,rrr-r.r/r$runallowed_click_alertrrrrmatches isinstancestrunallowed_click_alert_textanswer) rr%rr2r1handler_does_matchr0permissive_identifierr5alertrrrcheckzsF        zCallbackQueryHandler.checkcs|||IdH\}}|rO|rO|||jr&|js&|j|tj|jrKt|jr>|j||g|RIdHtj|j||g|Rtjt d|j ||g|RIdHdS)aZ Resolves the future or calls the callback of the listener. Will call the original handler if no listener. :param client: The Client object to resolve or call with. :param query: The CallbackQuery object to resolve or call with. :param args: The arguments to call the callback with. :return: None Nz0Listener must have either a future or a callback) r3remove_listenerfuturedone set_resultr(StopPropagationr r ValueErrorr)rr%rargsr2r1rrrrs"    z/CallbackQueryHandler.resolve_future_or_callbackr )__name__ __module__ __qualname____doc__rrrr$rboolrr3r=r __classcell__rrrrr s"  "5r )asynciortypingrrr(pyrogram.utilsrpyrogram.typesrrrhandlerr r rrrrs