console/SWSCloudWeb/common.py

from functools import wraps
from flask import redirect, url_for, session
from SWSCloudCore.models import Users


def requires_login(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        # Check session
        required = ['email', 'password', 'user_id']
        success = True

        for r in required:
            if r in session and success:
                continue
            else:
                return redirect(url_for("account.logout"))

        if not Users.auth(session.get('email'), session.get('password').encode(), 1):
            return redirect(url_for("account.logout"))

        return f(*args, **kwargs)
    return decorated_function