o @Mh{@sZddlmZmZddlmZmZddlmZmZddlm Z ddl m Z GdddZ dS) )State StatesGroup) CallbackQueryMessage)TeleBottypes)resolve_context)Unionc@seZdZdZdeeefdeddfddZdee efde fd d Z defd d Z de fd dZ de 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']) def start_ex(message: types.Message, state_context: StateContext): state_context.set(MyStates.name) 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 rN/var/www/html/venv/lib/python3.10/site-packages/telebot/states/sync/context.py__init__szStateContext.__init__statecCsBt|j|jj\}}}}}t|tr|j}|jj||||||dS)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']) def start_ex(message: types.Message, state_context: StateContext): state_context.set(MyStates.name) bot.send_message(message.chat.id, 'Hi, write me a name', reply_to_message_id=message.message_id) )chat_iduser_idrbusiness_connection_idrmessage_thread_id)rr r r isinstancername set_state)rrrrrrrrrrsets zStateContext.setcC0t|j|jj\}}}}}|jj|||||dS)zo Get current state for current user. :return: Current state name. :rtype: str rrrrr)rr r r get_staterrrrrrrrrget9zStateContext.getcCr)z Deletes state and data for current user. .. warning:: This method deletes state and associated data for current user. r)rr r r delete_staterrrrdeleteJr!zStateContext.deletecCr)zR Reset data for current user. State will not be changed. r)rr r r reset_datarrrrr$[szStateContext.reset_datacCr)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_datarrrrdatajs zStateContext.datacKs8t|j|jj\}}}}}|jjd|||||d|S)zl Add data for current user. :param kwargs: Data to add. :type kwargs: dict rNr)rr r radd_data)rkwargsrrrrrrrrr'~szStateContext.add_data)r N)__name__ __module__ __qualname____doc__r rrstrrrboolrr r#r$dictr&r'rrrrr sr N) telebot.statesrr telebot.typesrrtelebotrrrtypingr r rrrrs