diff options
Diffstat (limited to 'accounts/utils/login.py')
-rw-r--r-- | accounts/utils/login.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/accounts/utils/login.py b/accounts/utils/login.py index bd661b2..0cd1dc4 100644 --- a/accounts/utils/login.py +++ b/accounts/utils/login.py @@ -1,10 +1,11 @@ # -*- coding: utf-8 -*- -from flask import current_app from flask_login import LoginManager, current_user from functools import wraps from werkzeug.exceptions import Forbidden from itsdangerous import base64_decode, base64_encode import json +import flask_login.login_manager +from accounts.app import accounts_app class _compact_json: @@ -19,29 +20,29 @@ class _compact_json: return json.dumps(obj, **kwargs) -def create_login_manager(): +def create_login_manager() -> flask_login.login_manager.LoginManager: login_manager = LoginManager() login_manager.login_message = 'Bitte einloggen' login_manager.login_view = 'login.login' @login_manager.user_loader - def load_user(user_id): + def load_user(user_id: str): try: username, password = parse_userid(user_id) - return current_app.user_backend.auth(username, password) - except (current_app.user_backend.NoSuchUserError, - current_app.user_backend.InvalidPasswordError): + return accounts_app.user_backend.auth(username, password) + except (accounts_app.user_backend.NoSuchUserError, + accounts_app.user_backend.InvalidPasswordError): return None return login_manager -def create_userid(username, password): +def create_userid(username: str, password: str): userid = (username, password) return base64_encode(_compact_json.dumps(userid)) -def parse_userid(value): +def parse_userid(value: str): return _compact_json.loads(base64_decode(value)) |