import os
 
# setup global and environ variables
app_root = os.path.dirname(os.path.abspath(__name__))
os.environ['FAMILY_BUDGET_ROOT'] = app_root
 
from bottle import route, run, redirect, request, get, post, static_file
from controller import check_login, get_page
 
 
# static section
 
@route('/<filename:re:.*\.css>')
def stylesheets(filename):
    return static_file(filename, root=app_root)
 
# dynamic section
 
@route('<path:path>')
def family_budget(path):
    redirect('/family_budget/login')
 
@get('/family_budget/login')
def login():
    username = request.get_cookie("account", secret='very_secret_key')
    if username:
        redirect('/family_budget/main_page')
    else:
        login_page = get_page('templates/login_page.html')
        return login_page
 
@post('/family_budget/login')
def do_login():
    username = request.forms.get('username')
    password = request.forms.get('password')
    if check_login(username, password):
        request.set_cookie("account", username, secret='very_secret_key')
        redirect('/family_budget/main_page')
    else:
        return "<p>Login failed.</p>"
 
@route('/restricted')
def restricted_area():
    username = request.get_cookie("account", secret='very_secret_key')
    if username:
        return template("Hello {{name}}. Welcome back.", name=username)
    else:
        return "You are not logged in. Access denied."
 
run(host='0.0.0.0', port=5050)