moved accept_list from habitListe to User

This commit is contained in:
Luis 2024-03-07 18:44:13 +01:00
parent 69b3c257d1
commit 5f75997617
2 changed files with 6 additions and 6 deletions

View File

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

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, accept_List)
class User(UserMixin): class User(UserMixin):
@ -75,4 +75,7 @@ class User(UserMixin):
heatmap.append(value) heatmap.append(value)
heatmap.reverse() heatmap.reverse()
day = 27-weekday day = 27-weekday
return heatmap, day return heatmap, day
def accept_List(self, HabitList_id):
accept_List(self.id, HabitList_id)