Bifuración de sitios

funciones.py

@@ -7,9 +7,6 @@ import time
 
 async def procesa_evento(request: Request, stripe_signature: str = Header(None)):
 
-    print("Estoy en procesa evento....")
-    
-
     webhook_secret = globales.webhook
     data = await request.body()    
     
@@ -34,6 +31,10 @@ async def procesa_evento(request: Request, stripe_signature: str = Header(None))
     print("Sucedido a las: ", herramientas.imprimeTime())
     gaCliente = event_data['metadata'].get('gaCliente', None)
     print("Gaclient es: ", gaCliente)
+    
+    # Obtener de qué sitio viene el evento
+    sitio = event_data['metadata'].get('sitio', None)
+    print("Sitio es: ", sitio)
 
     if event_type == 'checkout.session.completed': 
         print("Checkout Session completado.")
@@ -53,7 +54,7 @@ async def procesa_evento(request: Request, stripe_signature: str = Header(None))
                 imagenes = event_data['metadata']['imagenes']
                 print("Imágenes:", imagenes)
                 firebase_user = event_data['metadata']['firebase_user']
-                herramientas.registrar_evento(cus_id, firebase_user, int(imagenes))  
+                herramientas.registrar_evento(cus_id, firebase_user, int(imagenes), sitio)  
                 #Analytics se hace al final después de registrar la venta.
                 print("gacliente:", gaCliente)
                 #Si el cliente es None, éste no fue enviado y no ejecutes esto. 
@@ -74,8 +75,9 @@ async def procesa_evento(request: Request, stripe_signature: str = Header(None))
         print("Customer directo:", cus_id)
         imagenes = event_data['metadata']['imagenes']
         print("Imágenes:")
-        print(imagenes)  
-        herramientas.registrar_evento(cus_id, int(imagenes))
+        print(imagenes)
+        firebase_user = event_data['metadata']['firebase_user']
+        herramientas.registrar_evento(cus_id, firebase_user, int(imagenes), sitio)
         print("gacliente:", gaCliente)
         print("Sending to GA4...")
         ga4Analiticas.send_ga4_purchase_event(event_data)

herramientas.py

@@ -22,28 +22,20 @@ def imprimeTime():
 
     return formatted_time
 
-def registrar_evento(cus: str, firebase_user: str, tokens: int):
+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.
     """   
-
-    #Si usamos firebaseuser directamente, entonces creo que no necesitaremos todo éste segmento de código:
-
-                                     #campo_buscado, var
-    # usuario = fireWhale.return_user('cus', cus) #Regresa el cus encontrado, None si no hay usuario con ese cus o 'Error' si hubo un error de conexión.
-    # print("Ésto es el usuario resultado de buscar el cus:")
-    # print(usuario)
-    # if usuario == 'Error':
-    #     print("Error en la búsqueda del resultado, quizá un error en la base de datos.")
-    #     return 'Error en Firestore'
-    # elif usuario is None: #El cus que traes no lo tiene ningun usuario.
-    #     print("Usuario nuevo, buscar el usuario entonces por email.")
-    #     usuario = fireWhale.return_user('email', cus)
-    #     print("Usuario ahora por email: ", usuario)
-    #     fireWhale.editaDato('usuarios', usuario, 'tokens', tokens)
-
-    fireWhale.incrementar_campo_numerico('usuarios', firebase_user, 'tokens', amount=tokens)
-    fireWhale.editaDato('usuarios', firebase_user, 'compro', True)
+    
+    if sitio == 'crea_imagen':
+        print("llegué al otro firestore")
+        # Aquí irá la lógica para el otro firestore
+    else:
+        fireWhale.incrementar_campo_numerico('usuarios', firebase_user, 'tokens', amount=tokens)
+        fireWhale.editaDato('usuarios', firebase_user, 'compro', True)