Moibe commited on
Commit
6c76d4b
·
1 Parent(s): eda7ff7

Obtener correctamente documento de usuario de splashmixai

Browse files
Files changed (2) hide show
  1. fireWhale.py +33 -1
  2. herramientas.py +2 -6
fireWhale.py CHANGED
@@ -208,4 +208,36 @@ def probar_conexion_splashmix():
208
  return True
209
  except Exception as e:
210
  print(f"❌ Error al conectar con Firestore de splashmix-ai: {e}")
211
- return False
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
208
  return True
209
  except Exception as e:
210
  print(f"❌ Error al conectar con Firestore de splashmix-ai: {e}")
211
+ return False
212
+
213
+ def obtener_usuario_splashmix(firebase_user):
214
+ """
215
+ Obtiene los datos de un usuario de la colección usuarios_ig en splashmix-ai.
216
+
217
+ Args:
218
+ firebase_user (str): El ID del usuario en Firebase.
219
+
220
+ Returns:
221
+ dict: Los datos del usuario si existe, None si no existe.
222
+ """
223
+ try:
224
+ print(f"\n--- Buscando usuario '{firebase_user}' en splashmix-ai (usuarios_ig) ---")
225
+
226
+ doc_ref = db_splashmix.collection('usuarios_ig').document(firebase_user)
227
+ documento = doc_ref.get()
228
+
229
+ if documento.exists:
230
+ datos_usuario = documento.to_dict()
231
+ print(f"✔️ Usuario encontrado en splashmix-ai!")
232
+ print(f" ID del documento: {firebase_user}")
233
+ print(f" Datos del usuario:")
234
+ for key, value in datos_usuario.items():
235
+ print(f" - {key}: {value}")
236
+ return datos_usuario
237
+ else:
238
+ print(f"❌ Usuario '{firebase_user}' NO encontrado en usuarios_ig de splashmix-ai.")
239
+ return None
240
+
241
+ except Exception as e:
242
+ print(f"❌ Error al obtener usuario de splashmix-ai: {e}")
243
+ return None
herramientas.py CHANGED
@@ -37,12 +37,8 @@ def registrar_evento(cus: str, firebase_user: str, tokens: int, sitio: str = Non
37
 
38
  if sitio == 'crea-imagen':
39
  print("llegué al otro firestore")
40
- # Probar conexión al Firestore de splashmix-ai
41
- conexion_exitosa = fireWhale.probar_conexion_splashmix()
42
- if conexion_exitosa:
43
- print("✔️ Prueba de conexión exitosa!")
44
- else:
45
- print("❌ Fallo en la prueba de conexión")
46
  else:
47
  fireWhale.incrementar_campo_numerico('usuarios', firebase_user, 'tokens', amount=tokens)
48
  fireWhale.editaDato('usuarios', firebase_user, 'compro', True)
 
37
 
38
  if sitio == 'crea-imagen':
39
  print("llegué al otro firestore")
40
+ # Obtener datos del usuario de splashmix-ai
41
+ datos_usuario = fireWhale.obtener_usuario_splashmix(firebase_user)
 
 
 
 
42
  else:
43
  fireWhale.incrementar_campo_numerico('usuarios', firebase_user, 'tokens', amount=tokens)
44
  fireWhale.editaDato('usuarios', firebase_user, 'compro', True)