JohnHammond
/
CTFd
mirror of https://github.com/JohnHammond/CTFd.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Extract user/team banning code into its own initialization function

update-jquery-3.5.0
Kevin Chung 2020-04-29 12:46:48 -04:00
parent e15e23f038
commit 7e3b1962c4
1 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
CTFd/utils/initialization/__init__.py
View File

@ -164,6 +164,32 @@ def init_request_processors(app):
else:
return redirect(url_for("views.setup"))
@app.before_request
def banned():
if request.endpoint == "views.themes":
return
if authed():
user = get_current_user()
team = get_current_team()
if user and user.banned:
return (
render_template(
"errors/403.html", error="You have been banned from this CTF"
),
403,
)
if team and team.banned:
return (
render_template(
"errors/403.html",
error="Your team has been banned from this CTF",
),
403,
)
@app.before_request
def tracker():
if request.endpoint == "views.themes":
@ -183,29 +209,6 @@ def init_request_processors(app):
db.session.rollback()
logout_user()
if authed():
user = get_current_user()
team = get_current_team()
if request.path.startswith("/themes") is False:
if user and user.banned:
return (
render_template(
"errors/403.html",
error="You have been banned from this CTF",
),
403,
)
if team and team.banned:
return (
render_template(
"errors/403.html",
error="Your team has been banned from this CTF",
),
403,
)
db.session.close()
@app.before_request