Improved user addition

2024-03-01 08:13:58 +01:00 · 2024-03-01 08:13:58 +01:00 · 1b665b8be0
commit 1b665b8be0
parent a8a3382f15
1 changed files with 4 additions and 1 deletions
--- 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)
+        if user:
            add_user(self.id, user.id)
+        else:
+            return None


    # Removes a User from the HabitList