Update modules/auth.py

modules/auth.py

@@ -1,17 +1,37 @@
+import os
 from azure.cosmos import CosmosClient, exceptions
 import bcrypt
-import os
+import base64
+
+def clean_and_validate_key(key):
+    key = key.strip()
+    while len(key) % 4 != 0:
+        key += '='
+    try:
+        base64.b64decode(key)
+        return key
+    except:
+        raise ValueError("La clave proporcionada no es válida")
 
 # Azure Cosmos DB configuration
 endpoint = os.environ.get("COSMOS_ENDPOINT")
 key = os.environ.get("COSMOS_KEY")
-database_name = "user_database"
-container_name = "users"
 
-# Initialize the Cosmos client
-client = CosmosClient(endpoint, key)
-database = client.get_database_client(database_name)
-container = database.get_container_client(container_name)
+if not endpoint or not key:
+    raise ValueError("Las variables de entorno COSMOS_ENDPOINT y COSMOS_KEY deben estar configuradas")
+
+key = clean_and_validate_key(key)
+
+try:
+    client = CosmosClient(endpoint, key)
+    database = client.get_database_client("user_database")
+    container = database.get_container_client("users")
+    # Prueba de conexión
+    database_list = list(client.list_databases())
+    print(f"Conexión exitosa. Bases de datos encontradas: {len(database_list)}")
+except Exception as e:
+    print(f"Error al conectar con Cosmos DB: {str(e)}")
+    raise
 
 def hash_password(password):
     """Hash a password for storing."""
@@ -24,17 +44,12 @@ def verify_password(stored_password, provided_password):
 def register_user(username, password, role, additional_info=None):
     """Register a new user."""
     try:
-        # Check if user already exists
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
-        existing_user = list(container.query_items(
-            query=query,
-            enable_cross_partition_query=True
-        ))
+        existing_user = list(container.query_items(query=query, enable_cross_partition_query=True))
         
         if existing_user:
             return False  # User already exists
         
-        # Create new user document
         new_user = {
             'id': username,
             'password': hash_password(password),
@@ -51,10 +66,7 @@ def authenticate_user(username, password):
     """Authenticate a user."""
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
-        results = list(container.query_items(
-            query=query,
-            partition_key=username
-        ))
+        results = list(container.query_items(query=query, partition_key=username))
         
         if results:
             stored_user = results[0]
@@ -62,15 +74,14 @@ def authenticate_user(username, password):
                 return True
     except exceptions.CosmosHttpResponseError:
         pass
+    
+    return False
 
 def get_user_role(username):
     """Get the role of a user."""
     try:
         query = f"SELECT c.role FROM c WHERE c.id = '{username}'"
-        results = list(container.query_items(
-            query=query,
-            partition_key=username
-        ))
+        results = list(container.query_items(query=query, partition_key=username))
         
         if results:
             return results[0]['role']
@@ -83,10 +94,7 @@ def update_user_info(username, new_info):
     """Update user information."""
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
-        results = list(container.query_items(
-            query=query,
-            partition_key=username
-        ))
+        results = list(container.query_items(query=query, partition_key=username))
         
         if results:
             user = results[0]
@@ -102,10 +110,7 @@ def delete_user(username):
     """Delete a user."""
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
-        results = list(container.query_items(
-            query=query,
-            partition_key=username
-        ))
+        results = list(container.query_items(query=query, partition_key=username))
         
         if results:
             user = results[0]