Update modules/database.py

modules/database.py

@@ -71,18 +71,30 @@ def initialize_mongodb_connection():
         return False
 
 #######################################################################################################
+def create_user(username, password, role):
+    try:
+        hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
+        user_data = {
+            'id': username,
+            'password': hashed_password,
+            'role': role,
+            'created_at': datetime.utcnow().isoformat()
+        }
+        user_container.create_item(body=user_data)
+        print(f"Usuario {role} creado: {username}")  # Log para depuración
+        return True
+    except Exception as e:
+        print(f"Error al crear usuario {role}: {str(e)}")  # Log para depuración
+        return False
 
+#######################################################################################################
 def create_admin_user(username, password):
-    hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
-    user_data = {
-        'id': username,
-        'password': hashed_password,
-        'is_invited': False,
-        'role': 'Administrador',
-        'created_at': datetime.utcnow().isoformat()
-    }
-    user_container.create_item(body=user_data)
-
+    return create_user(username, password, 'Administrador')
+    
+#######################################################################################################
+def create_student_user(username, password):
+    return create_user(username, password, 'Estudiante')
+       
 #######################################################################################################
 # Funciones para Cosmos DB SQL API (manejo de usuarios)
 def get_user(username):
@@ -99,15 +111,6 @@ def get_user(username):
         print(f"Error al obtener usuario {username}: {str(e)}")
         return None
 
-#######################################################################################################
-def create_user(user_data):
-    try:
-        user_container.create_item(body=user_data)
-        return True
-    except Exception as e:
-        logger.error(f"Error al crear usuario: {str(e)}")
-        return False
-
 ################################################################################
 # Funciones para Cosmos DB MongoDB API (análisis de texto)