mirror of
https://github.com/marcogll/vanessa_bot_vanity.git
synced 2026-01-13 13:25:16 +00:00
first commit
This commit is contained in:
31
app/modules/database.py
Normal file
31
app/modules/database.py
Normal file
@@ -0,0 +1,31 @@
|
||||
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
|
||||
Reference in New Issue
Block a user