Compare commits

...

2 Commits

Author SHA1 Message Date
Yapollon
17a3f7bacc Merge branch 'master' of https://repo.cimeyclust.com/CimeyClust/HabitTracker 2024-03-06 11:14:27 +01:00
Yapollon
adeab273a6 Update User.py 2024-03-06 11:14:15 +01:00

View File

@ -2,7 +2,7 @@ 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, update_user_profile)
class User(UserMixin): class User(UserMixin):
@ -34,6 +34,8 @@ class User(UserMixin):
def update(self): def update(self):
update_user(self.id, self.name, self.email, self.password if self.password else None) update_user(self.id, self.name, self.email, self.password if self.password else None)
def update_profile_image(self, profile_image: str):
update_user_profile(self.id, profile_image)
# Deletes the User # Deletes the User
def delete(self): def delete(self):
@ -72,4 +74,5 @@ class User(UserMixin):
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)
heatmap.reverse()
return heatmap return heatmap