console/SWSCloudAdministrator/Administrator/common.py
2016-10-11 03:37:04 +03:00

17 lines
687 B
Python

from functools import wraps
from flask import redirect, url_for, request, session
from SWSCloudCore.models import Admins
def requires_login(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if 'admin_email' not in session or 'admin_id' not in session or 'admin_password' not in session:
# flash(u'You need to be signed in for this page.')
return redirect(url_for('administrator.logout', next=request.path))
# auth user
if not Admins.auth(session.get('admin_email'), session.get('admin_password')):
return redirect(url_for("administrator.logout"))
return f(*args, **kwargs)
return decorated_function