Create sql_db.py

modules/database/sql_db.py

@@ -0,0 +1,42 @@
+# modules/database/sql_db.py
+
+from modules.database.database_init import get_database_client
+import logging
+
+logger = logging.getLogger(__name__)
+
+# Nombre de la base de datos y contenedor
+DATABASE_NAME = "user_database"
+USER_CONTAINER_NAME = "users"
+
+def get_user_container():
+    """Obtiene el contenedor de usuarios."""
+    db_client = get_database_client(DATABASE_NAME)
+    return db_client.get_container_client(USER_CONTAINER_NAME)
+
+def create_user(username, password, role):
+    """Crea un nuevo usuario."""
+    try:
+        container = get_user_container()
+        user_data = {
+            "id": username,
+            "password": password,
+            "role": role
+        }
+        container.create_item(body=user_data)
+        logger.info(f"Usuario creado: {username}")
+        return True
+    except Exception as e:
+        logger.error(f"Error creando usuario: {e}")
+        return False
+
+def get_user(username):
+    """Obtiene un usuario por su username."""
+    try:
+        container = get_user_container()
+        query = f"SELECT * FROM c WHERE c.id = '{username}'"
+        items = list(container.query_items(query=query, enable_cross_partition_query=True))
+        return items[0] if items else None
+    except Exception as e:
+        logger.error(f"Error obteniendo usuario: {e}")
+        return None