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

Fix team registration

Browse source
This commit is contained in:
James Wang 2016-04-08 10:57:36 -04:00
parent 765f6e0ec1
commit 5b549e0e70
No known key found for this signature in database
GPG key ID: 5B80C0B3F263CD5B
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
server/api/team.py
View file

@ -29,15 +29,14 @@ def team_create():
school = params.get("school") school = params.get("school")
team = Teams(teamname, school, _user.uid, _user.utype != 1) team = Teams(teamname, school, _user.uid, _user.utype != 1)
tid = team.tid
with app.app_context(): with app.app_context():
db.session.add(team) db.session.add(team)
db.session.commit() db.session.commit()
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
db.session.close() db.session.close()
session["tid"] = team.tid
return { "success": 1, "message": "Success!" } return { "success": 1, "message": "Success!" }
@blueprint.route("/delete", methods=["POST"]) @blueprint.route("/delete", methods=["POST"])

2
server/api/user.py
View file

@ -133,7 +133,7 @@ def user_status():
"logged_in": logged_in, "logged_in": logged_in,
"admin": is_admin(), "admin": is_admin(),
"competition": is_admin(), "competition": is_admin(),
"in_team": in_team(get_user()), "in_team": in_team(get_user().first()),
"username": session["username"] if logged_in else "", "username": session["username"] if logged_in else "",
} }
if logged_in: if logged_in: