Update app.py

app.py

@@ -170,34 +170,52 @@ def update_record(container, updated_record):
     except Exception as e:
         return False, f"Error: {traceback.format_exc()} 😱"
 
-# 🗑️ Delete record – query first to get the proper partition key value from "testPART"
-def delete_record(container, record, partition_key_field="testPART"):
+# 🗑️ Delete a record from Cosmos DB
+def delete_record(container, record, partition_key_field=None):
+    """
+    Delete a record from Cosmos DB using its ID and partition key.
+    
+    Args:
+        container: Cosmos DB container client
+        record: Dictionary containing at least 'id' and optionally the partition key field
+        partition_key_field: Name of the field containing the partition key value (optional)
+    
+    Returns:
+        tuple: (success: bool, message: str)
+    """
     try:
-        # Query for the document using its id
-        query = "SELECT * FROM c WHERE c.id=@id"
-        parameters = [{"name": "@id", "value": record["id"]}]
-        docs = list(container.query_items(
-            query=query,
-            parameters=parameters,
-            enable_cross_partition_query=True
-        ))
-        if not docs:
-            # Document not found; treat as already deleted
-            return True, f"Record {record['id']} not found (already deleted). 🗑️"
-        
-        doc = docs[0]
-        # Retrieve the partition key value from the document (adjust field name as needed)
-        partition_key_value = doc.get(partition_key_field)
-        if partition_key_value is None:
-            return False, f"Partition key field '{partition_key_field}' not found in document {doc['id']}."
-        
-        # Delete the document using its id and the actual partition key value
-        container.delete_item(item=doc["id"], partition_key=partition_key_value)
-        return True, f"Record {doc['id']} successfully deleted. 🗑️"
+        # Ensure record has an ID
+        if "id" not in record:
+            return False, "Record must contain an 'id' field. 🛑"
+
+        doc_id = record["id"]
+
+        # Determine partition key value
+        if partition_key_field:
+            # Use specified partition key field from the record
+            partition_key_value = record.get(partition_key_field)
+            if partition_key_value is None:
+                return False, f"Partition key field '{partition_key_field}' not found in record {doc_id}. 🛑"
+        else:
+            # If no partition key field specified, try to use the ID as the partition key
+            # (common when partition key is same as ID)
+            partition_key_value = doc_id
+
+        # Perform the deletion
+        container.delete_item(item=doc_id, partition_key=partition_key_value)
+        return True, f"Record {doc_id} successfully deleted. 🗑️"
+
+    except exceptions.CosmosResourceNotFoundError:
+        # Document doesn't exist, treat as success since goal is deletion
+        return True, f"Record {doc_id} not found (already deleted?). 🗑️"
     except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error: {str(e)} 🚨"
+        # Specific HTTP errors (e.g., wrong partition key)
+        return False, f"HTTP error deleting {doc_id}: {str(e)}. 🚨"
     except Exception as e:
-        return False, f"Unexpected error: {traceback.format_exc()} 😱"
+        # Unexpected errors with full traceback
+        return False, f"Unexpected error deleting {doc_id}: {str(traceback.format_exc())}. 😱"
+
+        
 
 
 # 💾 Save a new document to Cosmos DB with extra fields