archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Allow login_user() to load the user from the database

Browse Source 
This makes it easier to properly log someone in from a unit test.
This commit is contained in:
Sybren A. Stüvel 2017-05-12 13:40:59 +02:00
parent 2703617179
commit fdb9154b85
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pillar/auth/__init__.py
View File

@ -79,10 +79,13 @@ def config_login_manager(app):
return login_manager return login_manager
def login_user(oauth_token): def login_user(oauth_token, *, load_from_db=False):
"""Log in the user identified by the given token.""" """Log in the user identified by the given token."""
user = UserClass(oauth_token) if load_from_db:
user = _load_user(oauth_token)
else:
user = UserClass(oauth_token)
flask_login.login_user(user) flask_login.login_user(user)