informatic
/
flask-spaceauth
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make SpaceAuth subclass of LoginManager 

It seems to be the most common object to be accessed from the outside
(eg. custom AnonymousUser or custom unauthorized handlers)
authlib
informatic 2018-04-07 10:38:08 +02:00
parent 80c74deded
commit dd0654d137
1 changed files with 12 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
spaceauth/__init__.py
View File

@ -4,7 +4,7 @@ from flask_login import LoginManager, login_user, logout_user, current_user, log
from spaceauth.caps import cap_required from spaceauth.caps import cap_required
class SpaceAuth(object): class SpaceAuth(LoginManager):
def __init__(self, app=None, *args, **kwargs): def __init__(self, app=None, *args, **kwargs):
self.oauth = OAuth() self.oauth = OAuth()
self.remote = self.oauth.remote_app( self.remote = self.oauth.remote_app(
@ -17,22 +17,22 @@ class SpaceAuth(object):
self.remote.tokengetter(self.tokengetter) self.remote.tokengetter(self.tokengetter)
bp = Blueprint('spaceauth', __name__) bp = Blueprint('spaceauth', __name__)
bp.add_url_rule('/login', 'login', self.login_view) bp.add_url_rule('/login', 'login', self.login_view_handler)
bp.add_url_rule('/logout', 'logout', self.logout_view) bp.add_url_rule('/logout', 'logout', self.logout_view_handler)
bp.add_url_rule('/callback', 'callback', self.callback_view) bp.add_url_rule('/callback', 'callback', self.callback_view_handler)
self.blueprint = bp self.blueprint = bp
self.login_manager = LoginManager() super(SpaceAuth, self).__init__()
self.login_manager.refresh_view = 'spaceauth.login' self.refresh_view = 'spaceauth.login'
self.login_manager.login_view = 'spaceauth.login' self.login_view = 'spaceauth.login'
self.login_manager.user_loader(self.user_loader_handler) self.user_loader(self.user_loader_handler)
if app: if app:
self.init_app(app, *args, **kwargs) self.init_app(app, *args, **kwargs)
def init_app(self, app, url_prefix='/oauth'): def init_app(self, app, url_prefix='/oauth'):
self.oauth.init_app(app) self.oauth.init_app(app)
self.login_manager.init_app(app) super(SpaceAuth, self).init_app(app)
app.register_blueprint(self.blueprint, url_prefix=url_prefix) app.register_blueprint(self.blueprint, url_prefix=url_prefix)
@app.errorhandler(OAuthException) @app.errorhandler(OAuthException)
@ -40,20 +40,20 @@ class SpaceAuth(object):
flash('OAuth error occured', 'error') flash('OAuth error occured', 'error')
return redirect('/') return redirect('/')
def login_view(self): def login_view_handler(self):
session['spaceauth_next'] = request.args.get('next') or request.referrer session['spaceauth_next'] = request.args.get('next') or request.referrer
return self.remote.authorize( return self.remote.authorize(
callback=url_for('spaceauth.callback', _external=True) callback=url_for('spaceauth.callback', _external=True)
) )
def logout_view(self): def logout_view_handler(self):
# TODO revoke token # TODO revoke token
session.pop('spaceauth_token', None) session.pop('spaceauth_token', None)
session.pop('spaceauth_next', None) session.pop('spaceauth_next', None)
logout_user() logout_user()
return redirect('/') return redirect('/')
def callback_view(self): def callback_view_handler(self):
resp = self.remote.authorized_response() resp = self.remote.authorized_response()
if resp is None: if resp is None:
raise OAuthException( raise OAuthException(
@ -79,16 +79,5 @@ class SpaceAuth(object):
user.id = uid user.id = uid
return user return user
def user_loader(self, func):
"""
Define flask_login-like user loader. Application is supposed to create
its user model when missing. Additional `profile` argument is passed
with user profile information right after login.
"""
self.user_loader_handler = func
self.login_manager.user_loader(self.user_loader_handler)
return func
def user_profile(self): def user_profile(self):
return self.remote.get('profile').data return self.remote.get('profile').data