From 1b665b8be0f322230b13605c84e23468934f7879 Mon Sep 17 00:00:00 2001 From: Yapollon Date: Fri, 1 Mar 2024 08:13:58 +0100 Subject: [PATCH] Improved user addition --- models/HabitList.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/models/HabitList.py b/models/HabitList.py index 110af65..e20ac25 100644 --- a/models/HabitList.py +++ b/models/HabitList.py @@ -62,7 +62,10 @@ class HabitList: # Adds a User by email to the HabitList def add_user(self, email: str): user = User.get_by_email(email) - add_user(self.id, user.id) + if user: + add_user(self.id, user.id) + else: + return None # Removes a User from the HabitList