moved accept_list from habitListe to User
This commit is contained in:
parent
69b3c257d1
commit
5f75997617
@ -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)
|
||||||
|
|||||||
@ -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):
|
||||||
@ -76,3 +76,6 @@ class User(UserMixin):
|
|||||||
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)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user