x'hTXddlmZddlmZmZddlmZddlmZddlm Z GddZ y) )State) CallbackQueryMessage) AsyncTeleBot)resolve_context)UnionceZdZdZdeeefdeddfdZdee e fde fdZ de fd Z de fd Zde fd Zdefd Zdd Zy) 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. messagebotreturnNcV||_||_|jj|_y)N)r r bot_id)selfr r s V/var/www/flask-api/venv/lib/python3.12/site-packages/telebot/states/asyncio/context.py__init__zStateContext.__init__s6= !$hhoo statecKt|j|jj\}}}}}t |t r |j }|jj||||||d{S7w)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_idrmessage_thread_idN)rr r r isinstancername set_state)rrrrrrrs rsetzStateContext.sets} DLL$((// : L0&:K eU #JJEXX''#9/ (    sA1A:3A84A:cKt|j|jj\}}}}}|jj |||||d{S7w)zo Get current state for current user. :return: Current state name. :rtype: str rrrrrN)rr r r get_staterrrrrrs rgetzStateContext.get:sg DLL$((// : L0&:KXX''#9/ (    AAAAcKt|j|jj\}}}}}|jj |||||d{S7w)z Deletes state and data for current user. .. warning:: This method deletes state and associated data for current user. rN)rr r r delete_stater!s rdeletezStateContext.deleteMsg DLL$((// : L0&:KXX**#9/ +    r#cKt|j|jj\}}}}}|jj |||||d{S7w)zQ Reset data for current user. State will not be changed. rN)rr r r reset_datar!s rr(zStateContext.reset_data`sg DLL$((// : L0&:KXX((#9/ )    r#ct|j|jj\}}}}}|jj |||||S)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_datar!s rdatazStateContext.dataqsY DLL$((// : L0&:Kxx%%#9/ &  rc Kt|j|jj\}}}}}|jjd|||||d|d{S7w)zl Add data for current user. :param kwargs: Data to add. :type kwargs: dict rN)rr r radd_data)rkwargsrrrrrs rr.zStateContext.add_datass DLL$((// : L0&:K'TXX&& #9/       sAAAA)r N)__name__ __module__ __qualname____doc__rrrrrrstrboolrr"r&r(dictr+r.r-rrr r s} &g}&< =&L&UY&  uUCZ0 T : 3 & d & $ " d , rr N) telebot.statesr telebot.typesrrtelebot.async_telebotrrtypingrr r-rrr;s  0.*P P r