From aaeb04d4abd8d0f0060490e772b9fa24e0a29c82 Mon Sep 17 00:00:00 2001
From: Yapollon <yasschueler@gmail.com>
Date: Wed, 14 Feb 2024 10:52:25 +0100
Subject: [PATCH] Heatmap done?

yes
---
 models/User.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

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