easyctf/easyctf-2017
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add team id to session

Browse source
This commit is contained in:
James Wang 2016-01-17 00:58:29 +00:00
parent 796fac7f85
commit f38063a2d5
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
server/api/team.py
View file

@ -1,4 +1,4 @@
from flask import Blueprint, request from flask import Blueprint, request, session
from flask import current_app as app from flask import current_app as app
from voluptuous import Schema, Length, Required from voluptuous import Schema, Length, Required
@ -35,6 +35,7 @@ def team_create():
Users.query.filter_by(uid=_user.uid).update({ "tid": team.tid }) Users.query.filter_by(uid=_user.uid).update({ "tid": team.tid })
db.session.commit() db.session.commit()
session["tid"] = team.tid
return { "success": 1, "message": "Success!" } return { "success": 1, "message": "Success!" }
@blueprint.route("/invite", methods=["POST"]) @blueprint.route("/invite", methods=["POST"])

2
server/api/user.py
View file

@ -237,6 +237,8 @@ def login_user(username, password):
session["sid"] = token.sid session["sid"] = token.sid
session["username"] = token.username session["username"] = token.username
session["admin"] = user.admin == True session["admin"] = user.admin == True
if user.tid is not None and user.tid >= 0:
session["tid"] = user.tid
return True return True