Add user to IAP group (#12)

google · Feb 20, 2024 · 7b6a825 · 7b6a825
1 parent 3be98d2
commit 7b6a825
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/timesketch/views/auth.py b/timesketch/views/auth.py
@@ -104,6 +104,9 @@ def login():
                 email = decoded_jwt.get("email")
                 if email:
                     user = User.get_or_create(username=email, name=email)
+                    group = Group.get_or_create(name="GOOGLE-IAP")
+                    user.groups.append(group)
+                    db_session.commit()
                     login_user(user)
 
             except (ImportError, NameError, UnboundLocalError):