x'h{`ddlmZmZddlmZmZddlmZmZddlm Z ddl m Z GddZ y) )State StatesGroup) CallbackQueryMessage)TeleBottypes)resolve_context)UnionceZdZdZdeeefdeddfdZdee efde fdZ defd Z de fd Z de 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']) 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. messagebotreturnNcV||_||_|jj|_y)N)r rbot_id)selfr rs S/var/www/flask-api/venv/lib/python3.12/site-packages/telebot/states/sync/context.py__init__zStateContext.__init__s6= hhoo statect|j|jj\}}}}}t |t r |j }|jj||||||S)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)r r rr isinstancername set_state)rrrrrrrs rsetzStateContext.setstO^^b^j^jlpltltl{l{N|K0&:K eU #JJExx!!#9/ "  rct|j|jj\}}}}}|jj |||||S)zo Get current state for current user. :return: Current state name. :rtype: str rrrrr)r r rr get_staterrrrrrs rgetzStateContext.get9s^O^^b^j^jlpltltl{l{N|K0&:Kxx!!#9/ "  rct|j|jj\}}}}}|jj |||||S)z Deletes state and data for current user. .. warning:: This method deletes state and associated data for current user. r!)r r rr delete_stater#s rdeletezStateContext.deleteJs^O^^b^j^jlpltltl{l{N|K0&:Kxx$$#9/ %  rct|j|jj\}}}}}|jj |||||S)zR Reset data for current user. State will not be changed. r!)r r rr reset_datar#s rr)zStateContext.reset_data[s^ O^^b^j^jlpltltl{l{N|K0&:Kxx""#9/ #  rct|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!)r r rr retrieve_datar#s rdatazStateContext.datajs^O^^b^j^jlpltltl{l{N|K0&:Kxx%%#9/ &  rc t|j|jj\}}}}}|jjd|||||d|S)zl Add data for current user. :param kwargs: Data to add. :type kwargs: dict r!)r r rradd_data)rkwargsrrrrrs rr/zStateContext.add_data~sjO^^b^j^jlpltltl{l{N|K0&:K txx   #9/     r)rN)__name__ __module__ __qualname____doc__r rrstrrrboolrr$r'r)dictr,r/r.rrr r s| &g}&< =&C&D&  ucz* t 6 S "  "  D   d ( rr N) telebot.statesrr telebot.typesrrtelebotrrr typingr r r.rrr<s -0"*D D r