CimeyClust/HabitTracker
1
1
Fork 0
Code Issues Pull Requests Packages Projects 2 Releases Wiki Activity

Fixed user ownership check if habit_lists

Browse Source
This commit is contained in:
Verox 2024-03-14 19:50:50 +01:00
parent e57cc58b3e
commit fce1593e32
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
db/SQLiteClient.py
View File

@ -324,7 +324,7 @@ def habitList_update_slot(id: int, slot: int):
def get_habitLists(user_id: int): def get_habitLists(user_id: int):
query = (f"SELECT habit_lists.*, habit_users.user_id, habit_users.accepted FROM habit_lists JOIN habit_users ON habit_lists.id = habit_users.list_id " query = (f"SELECT habit_lists.*, habit_users.accepted FROM habit_lists JOIN habit_users ON habit_lists.id = habit_users.list_id "
f"WHERE habit_users.user_id = {user_id};") f"WHERE habit_users.user_id = {user_id};")
conn = con3() conn = con3()
cursor = conn.cursor() cursor = conn.cursor()

3
models/User.py
View File

@ -54,10 +54,9 @@ class User(UserMixin):
raw_habitLists = get_habitLists(self.id) raw_habitLists = get_habitLists(self.id)
habitLists = [] habitLists = []
for habitList in raw_habitLists: for habitList in raw_habitLists:
user_id = habitList[5]
accepted = habitList[6] accepted = habitList[6]
habitList = HabitList(habitList[0], habitList[1], habitList[2], habitList[3]) habitList = HabitList(habitList[0], habitList[1], habitList[2], habitList[3])
if accepted == 1 or habitList.get_users()[0].id == user_id: if accepted == 1:
habitLists.append(habitList) habitLists.append(habitList)
return habitLists return habitLists