models/HabitList.py

@@ -3,7 +3,7 @@ from dataclasses import dataclass
 from models.Habit import Habit
 from models.User import User
 from db.SQLiteClient import (create_habitList, get_habitList, get_habits, get_users, add_user, remove_user,
-                             update_habitList, delete_habitList)
+                             update_habitList, delete_habitList, accept_List)
 
 
 @dataclass
@@ -66,6 +66,9 @@ class HabitList:
         else:
             return None
 
+    def accept_List(self, user:User):
+        accept_List(self.id, user.id)
+
     # Removes a User from the HabitList
     def remove_user(self, user_id):
         remove_user(self.id, user_id)

models/User.py

@@ -2,7 +2,7 @@ from datetime import datetime
 
 from flask_login import UserMixin
 from db.SQLiteClient import (create_user, get_user, get_user_by_email, update_user, delete_user,
-                             get_habitLists, get_heatmap_value, accept_List)
+                             get_habitLists, get_heatmap_value)
 
 
 class User(UserMixin):
@@ -76,6 +76,3 @@ class User(UserMixin):
         heatmap.reverse()
         day = 27-weekday
         return heatmap, day
-
-    def accept_List(self, HabitList_id):
-        accept_List(self.id, HabitList_id)