Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -61,19 +61,20 @@ def create_admin_user(username, password, additional_info=None):
 def record_login(username):
     """Registra el inicio de sesión de un usuario"""
     try:
-        container = get_container("user_sessions")
-        if not container:
-            return None
-
+        user_container, _, session_container = get_sql_containers()  # Obtener el contenedor correcto
+        
         session_doc = {
             "id": str(uuid.uuid4()),
             "type": "session",
             "username": username,
             "loginTime": datetime.now(timezone.utc).isoformat(),
-            "partitionKey": username
+            "partitionKey": username  # Asegurarse que esto coincida con la configuración de la colección
         }
 
-        result = container.create_item(body=session_doc)
+        result = session_container.create_item(
+            body=session_doc,
+            enable_cross_partition_query=True  # Agregado para manejar particiones
+        )
         return result['id']
     except Exception as e:
         logger.error(f"Error registrando login: {str(e)}")
@@ -82,18 +83,15 @@ def record_login(username):
 def record_logout(username, session_id):
     """Registra el cierre de sesión y calcula la duración"""
     try:
-        container = get_container("user_sessions")
-        if not container:
-            return False
-
-        # Obtener la sesión actual
+        user_container, _, session_container = get_sql_containers()
+        
         query = "SELECT * FROM c WHERE c.id = @id AND c.username = @username"
         params = [
             {"name": "@id", "value": session_id},
             {"name": "@username", "value": username}
         ]
         
-        sessions = list(container.query_items(
+        sessions = list(session_container.query_items(
             query=query,
             parameters=params,
             enable_cross_partition_query=True
@@ -107,13 +105,12 @@ def record_logout(username, session_id):
         logout_time = datetime.now(timezone.utc)
         duration = int((logout_time - login_time).total_seconds())
 
-        # Actualizar el documento
         session.update({
             "logoutTime": logout_time.isoformat(),
             "sessionDuration": duration
         })
 
-        container.upsert_item(body=session)
+        session_container.upsert_item(body=session)
         return True
     except Exception as e:
         logger.error(f"Error registrando logout: {str(e)}")
@@ -122,22 +119,19 @@ def record_logout(username, session_id):
 def get_recent_sessions(limit=10):
     """Obtiene las sesiones más recientes"""
     try:
-        container = get_container("user_sessions")
-        if not container:
-            return []
-
+        user_container, _, session_container = get_sql_containers()
+        
         query = """
         SELECT c.username, c.loginTime, c.logoutTime, c.sessionDuration
         FROM c
         WHERE c.type = 'session'
-        AND c.logoutTime != null
         ORDER BY c.loginTime DESC
         OFFSET 0 LIMIT @limit
         """
         
         params = [{"name": "@limit", "value": limit}]
         
-        sessions = container.query_items(
+        sessions = session_container.query_items(
             query=query,
             parameters=params,
             enable_cross_partition_query=True
@@ -148,10 +142,11 @@ def get_recent_sessions(limit=10):
         logger.error(f"Error obteniendo sesiones recientes: {str(e)}")
         return []
 
+
 def get_user_total_time(username):
     """Obtiene el tiempo total que un usuario ha pasado en la plataforma"""
     try:
-        container = get_container("user_sessions")
+        container = get_sql_container("users_sessions")
         if not container:
             return None