haldis/app/login.py
2015-03-26 22:17:50 +01:00

32 lines
716 B
Python

from flask import redirect, abort, session, url_for
from flask.ext.login import LoginManager, current_user, logout_user
from app import app
from models import User
from zeus import zeus_login
login_manager = LoginManager()
login_manager.init_app(app)
@login_manager.user_loader
def load_user(userid):
return User.query.filter_by(id=userid).first()
@app.route('/foodbot/login')
def login():
return zeus_login()
@app.route('/foodbot/logout')
def logout():
if 'zeus_token' in session:
session.pop('zeus_token', None)
logout_user()
return redirect(url_for('admin.index'))
def before_request():
if current_user.is_anonymous() or not current_user.is_allowed():
abort(401)