17 lines
739 B
Python
17 lines
739 B
Python
class UserStateManager:
|
|
def __init__(self):
|
|
# Словарь для хранения состояния каждого пользователя
|
|
self.user_states = {}
|
|
|
|
def set_state(self, chat_id, state):
|
|
"""Устанавливает состояние для пользователя."""
|
|
self.user_states[chat_id] = state
|
|
|
|
def get_state(self, chat_id):
|
|
"""Получает текущее состояние пользователя."""
|
|
return self.user_states.get(chat_id, "MAIN_MENU")
|
|
|
|
def reset_state(self, chat_id):
|
|
"""Сбрасывает состояние пользователя в главное меню."""
|
|
self.user_states[chat_id] = "MAIN_MENU"
|