mirror of https://github.com/JohnHammond/CTFd.git
Invalidate any previously cached attributes on user login
parent
7779175eea
commit
625c17db30
|
@ -3,6 +3,7 @@ import os
|
||||||
|
|
||||||
from flask import session
|
from flask import session
|
||||||
|
|
||||||
|
from CTFd.cache import clear_user_session
|
||||||
from CTFd.exceptions import UserNotFoundException, UserTokenExpiredException
|
from CTFd.exceptions import UserNotFoundException, UserTokenExpiredException
|
||||||
from CTFd.models import UserTokens, db
|
from CTFd.models import UserTokens, db
|
||||||
from CTFd.utils.encoding import hexencode
|
from CTFd.utils.encoding import hexencode
|
||||||
|
@ -15,6 +16,9 @@ def login_user(user):
|
||||||
session["email"] = user.email
|
session["email"] = user.email
|
||||||
session["nonce"] = generate_nonce()
|
session["nonce"] = generate_nonce()
|
||||||
|
|
||||||
|
# Clear out any currently cached user attributes
|
||||||
|
clear_user_session(user_id=user.id)
|
||||||
|
|
||||||
|
|
||||||
def logout_user():
|
def logout_user():
|
||||||
session.clear()
|
session.clear()
|
||||||
|
|
|
@ -28,18 +28,6 @@ def test_clear_user_session():
|
||||||
user.type = "admin"
|
user.type = "admin"
|
||||||
app.db.session.commit()
|
app.db.session.commit()
|
||||||
|
|
||||||
# The user shouldn't be considered admin because their type is still cached
|
|
||||||
user = Users.query.filter_by(id=2).first()
|
|
||||||
with app.test_request_context("/"):
|
|
||||||
login_user(user)
|
|
||||||
user = get_current_user()
|
|
||||||
assert user.id == 2
|
|
||||||
assert user.type == "admin"
|
|
||||||
assert is_admin() is False
|
|
||||||
|
|
||||||
# Clear the user's cached session (for now just the type)
|
|
||||||
clear_user_session(user_id=2)
|
|
||||||
|
|
||||||
# The user's type should now be admin
|
# The user's type should now be admin
|
||||||
user = Users.query.filter_by(id=2).first()
|
user = Users.query.filter_by(id=2).first()
|
||||||
with app.test_request_context("/"):
|
with app.test_request_context("/"):
|
||||||
|
|
Loading…
Reference in New Issue