mirror of
https://github.com/marcogll/vanessa_bot_vanity.git
synced 2026-01-13 13:25:16 +00:00
606b9c01de
This commit introduces a `/reset` command that allows users to clear their registration status. This is useful for testing the registration conversation flow multiple times without needing to restart the bot. The command works by removing the user's chat ID from the in-memory set of registered users and then displaying the main menu again.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import logging
|
|
|
|
# MOCK DATABASE MODULE
|
|
# Since the actual DB is removed temporarily, this module provides mock implementations.
|
|
|
|
SessionUsersAlma = None
|
|
SessionVanityHr = None
|
|
SessionVanityAttendance = None
|
|
|
|
_REGISTERED_USERS = set()
|
|
|
|
def chat_id_exists(chat_id: int) -> bool:
|
|
"""Mock check: returns True if user is in in-memory set."""
|
|
return int(chat_id) in _REGISTERED_USERS
|
|
|
|
def register_user(user_data: dict) -> bool:
|
|
"""Mock register: adds user to in-memory set."""
|
|
try:
|
|
meta = user_data.get("meta", {})
|
|
metadata = user_data.get("metadata", {})
|
|
tid = meta.get("telegram_id") or metadata.get("telegram_id") or metadata.get("chat_id")
|
|
|
|
if tid:
|
|
_REGISTERED_USERS.add(int(tid))
|
|
logging.info(f"[MockDB] User {tid} registered in memory.")
|
|
return True
|
|
logging.warning("[MockDB] Could not find telegram_id in user_data.")
|
|
return False
|
|
except Exception as e:
|
|
logging.error(f"[MockDB] Register error: {e}")
|
|
return False
|
|
|
|
|
|
def deregister_user(chat_id: int) -> bool:
|
|
"""Mock deregister: removes user from in-memory set."""
|
|
try:
|
|
_REGISTERED_USERS.discard(int(chat_id))
|
|
logging.info(f"[MockDB] User {chat_id} deregistered from memory.")
|
|
return True
|
|
except Exception as e:
|
|
logging.error(f"[MockDB] Deregister error: {e}")
|
|
return False
|