Almost done with voting!!

1 files changed, 8 insertions, 0 deletions

diff --git a/src/db.py b/src/db.py
index b971a13..f1ae4f6 100644
--- a/src/db.py
+++ b/src/db.py
@@ -168,3 +168,11 @@ async def get_user_vote(deployment_id: int, voter_id: str):
 async def get_vote_counts(deployment_id: int):
     votes = await get_votes(deployment_id)
     return len(votes["approve"]), len(votes["reject"])
+
+async def update_voting_message_ts(deployment_id: int, message_ts: str):
+    async with aiosqlite.connect(DB_PATH) as db:
+        await db.execute(
+            "UPDATE Deployments SET voting_message_ts = ? WHERE id = ?",
+            (message_ts, deployment_id)
+        )
+        await db.commit()