Heatmap done?

yes
This commit is contained in:
Yapollon 2024-02-14 10:52:25 +01:00
parent 0f060e81a4
commit aaeb04d4ab

View File

@ -1,8 +1,8 @@
from datetime import datetime from datetime import datetime
from flask_login import UserMixin from flask_login import UserMixin
from db.SQLiteClient import create_user, get_user, get_user_by_email, get_habits, delete_user, update_user, \ from db.SQLiteClient import create_user, get_user, get_user_by_email, delete_user, update_user, \
get_habitLists get_habitLists, get_heatmap_value
class User(UserMixin): class User(UserMixin):
@ -43,3 +43,10 @@ class User(UserMixin):
habitLists.append(habitList) habitLists.append(habitList)
return habitLists 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