lili
/
Tlapbot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 6 Wiki Activity

fix reset_milestone

This commit is contained in:
Lili (Tlapka) 2023-07-03 12:18:29 +02:00
parent 348e674f74
commit d484a0a363
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tlapbot/db.py
View File

@ -127,21 +127,22 @@ def refresh_milestones():
def reset_milestone(milestone): def reset_milestone(milestone):
if not redeem_name in current_app.config['REDEEMS']: if not milestone in current_app.config['REDEEMS']:
print(f"Failed resetting milestone, {milestone} not in redeems file.") print(f"Failed resetting milestone, {milestone} not in redeems file.")
return False return False
try: try:
db = get_db()
db.execute( db.execute(
"DELETE FROM milestones WHERE name = ?", "DELETE FROM milestones WHERE name = ?",
(milestone,) (milestone,)
) )
db.execute( db.execute(
"INSERT INTO milestones(name, progress, goal) VALUES(?, ?, ?)", "INSERT INTO milestones(name, progress, goal, complete) VALUES(?, ?, ?, FALSE)",
(milestone, 0, current_app.config['REDEEMS'][milestone]['goal']) (milestone, 0, current_app.config['REDEEMS'][milestone]['goal'])
) )
db.commit() db.commit()
return True return True
except Error as e: except sqlite3.Error as e:
current_app.logger.error(f"Error occured adding a milestone: {e.args[0]}") current_app.logger.error(f"Error occured adding a milestone: {e.args[0]}")
return False return False
@ -193,7 +194,7 @@ def refresh_milestones_command():
@click.argument('milestone') @click.argument('milestone')
def reset_milestone_command(milestone): def reset_milestone_command(milestone):
"""Resets a completed milestone back to zero.""" """Resets a completed milestone back to zero."""
if milestone_complete(milestone): if milestone_complete(get_db(), milestone):
if reset_milestone(milestone): if reset_milestone(milestone):
click.echo(f"Reset milestone {milestone}.") click.echo(f"Reset milestone {milestone}.")
else: else: