|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102 |
- #! /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
|