calefaction/app.py

#! /usr/bin/env python3
# -*- coding: utf-8  -*-

from pathlib import Path

from flask import Flask, abort, flash, g, redirect, request, url_for
from flask_mako import MakoTemplates, render_template

import calefaction
from calefaction.auth import AuthManager
from calefaction.config import Config
from calefaction.database import Database
from calefaction.eve import EVE
from calefaction.messages import Messages
from calefaction.util import (
    try_func, make_error_catcher, make_route_restricter,
    set_up_asset_versioning)

app = Flask(__name__)

basepath = Path(__file__).resolve().parent
config = Config(basepath)
Database.path = str(basepath / "data" / "db.sqlite3")
eve = EVE(config)
auth = AuthManager(config, eve)

app.catch_exceptions = make_error_catcher(app, "error.mako")
app.route_restricted = make_route_restricter(
    auth, lambda: redirect(url_for("index"), 303))

MakoTemplates(app)
set_up_asset_versioning(app)
calefaction.enable_logging()
config.install(app)

@app.before_request
def prepare_request():
    """Set up the Flask global context variable with important objects."""
    g.auth = auth
    g.config = config
    g.eve = eve
    g.modules = config.modules
    g.version = calefaction.__version__

app.before_request(Database.pre_hook)
app.teardown_appcontext(Database.post_hook)

@app.route("/")
@app.catch_exceptions
def index():
    """Render and return the index page.

    This is a informational landing page for non-logged-in users, and the corp
    homepage for those who are logged in.
    """
    success, _ = try_func(auth.is_authenticated)
    if success:
        module = config.get("modules.home")
        if module:
            return config.modules[module].home()
        return render_template("default_home.mako")
    return render_template("landing.mako")

@app.route("/login", methods=["GET", "POST"])
@app.catch_exceptions
def login():
    """Handle the last step of a SSO login request."""
    code = request.values.get("code")
    state = request.values.get("state")

    success, caught = try_func(lambda: auth.handle_login(code, state))
    if success:
        flash(Messages.LOGGED_IN, "success")
    elif not caught:
        flash(Messages.LOGIN_FAILED, "error")
    return redirect(url_for("index"), 303)

@app.route("/logout", methods=["GET", "POST"])
@app.catch_exceptions
def logout():
    """Log the user out (POST), or ask them to confirm a log out (GET)."""
    if request.method == "GET":
        return render_template("logout.mako")

    auth.handle_logout()
    flash(Messages.LOGGED_OUT, "success")
    return redirect(url_for("index"), 303)

@app.route("/settings/style/<style>", methods=["POST"])
@app.catch_exceptions
@app.route_restricted
def set_style(style):
    """Set the user's style preference."""
    if not auth.set_character_style(style):
        abort(404)
    return "", 204

@app.errorhandler(404)
@app.catch_exceptions
def page_not_found(err):
    """Render and return the 404 error template."""
    return render_template("404.mako"), 404