diff --git a/models/Habit.py b/models/Habit.py index 34edf10..9c26954 100644 --- a/models/Habit.py +++ b/models/Habit.py @@ -80,6 +80,9 @@ class Habit: count = 0 self.checked = False for tracking in self.get_habitTrackings(): + if tracking.created_at.date() == datetime.today().date(): + self.checked = True + # day if self.unit == 0: if tracking.created_at.date() == datetime.today().date(): diff --git a/models/User.py b/models/User.py index e4f2384..13f9f72 100644 --- a/models/User.py +++ b/models/User.py @@ -1,8 +1,8 @@ from datetime import datetime from flask_login import UserMixin -from db.SQLiteClient import create_user, get_user, get_user_by_email, get_habits, delete_user, update_user, \ - get_habitLists +from db.SQLiteClient import create_user, get_user, get_user_by_email, delete_user, update_user, \ + get_habitLists, get_heatmap_value class User(UserMixin): @@ -43,3 +43,10 @@ class User(UserMixin): habitLists.append(habitList) return habitLists + + def get_heatmap(self): + heatmap = [] + for day in range (0, 27): + value = get_heatmap_value(self.id, day) + heatmap.append(value) + return heatmap \ No newline at end of file diff --git a/templates/index.html b/templates/index.html index caaf011..e3a58d3 100644 --- a/templates/index.html +++ b/templates/index.html @@ -1,187 +1,180 @@ {% extends 'layouts/main.html' %} {% block content %} -