Fix template ID mismatch - Support both 'default' and 'default-audit-checklist' IDs - Use existing template in database - All CRUD operations now work with existing template

main.py

@@ -745,8 +745,13 @@ async def ensure_default_template_exists():
     """
     try:
         logger.info("Checking if default template exists...")
-        # Check if template exists
-        template = await db.checklist_templates.find_one({"templateId": "default"})
+        # Check if template exists (check both old and new IDs)
+        template = await db.checklist_templates.find_one({
+            "$or": [
+                {"templateId": "default"},
+                {"templateId": "default-audit-checklist"}
+            ]
+        })
         
         if not template:
             logger.info("Default template not found, creating it now...")
@@ -1031,11 +1036,8 @@ async def add_template_item(item_data: dict):
     try:
         logger.info(f"Adding new item to shared template")
         
-        # Ensure template exists first
-        await ensure_default_template_exists()
-        
-        # Get the template
-        template = await db.checklist_templates.find_one({"templateId": "default"})
+        # Ensure template exists first and get it
+        template = await ensure_default_template_exists()
         
         if not template:
             raise HTTPException(
@@ -1043,6 +1045,8 @@ async def add_template_item(item_data: dict):
                 detail="Template not found"
             )
         
+        template_id = template.get("templateId", "default")
+        
         # Find the section and add the new item
         section_id = item_data.get('sectionId')
         section_index = next((i for i, s in enumerate(template['sections']) if s['id'] == section_id), None)
@@ -1060,7 +1064,7 @@ async def add_template_item(item_data: dict):
         
         # Update the template
         result = await db.checklist_templates.update_one(
-            {"templateId": "default", "sections.id": section_id},
+            {"templateId": template_id, "sections.id": section_id},
             {"$push": {"sections.$.items": new_item}}
         )
         
@@ -1104,8 +1108,9 @@ async def update_template_item(item_id: str, item_data: dict):
     try:
         logger.info(f"Updating item {item_id} in shared template")
         
-        # Ensure template exists first
-        await ensure_default_template_exists()
+        # Ensure template exists first and get it
+        template = await ensure_default_template_exists()
+        template_id = template.get("templateId") if template else "default"
         
         # Prepare update fields
         update_fields = {}
@@ -1118,9 +1123,9 @@ async def update_template_item(item_id: str, item_data: dict):
                 detail="No valid fields to update"
             )
         
-        # Update the item in template
+        # Update the item in template (use the template ID we found)
         result = await db.checklist_templates.update_one(
-            {"templateId": "default"},
+            {"templateId": template_id},
             {"$set": update_fields},
             array_filters=[{"item.id": item_id}]
         )
@@ -1164,12 +1169,13 @@ async def delete_template_item(item_id: str):
     try:
         logger.info(f"Deleting item {item_id} from shared template")
         
-        # Ensure template exists first
-        await ensure_default_template_exists()
+        # Ensure template exists first and get it
+        template = await ensure_default_template_exists()
+        template_id = template.get("templateId") if template else "default"
         
         # Remove the item from the template
         result = await db.checklist_templates.update_one(
-            {"templateId": "default"},
+            {"templateId": template_id},
             {"$pull": {"sections.$[].items": {"id": item_id}}}
         )
         
@@ -1211,11 +1217,8 @@ async def add_template_section(section_data: dict):
     try:
         logger.info(f"Adding new section to shared template")
         
-        # Ensure template exists first
-        await ensure_default_template_exists()
-        
-        # Get the template
-        template = await db.checklist_templates.find_one({"templateId": "default"})
+        # Ensure template exists first and get it
+        template = await ensure_default_template_exists()
         
         if not template:
             raise HTTPException(
@@ -1223,6 +1226,8 @@ async def add_template_section(section_data: dict):
                 detail="Template not found"
             )
         
+        template_id = template.get("templateId", "default")
+        
         # Create new section
         new_section = {
             "id": section_data.get('id', f"S{len(template['sections'])+1}"),
@@ -1233,7 +1238,7 @@ async def add_template_section(section_data: dict):
         
         # Add the new section
         result = await db.checklist_templates.update_one(
-            {"templateId": "default"},
+            {"templateId": template_id},
             {"$push": {"sections": new_section}}
         )
         
@@ -1276,12 +1281,13 @@ async def delete_template_section(section_id: str):
     try:
         logger.info(f"Deleting section {section_id} from shared template")
         
-        # Ensure template exists first
-        await ensure_default_template_exists()
+        # Ensure template exists first and get it
+        template = await ensure_default_template_exists()
+        template_id = template.get("templateId") if template else "default"
         
         # Remove the section from the template
         result = await db.checklist_templates.update_one(
-            {"templateId": "default"},
+            {"templateId": template_id},
             {"$pull": {"sections": {"id": section_id}}}
         )