Preparado para usuarios nuevos

fireWhale.py

@@ -122,7 +122,7 @@ def verificar_token(id_token):
         print(f"Token inválido: {e}")
         return None  # Retorna None si el token es inválido
     
-def encontrar_documento_por_cus(valor_cus):
+def return_user(campo_buscado, valor_cus):
     """
     Busca un documento en la colección 'usuario' donde el campo 'cus'
     coincide con el valor_cus especificado.
@@ -140,12 +140,12 @@ def encontrar_documento_por_cus(valor_cus):
     try:
         # Realizamos la consulta: filtra por documentos donde 'cus' sea igual a valor_cus
         query_results = db.collection('usuarios').where(
-            filter=firestore.FieldFilter('cus', '==', valor_cus)
+            filter=firestore.FieldFilter(campo_buscado, '==', valor_cus)
         ).limit(1).get() # Agregamos .limit(1) porque esperas un único documento
 
         if not query_results:
-            print(f"❌ No se encontró ningún documento con 'cus' = '{valor_cus}'.")
-            return None, None
+            print(f"❌ No se encontró ningún documento con {campo_buscado} = '{valor_cus}'.")
+            return None
 
         # Si se encuentra, query_results será una lista de instantáneas de documentos
         # Tomamos el primer resultado (ya que usamos limit(1))
@@ -158,8 +158,8 @@ def encontrar_documento_por_cus(valor_cus):
         return doc_id
 
     except Exception as e:
-        print(f"❌ Error al buscar documento por 'cus': {e}")
-        return None, None
+        print(f"❌ Error al buscar documento por {campo_buscado}: {e}")
+        return "Error"
     
 def incrementar_campo_numerico(collection_name, document_id, field_name, amount=1):
     """

herramientas.py

@@ -30,9 +30,17 @@ def registrar_evento(cus: str, tokens: int):
     Args:
         cus (str): customer de stripe.
     """   
-
-    resultado = fireWhale.encontrar_documento_por_cus(cus)
-    print("Ésto es el resultado de buscar el cus:")
-    print(resultado)
-    #fireWhale.editaDato('usuarios', resultado, 'tokens', tokens)
-    fireWhale.incrementar_campo_numerico('usuarios', resultado, 'tokens', amount=tokens)
+                                     #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)

main.py

@@ -58,7 +58,6 @@ async def webhook_received(request: Request, stripe_signature: str = Header(None
         #Si hay un elemento que es card abona inmediatamente. 
         #Si no, por ahora ignora. 
         if len(event_data['payment_method_options']) > 0: #Si es mayor que cero checa si existe card.
-            print("Ancho del contenido de card: ")
             if len(event_data['payment_method_options']["card"]) > 1: #Si el contenido de card es mayor de uno, si existe y puedes finalmente procesar el pago.
                 cus_id = event_data['customer']
                 print("Customer directo:", cus_id)