diff --git a/server/api/team.py b/server/api/team.py index 6c93ffc..fdbdd34 100644 --- a/server/api/team.py +++ b/server/api/team.py @@ -1,4 +1,4 @@ -from flask import Blueprint, request +from flask import Blueprint, request, session from flask import current_app as app from voluptuous import Schema, Length, Required @@ -35,6 +35,7 @@ def team_create(): Users.query.filter_by(uid=_user.uid).update({ "tid": team.tid }) db.session.commit() + session["tid"] = team.tid return { "success": 1, "message": "Success!" } @blueprint.route("/invite", methods=["POST"]) diff --git a/server/api/user.py b/server/api/user.py index 6f67e8b..2eff859 100644 --- a/server/api/user.py +++ b/server/api/user.py @@ -237,6 +237,8 @@ def login_user(username, password): session["sid"] = token.sid session["username"] = token.username session["admin"] = user.admin == True + if user.tid is not None and user.tid >= 0: + session["tid"] = user.tid return True