Flask Login Example

flask-login-example

from flask import session
from flask import request
from flask import redirect
from functools import wraps

app = Flask(__name__)
app.secret_key = '86A611287CE4DC76545575687F84F'


def login_required(test):
    @wraps(test)
    def wrap(*args, **kwargs):
        if 'username' in session:
            return test(*args, **kwargs)
        else:
            app.logger.info("Unauthorized access attempted!")
            return redirect('login')
    return wrap


@app.route('/admin')
@login_required
def admin():
    return "<h1>Admin Page</h1>"


@app.route('/login', methods=['GET'])
def login():
    return """
    <h1>Login Page!</h1>
    <form method='post' action='/login'>
        <input type='text' name='username' /><br />
        <input type='password' name='password' /><br />
        <input type='submit' />
    </form>
    """


@app.route('/login', methods=['POST'])
def login_submit():
    username = request.form["username"]
    password = request.form["password"]

    if password != "password":
        return "Failed!"
    else:
        print username
        session['username'] = "Benjie"
        return "Success! Logged in as %s" % username


@app.route('/logout')
def logout():
    session.pop('username', None)
    redirect('login')