Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -9,12 +9,15 @@ import uuid
 logger = logging.getLogger(__name__)
 
 def get_user(username, role=None):
-    user_container, _, _ = get_sql_containers()
+    container = get_container("users")
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
         if role:
             query += f" AND c.role = '{role}'"
-        items = list(user_container.query_items(query=query, enable_cross_partition_query=True))
+        items = list(container.query_items(
+            query=query,
+            partition_key=username
+        ))
         return items[0] if items else None
     except Exception as e:
         logger.error(f"Error al obtener usuario {username}: {str(e)}")
@@ -62,20 +65,20 @@ def create_admin_user(username, password, additional_info=None):
 def record_login(username):
     """Registra el inicio de sesión de un usuario"""
     try:
-        _, _, session_container = get_sql_containers()
-        
+        container = get_container("users_sessions")
+        if not container:
+            return None
+
         session_doc = {
             "id": str(uuid.uuid4()),
             "type": "session",
             "username": username,
-            "loginTime": datetime.now(timezone.utc).isoformat(),
-            "partitionKey": username  # Asegurarse que coincida con la configuración
+            "loginTime": datetime.now(timezone.utc).isoformat()
         }
 
-        # Crear el documento sin enable_cross_partition_query
-        result = session_container.create_item(
+        result = container.create_item(
             body=session_doc,
-            partition_key=username  # Usar partition_key en lugar de partitionKey
+            partition_key=username
         )
         logger.info(f"Sesión registrada para {username}")
         return result['id']