From 47b38ea7f9e8c4e57d5463dc68271819b823d023 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Mon, 1 Apr 2019 15:58:14 +0100 Subject: Move task assignment to separate module Also, add automatic assign on user signup --- app/auth/routes.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'app/auth') diff --git a/app/auth/routes.py b/app/auth/routes.py index c5c175c..73cbd6a 100644 --- a/app/auth/routes.py +++ b/app/auth/routes.py @@ -22,6 +22,7 @@ from app.auth.email import ( send_password_reset_email, send_email_confirmation_email, ) +from app.utils.tasks import create_initial_user_tasks @bp.route("/login", methods=("GET", "POST")) @@ -122,7 +123,13 @@ def confirm_email(token): else: user.is_confirmed = True db.session.commit() + for task in create_initial_user_tasks(user): + if task is None: + break + db.session.add(task) + db.session.commit() flash("Account confirmed successfully. Thank you!", "success") + return redirect(url_for("auth.login")) return redirect(url_for("main.index")) -- cgit v1.2.3