Customer as index to add tokens

fireWhale.py

@@ -123,4 +123,43 @@ def verificar_token(id_token):
         return uid  # Retorna el UID del usuario si el token es válido
     except auth.InvalidIdTokenError as e:
         print(f"Token inválido: {e}")
-        return None  # Retorna None si el token es inválido
+        return None  # Retorna None si el token es inválido
+    
+def encontrar_documento_por_cus(valor_cus):
+    """
+    Busca un documento en la colección 'usuario' donde el campo 'cus'
+    coincide con el valor_cus especificado.
+
+    Args:
+        valor_cus (str): El valor exacto del campo 'cus' a buscar.
+
+    Returns:
+        tuple: Una tupla (document_id, document_data) si se encuentra,
+                o (None, None) si no se encuentra o hay un error.
+                Retorna solo el primer documento encontrado si hay múltiples coincidencias.
+    """
+    print(f"\n--- Buscando documento en 'usuario' con cus: '{valor_cus}' ---")
+
+    try:
+        # Realizamos la consulta: filtra por documentos donde 'cus' sea igual a valor_cus
+        query_results = db.collection('usuario').where(
+            filter=firestore.FieldFilter('cus', '==', 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
+
+        # Si se encuentra, query_results será una lista de instantáneas de documentos
+        # Tomamos el primer resultado (ya que usamos limit(1))
+        doc_snapshot = query_results[0]
+        
+        doc_id = doc_snapshot.id
+        doc_data = doc_snapshot.to_dict()
+        
+        print(f"✔️ Documento encontrado: ID='{doc_id}', Datos: {doc_data}")
+        return doc_id, doc_data
+
+    except Exception as e:
+        print(f"❌ Error al buscar documento por 'cus': {e}")
+        return None, None

herramientas.py

@@ -23,7 +23,7 @@ def imprimeTime():
 
     return formatted_time
 
-def registrar_evento(event_type: str):
+def registrar_evento(cus: str):
     """
     Guarda la fecha y hora actual junto con un tipo de evento
     en un archivo 'registro.txt' en la raíz del proyecto.
@@ -35,24 +35,28 @@ def registrar_evento(event_type: str):
     fecha = imprimeTime() # Utiliza tu función existente
 
     # 2. Definir el nombre del archivo de registro
-    file_name = "registro.txt"
+    #file_name = "registro.txt"
 
     # 3. Obtener la ruta completa al archivo en la raíz del proyecto
     # Esto asume que el script que llama a esta función está en la raíz.
-    script_dir = os.path.dirname(os.path.abspath(__file__))
-    registro_path = os.path.join(script_dir, file_name)
+    # script_dir = os.path.dirname(os.path.abspath(__file__))
+    # registro_path = os.path.join(script_dir, file_name)
 
     # 4. Formatear la línea que se guardará en el archivo
     # Puedes elegir el formato que prefieras. Un buen formato sería CSV o algo legible.
     # Por ejemplo: "FECHA_HORA,TIPO_DE_EVENTO\n"
-    log_line = f"{fecha},{event_type}\n"
+    #log_line = f"{fecha},{cus}\n"
 
-    fireWhale.creaDato('usuarios', 'log', 'registro', log_line)
+    resultado = fireWhale.encontrar_documento_por_cus(cus)
+    print("Ésto es el resultado de buscar el cus:")
+    print(resultado)
+
+    fireWhale.editaDato('usuarios', resultado, 'tokens', 182)
 
     # 5. Abrir el archivo en modo de añadir ('a') y escribir la línea
-    try:
-        with open(registro_path, 'a') as f:
-            f.write(log_line)
-        print(f"[{fecha}] Evento '{event_type}' registrado con éxito en '{registro_path}'")
-    except Exception as e:
-        print(f"Error al escribir en el archivo de registro '{registro_path}': {e}")
+    # try:
+    #     with open(registro_path, 'a') as f:
+    #         f.write(log_line)
+    #     print(f"[{fecha}] Evento '{event_type}' registrado con éxito en '{registro_path}'")
+    # except Exception as e:
+    #     print(f"Error al escribir en el archivo de registro '{registro_path}': {e}")

main.py

@@ -34,8 +34,7 @@ async def webhook_received(request: Request, stripe_signature: str = Header(None
         )
         # print("Evento construido...") 
         # print(event)   
-        # print("Evento impreso")
-            
+        # print("Evento impreso")            
 
     except Exception as e:     
         print("Excepción es: ", e)         
@@ -45,20 +44,20 @@ async def webhook_received(request: Request, stripe_signature: str = Header(None
     print("Voy a imprimir el event type:")
     print(event_type)
     
-    if event_type == 'charge.succeeded':
-        print('charge succeed')
-        herramientas.registrar_evento(event_type)
+    if event_type == 'payment_intent.succeeded':
+        print('payment intent succeed')        
         print(event_data)
-        print("Ready")
-        #time.sleep(80)
+        print("Ready")        
         print(event_data['created'])
         print(event_data['id'])
         print(event_data['payment_intent'])
         print(event_data['payment_method'])
         print(event_data['receipt_url'])
+        print("Customer:")
+        cus = event_data['customer']
+        print(cus)
+        herramientas.registrar_evento(cus)
         
-        # autorizacion = sulkuPypi.authorize(19, 'picswap')
-        # print("Autorización: ", autorizacion)
     else:
         print(f'unhandled event: {event_type}')