db_utils.py

from firebase_admin import firestore
from typing import Dict, List, Optional

def add_user_data(db: firestore.Client, user_id: str, data: Dict) -> bool:
    """Add user data to Firestore"""
    try:
        db.collection('users').document(user_id).set(data)
        return True
    except Exception:
        return False

def get_user_data(db: firestore.Client, user_id: str) -> Optional[Dict]:
    """Get user data from Firestore"""
    try:
        doc = db.collection('users').document(user_id).get()
        if doc.exists:
            return doc.to_dict()
        return None
    except Exception:
        return None

def update_user_data(db: firestore.Client, user_id: str, data: Dict) -> bool:
    """Update user data in Firestore"""
    try:
        db.collection('users').document(user_id).update(data)
        return True
    except Exception:
        return False

def delete_user_data(db: firestore.Client, user_id: str) -> bool:
    """Delete user data from Firestore"""
    try:
        db.collection('users').document(user_id).delete()
        return True
    except Exception:
        return False