UPSERT when creating timers

matrix-org · Apr 27, 2020 · 7d9c9e7 · 7d9c9e7
1 parent 8b404a5
commit 7d9c9e7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/fcp_timers.py b/fcp_timers.py
@@ -95,8 +95,15 @@ def _db_save_timer(self, timestamp: datetime, proposal_num: int):
         with self.store.conn:
             self.store.cur.execute("""
             INSERT INTO fcp_timers (proposal_num, end_timestamp)
-            VALUES (%s, %s)
-            """, (proposal_num, timestamp.timestamp()))
+                VALUES (%s, %s)
+            ON CONFLICT (proposal_num)
+            DO UPDATE
+                SET end_timestamp = %s
+                WHERE fcp_timers.proposal_num = %s
+            """, (
+                proposal_num, timestamp.timestamp(),
+                proposal_num, timestamp.timestamp(),
+            ))
 
     def _db_delete_timer(self, proposal_num: int):
         """Deletes a timer from the database"""