Compare commits
2 Commits
45f23c88e2
...
69b3c257d1
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
69b3c257d1 | ||
|
|
db45e7f059 |
@ -322,6 +322,15 @@ def add_user(list_id: int, user_id: int):
|
|||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
|
def accept_List(list_id: int, user_id: int):
|
||||||
|
query = (f"UPDATE habit_users SET accepted = true WHERE {user_id} = habit_users.user_id AND {list_id} = habit_users.list_id;")
|
||||||
|
conn = con3()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(query)
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def remove_user(list_id: int, user_id: int):
|
def remove_user(list_id: int, user_id: int):
|
||||||
query = f"DELETE FROM habit_users WHERE user_id = {user_id} AND list_id = {list_id};"
|
query = f"DELETE FROM habit_users WHERE user_id = {user_id} AND list_id = {list_id};"
|
||||||
|
|||||||
@ -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)
|
update_habitList, delete_habitList, accept_List)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
@ -66,6 +66,8 @@ 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):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user