import pytz
import fireWhale
from datetime import datetime

def imprimeTime(): 
    """
    Devuelve la fecha y hora actual en la zona horaria de la Ciudad de México (GMT-6).
    """
    # 1. Definir la zona horaria de la Ciudad de México
    # Puedes usar 'America/Mexico_City' para que pytz maneje el horario de verano automáticamente.
    mexico_city_tz = pytz.timezone('America/Mexico_City')

    # 2. Obtener la hora actual en UTC
    utc_now = datetime.now(pytz.utc)

    # 3. Convertir la hora UTC a la zona horaria deseada
    mexico_city_now = utc_now.astimezone(mexico_city_tz)

    # 4. Formatear la fecha y hora
    # El formato que deseas es "YYYY-MM-DD HH:MM:SS"
    formatted_time = mexico_city_now.strftime("%Y-%m-%d %H:%M:%S")

    return formatted_time

def registrar_evento(cus: str, firebase_user: str, tokens: int, sitio: str = None):
    """
    Registra los tokens correspondientes a la compra realizada por el usuario.

    Args:
        cus (str): customer de stripe.
        firebase_user (str): usuario de firebase.
        tokens (int): cantidad de tokens a registrar.
        sitio (str, optional): sitio de origen del evento.
    """   

    print("El sitio recibido es: ", sitio)
    
    if sitio == 'crea-imagen':
        print("llegué al otro firestore")
        # Obtener datos del usuario de splashmix-ai
        datos_usuario = fireWhale.obtener_usuario_splashmix(firebase_user, tokens)
    else:
        fireWhale.incrementar_campo_numerico('usuarios', firebase_user, 'tokens', amount=tokens)
        fireWhale.editaDato('usuarios', firebase_user, 'compro', True)