etm-terrain/models/session.py

30 lines
725 B
Python

MENU = "MENU"
ATTENTE_PHOTO_FIN = "ATTENTE_PHOTO_FIN"
ATTENTE_CONTENU_SAV = "ATTENTE_CONTENU_SAV"
ATTENTE_CONTENU_MATERIEL = "ATTENTE_CONTENU_MATERIEL"
SAISIE_MANUELLE_CHANTIER = "SAISIE_MANUELLE_CHANTIER"
ATTENTE_FAQ = "ATTENTE_FAQ"
_states: dict[int, str] = {}
_data: dict[int, dict] = {}
def get_state(user_id: int) -> str:
return _states.get(user_id, MENU)
def set_state(user_id: int, state: str) -> None:
_states[user_id] = state
def get_data(user_id: int, key: str, default=None):
return _data.get(user_id, {}).get(key, default)
def set_data(user_id: int, key: str, value) -> None:
_data.setdefault(user_id, {})[key] = value
def clear_data(user_id: int) -> None:
_data.pop(user_id, None)