From 3858a14f1ac01648390b3b70a79912ed6712c6c9 Mon Sep 17 00:00:00 2001 From: Lili Date: Sun, 26 Mar 2023 12:22:54 +0200 Subject: [PATCH] add goal changing to refresh_milestones() --- tlapbot/db.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tlapbot/db.py b/tlapbot/db.py index cc2d443..7032acd 100644 --- a/tlapbot/db.py +++ b/tlapbot/db.py @@ -108,11 +108,18 @@ def refresh_milestones(): "SELECT goal FROM milestones WHERE name = ?", (redeem,) ) - if cursor.fetchone() is None: + result = cursor.fetchone() + if result is None: cursor.execute( "INSERT INTO milestones(name, progress, goal, complete) VALUES(?, 0, ?, FALSE)", (redeem, redeem_info['goal']) ) + # update existing milestone to new goal + elif result != redeem_info["goal"]: + cursor.execute( + "UPDATE milestones SET goal = ? WHERE name = ?", + (redeem_info["goal"], redeem) + ) db.commit() except sqlite3.Error as e: print("Failed inserting milestones to db:", e.args[0])