from flask import Flask, redirect, abort, request, session, render_template
from jinja2 import Environment, FileSystemLoader

env = Environment(loader=FileSystemLoader('templates'))
app = Flask(__name__)
app.secret_key = "secret key"


@app.route('/')
def hi():
    return 'hi'

@app.route('/hello', methods=['GET', 'POST'])
def hello():
    if 'log' in session:
        if session.get('log') == True:
        	template = env.get_template('index.html')
       		return template.render(user=session['username'])
        return redirect('/login')
    return redirect('/login')



@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.authorization and request.authorization.username == 'TRAIN' and request.authorization.password == 'TuN3L':
        session['log'] = True
        session['username'] = request.authorization.username
        return redirect('/hello')
    else:
        return abort(401)

@app.route('/logout',methods=['GET', 'POST'])
def logout():
    if 'log' in session:
        if session.get('log') == True:
            session['log'] = False
            return redirect('/')
    return redirect('/login')

#
if __name__ == '__main__':
    app.run(debug=False)