o @MhT@sRddlmZddlmZmZddlmZddlmZddlm Z GdddZ dS) )State) CallbackQueryMessage) AsyncTeleBot)resolve_context)Unionc@seZdZdZdeeefdeddfddZdee e fde fd d Z de fd d Z de fd dZde fddZdefddZdddZdS) StateContexta Class representing a state context. Passed through a middleware to provide easy way to set states. .. code-block:: python3 @bot.message_handler(commands=['start']) async def start_ex(message: types.Message, state_context: StateContext): await state_context.set(MyStates.name) await bot.send_message(message.chat.id, 'Hi, write me a name', reply_to_message_id=message.message_id) # also, state_context.data(), .add_data(), .reset_data(), .delete() methods available. messagebotreturnNcCs||_||_|jj|_dS)N)r r bot_id)selfr r rQ/var/www/html/venv/lib/python3.10/site-packages/telebot/states/asyncio/context.py__init__szStateContext.__init__statecsJt|j|jj\}}}}}t|tr|j}|jj||||||dIdHS)a Set state for current user. :param state: State object or state name. :type state: Union[State, str] .. code-block:: python3 @bot.message_handler(commands=['start']) async def start_ex(message: types.Message, state_context: StateContext): await state_context.set(MyStates.name) await bot.send_message(message.chat.id, 'Hi, write me a name', reply_to_message_id=message.message_id) )chat_iduser_idrbusiness_connection_idr message_thread_idN)rr r r isinstancername set_state)r rrrrr rrrrsets  zStateContext.setc8t|j|jj\}}}}}|jj|||||dIdHS)zo Get current state for current user. :return: Current state name. :rtype: str rrrr rN)rr r r get_stater rrrr rrrrget:  zStateContext.getcr)z Deletes state and data for current user. .. warning:: This method deletes state and associated data for current user. rN)rr r r delete_staterrrrdeleteMrzStateContext.deletecr)zQ Reset data for current user. State will not be changed. rN)rr r r reset_datarrrrr"`s  zStateContext.reset_datacCs0t|j|jj\}}}}}|jj|||||dS)z Get data for current user. .. code-block:: python3 with state_context.data() as data: print(data) data['name'] = 'John' r)rr r r retrieve_datarrrrdataqs zStateContext.datacs@t|j|jj\}}}}}|jjd|||||d|IdHS)zl Add data for current user. :param kwargs: Data to add. :type kwargs: dict rNr)rr r r add_data)r kwargsrrrr rrrrr%s  zStateContext.add_data)r N)__name__ __module__ __qualname____doc__rrrrrrstrboolrrr!r"dictr$r%rrrrr srN) telebot.statesr telebot.typesrrtelebot.async_telebotrrtypingrrrrrrs