From 428e2498ddd98d83f4293c9d83286b27f8bd7e8b Mon Sep 17 00:00:00 2001 From: Verox001 Date: Fri, 26 Jan 2024 11:17:09 +0100 Subject: [PATCH] Optimized habit tracking --- app.py | 28 ++++++++++++++++++++++++---- 1 file changed, 24 insertions(+), 4 deletions(-) diff --git a/app.py b/app.py index 3f2615a..111444e 100644 --- a/app.py +++ b/app.py @@ -126,9 +126,8 @@ def index(): habits = [] name = "Bitte melde dich an." - # Add checked attribute to habits (if they have been checked today) - for habit in habits: - trackings = habit.get_habitTrackings() + # old code + """trackings = habit.get_habitTrackings() for tracking in trackings: # day if habit.unit == 0: @@ -151,6 +150,16 @@ def index(): habit.checked = True break + else: + habit.checked = False""" + + # Add checked attribute to habits (if they have been checked today) + for habit in habits: + trackings = habit.get_habitTrackings() + for tracking in trackings: + if tracking.created_at.date() == datetime.date.today(): + habit.checked = True + break else: habit.checked = False @@ -324,7 +333,7 @@ def check_habit(): trackings = habit.get_habitTrackings() # Check if habit has been tracked today - delete_tracking = None + """delete_tracking = None for tracking in trackings: # day if habit.unit == 0: @@ -347,6 +356,17 @@ def check_habit(): delete_tracking = tracking break + if not delete_tracking: + HabitTrackings.create(habit_id, 1) + else: + delete_tracking.delete()""" + + delete_tracking = None + for tracking in trackings: + if tracking.created_at.date() == datetime.date.today(): + delete_tracking = tracking + break + if not delete_tracking: HabitTrackings.create(habit_id, 1) else: