From 9345490bf359d9fb5059f986949a7a29f3f1f4c9 Mon Sep 17 00:00:00 2001
From: Anuj-Gupta4 <anujgupta.4388@gmail.com>
Date: Mon, 3 Mar 2025 17:40:09 +0545
Subject: [PATCH] feat(project): populate total task field after task creation

---
 src/backend/app/db/models.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/backend/app/db/models.py b/src/backend/app/db/models.py
index fff93caf9..e542cc3d1 100644
--- a/src/backend/app/db/models.py
+++ b/src/backend/app/db/models.py
@@ -1005,7 +1005,18 @@ async def create(
             await cur.execute(sql, data)
             result = await cur.fetchall()
 
-        return bool(result)
+            if success := bool(result):
+                update_project_sql = """
+                    UPDATE projects
+                    SET total_tasks = (
+                        SELECT COALESCE(MAX(project_task_index), 0)
+                        FROM public.tasks WHERE project_id = %(project_id)s
+                    )
+                    WHERE id = %(project_id)s;
+                """
+                await cur.execute(update_project_sql, {"project_id": project_id})
+
+        return success
 
 
 class DbProject(BaseModel):