Simplificación de adición de tokens

herramientas.py

@@ -23,24 +23,27 @@ def imprimeTime():
 
     return formatted_time
 
-def registrar_evento(cus: str, tokens: int):
+def registrar_evento(cus: str, firebase_user: str, tokens: int):
     """
     Registra los tokens correspondientes a la compra realizada por el usuario.
 
     Args:
         cus (str): customer de stripe.
     """   
+
+    #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', usuario, 'tokens', amount=tokens)
+    # 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)

main.py

@@ -1,4 +1,3 @@
-import time
 import stripe
 from fastapi import FastAPI, Request, Header
 import sulkuPypi
@@ -64,7 +63,8 @@ async def webhook_received(request: Request, stripe_signature: str = Header(None
                 imagenes = event_data['metadata']['imagenes']
                 print("Imágenes:")
                 print(imagenes)  
-                herramientas.registrar_evento(cus_id, int(imagenes))             
+                firebase_user = event_data['metadata']['firebase_user']
+                herramientas.registrar_evento(cus_id, firebase_user, int(imagenes))             
 
     if event_type == 'checkout.session.async_payment_succeeded': 
         print("Pago asíncrono completado.")