Class StateMemoryStorage

telebot/storage/memory_storage.py:5–225

Memory storage for states. Stores states in memory as a dictionary. .. code-block:: python3 storage = StateMemoryStorage() bot = TeleBot(token, storage=storage) :param separator: Separator for keys, default is ":". :type separator: Optional[str] :param p

class StateMemoryStorage(StateStorageBase):
    """
    Memory storage for states.

    Stores states in memory as a dictionary.

    .. code-block:: python3

        storage = StateMemoryStorage()
        bot = TeleBot(token, storage=storage)

    :param separator: Separator for keys, default is ":".
    :type separator: Optional[str]

    :param prefix: Prefix for keys, default is "telebot".
    :type prefix: Optional[str]
    """

    def __init__(
        self, separator: Optional[str] = ":", prefix: Optional[str] = "telebot"
    ) -> None:
        self.separator = separator
        self.prefix = prefix
        if not self.prefix:
            raise ValueError("Prefix cannot be empty")

        self.data = (
            {}
        )  # key: telebot:bot_id:business_connection_id:message_thread_id:chat_id:user_id

    def set_state(
        self,
        chat_id: int,
        user_id: int,
        state: str,
        business_connection_id: Optional[str] = None,
        message_thread_id: Optional[int] = None,
        bot_id: Optional[int] = None,
    ) -> bool:
        if hasattr(state, "name"):
            state = state.name

        _key = self._get_key(
            chat_id,
            user_id,
            self.prefix,
            self.separator,
            business_connection_id,
            message_thread_id,
            bot_id,
        )

        if self.data.get(_key) is None:
            self.data[_key] = {"state": state, "data": {}}
        else:
            self.data[_key]["state"] = state

        return True