Update modules/auth.py

modules/auth.py

@@ -26,7 +26,10 @@ def register_user(username, password, role, additional_info=None):
     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
@@ -39,7 +42,7 @@ def register_user(username, password, role, additional_info=None):
             'additional_info': additional_info or {}
         }
         
-        container.create_item(body=new_user)
+        container.create_item(body=new_user, partition_key=username)
         return True
     except exceptions.CosmosHttpResponseError:
         return False
@@ -48,7 +51,10 @@ 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, enable_cross_partition_query=True))
+        results = list(container.query_items(
+            query=query,
+            partition_key=username
+        ))
         
         if results:
             stored_user = results[0]
@@ -56,14 +62,15 @@ 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, enable_cross_partition_query=True))
+        results = list(container.query_items(
+            query=query,
+            partition_key=username
+        ))
         
         if results:
             return results[0]['role']
@@ -76,12 +83,15 @@ 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, enable_cross_partition_query=True))
+        results = list(container.query_items(
+            query=query,
+            partition_key=username
+        ))
         
         if results:
             user = results[0]
             user['additional_info'].update(new_info)
-            container.upsert_item(user)
+            container.upsert_item(user, partition_key=username)
             return True
     except exceptions.CosmosHttpResponseError:
         pass
@@ -92,11 +102,14 @@ def delete_user(username):
     """Delete a user."""
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
-        results = list(container.query_items(query=query, enable_cross_partition_query=True))
+        results = list(container.query_items(
+            query=query,
+            partition_key=username
+        ))
         
         if results:
             user = results[0]
-            container.delete_item(item=user, partition_key=username)
+            container.delete_item(item=user['id'], partition_key=username)
             return True
     except exceptions.CosmosHttpResponseError:
         pass