haldis/app/zeus.py

70 lines
2.1 KiB
Python
Raw Normal View History

from flask import (current_app, flash, redirect, request, session,
url_for, Blueprint)
2017-01-06 12:05:31 +01:00
from flask_login import login_user
from flask_oauthlib.client import OAuthException, OAuth
2015-03-31 20:15:22 +02:00
from models import User, db
2015-03-31 20:15:22 +02:00
oauth_bp = Blueprint("oauth_bp", __name__)
2015-03-31 20:15:22 +02:00
def zeus_login():
return current_app.zeus.authorize(
callback=url_for('oauth_bp.authorized', _external=True))
2015-03-31 20:15:22 +02:00
@oauth_bp.route('/login/zeus/authorized')
2015-03-31 20:15:22 +02:00
def authorized():
resp = current_app.zeus.authorized_response()
2015-03-31 20:15:22 +02:00
if resp is None:
return 'Access denied: reason=%s error=%s' % (
request.args['error'], request.args['error_description'])
2015-03-31 20:15:22 +02:00
if isinstance(resp, OAuthException):
return 'Access denied: %s' % resp.message + '<br>' + str(resp.data)
session['zeus_token'] = (resp['access_token'], '')
me = current_app.zeus.get('current_user/')
2015-03-31 20:15:22 +02:00
username = me.data.get('username', '').lower()
user = User.query.filter_by(username=username).first()
if len(username) > 0 and user:
return login_and_redirect_user(user)
elif len(username) > 0:
user = create_user(username)
return login_and_redirect_user(user)
flash("You're not allowed to enter, please contact a system administrator")
return redirect(url_for("general_bp.home"))
def init_oauth(app):
oauth = OAuth(app)
zeus = oauth.remote_app(
'zeus',
consumer_key=app.config['ZEUS_KEY'],
consumer_secret=app.config['ZEUS_SECRET'],
request_token_params={},
base_url='https://adams.ugent.be/oauth/api/',
access_token_method='POST',
access_token_url='https://adams.ugent.be/oauth/oauth2/token/',
authorize_url='https://adams.ugent.be/oauth/oauth2/authorize/')
2015-03-31 20:15:22 +02:00
@zeus.tokengetter
def get_zeus_oauth_token():
return session.get('zeus_token')
2015-03-31 20:15:22 +02:00
return zeus
2015-03-31 20:15:22 +02:00
def login_and_redirect_user(user):
login_user(user)
return redirect(url_for("general_bp.home"))
2015-03-31 20:15:22 +02:00
def create_user(username):
user = User()
user.configure(username, False, 1)
db.session.add(user)
db.session.commit()
return user