CimeyClust/HabitTracker
1
1
Fork 0
Code Issues Pull Requests Packages Projects 2 Releases Wiki Activity

Reverted merge commit from model

Browse Source
This commit is contained in:
Verox001 2024-02-20 09:58:34 +01:00
parent f0275108a5
commit 94fa2f567f
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
models/User.py
View File

@ -1,3 +1,5 @@
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, update_user, delete_user, from db.SQLiteClient import (create_user, get_user, get_user_by_email, update_user, delete_user,
get_habitLists, get_heatmap_value) get_habitLists, get_heatmap_value)
@ -59,7 +61,14 @@ class User(UserMixin):
# Returns all heatmap-values from the last 28 days # Returns all heatmap-values from the last 28 days
def get_heatmap(self) -> list: def get_heatmap(self) -> list:
# get current day of week as integer. monday is 1 and sunday is 7
weekday = datetime.today().weekday() + 1
heatmap = [] heatmap = []
# append the heatmap values of the current week
for day in range(0, weekday):
heatmap.append(0)
for day in range (0, 28): for day in range (0, 28):
value = get_heatmap_value(self.id, day) value = get_heatmap_value(self.id, day)
heatmap.append(value) heatmap.append(value)