haldis/app/app.py

190 lines
5.9 KiB
Python
Raw Normal View History

2020-01-24 15:38:03 +01:00
#!/usr/bin/env python3
2019-09-10 15:17:35 +02:00
"Main Haldis script"
2015-04-02 12:41:19 +02:00
import logging
2019-09-10 15:17:35 +02:00
from logging.handlers import TimedRotatingFileHandler
2019-09-08 01:58:21 +02:00
import typing
from datetime import datetime
2020-07-17 10:42:16 +02:00
try:
import airbrake
except ImportError:
airbrake = None
from flask import Flask, render_template
2017-01-06 12:05:31 +01:00
from flask_bootstrap import Bootstrap, StaticCDN
2015-06-09 22:24:48 +02:00
from flask_debugtoolbar import DebugToolbarExtension
from flask_login import LoginManager
from flask_migrate import Migrate, MigrateCommand
from flask_oauthlib.client import OAuth, OAuthException
2019-09-03 13:56:13 +02:00
from flask_script import Manager, Server
from markupsafe import Markup
2015-03-31 20:15:22 +02:00
from admin import init_admin
from login import init_login
from models import db
from models.anonymous_user import AnonymouseUser
2020-02-29 21:56:04 +01:00
from utils import euro_string, price_range_string
from zeus import init_oauth
2019-09-10 15:17:35 +02:00
def register_plugins(app: Flask) -> Manager:
"Register Airbrake and logrotation plugins"
2019-09-10 15:17:35 +02:00
# pylint: disable=W0612
if not app.debug:
2019-09-05 03:33:29 +02:00
timedFileHandler = TimedRotatingFileHandler(
app.config["LOGFILE"], when="midnight", backupCount=100
)
timedFileHandler.setLevel(logging.DEBUG)
2019-09-05 03:33:29 +02:00
loglogger = logging.getLogger("werkzeug")
loglogger.setLevel(logging.DEBUG)
loglogger.addHandler(timedFileHandler)
app.logger.addHandler(timedFileHandler)
2020-07-17 10:42:16 +02:00
if app.config["AIRBRAKE_ID"]:
if airbrake is None:
raise Exception(
"Airbrake support was requested (AIRBRAKE_ID is present in config), "
2020-07-17 11:06:27 +02:00
"but could not import airbrake. Make sure `airbrake` is installed"
2020-07-17 10:42:16 +02:00
)
2020-07-17 10:42:16 +02:00
airbrakelogger = logging.getLogger("airbrake")
2020-07-18 03:38:55 +02:00
airbrake_obj = airbrake.Airbrake(
2020-07-17 11:06:27 +02:00
project_id=app.config["AIRBRAKE_ID"], api_key=app.config["AIRBRAKE_KEY"]
)
# Change URL in a hacky way to make this work for our errbit
2020-07-18 03:38:55 +02:00
airbrake_obj._api_url = "http://errbit.awesomepeople.tv/api/v3/projects/{}/notices".format( # pylint: disable=protected-access
airbrake_obj.project_id
2020-07-17 10:42:16 +02:00
)
2020-07-18 03:38:55 +02:00
airbrakelogger.addHandler(airbrake.AirbrakeHandler(airbrake=airbrake_obj))
app.logger.addHandler(airbrake.AirbrakeHandler(airbrake=airbrake_obj))
# Initialize SQLAlchemy
db.init_app(app)
# Initialize Flask-Migrate
migrate = Migrate(app, db)
2019-09-10 15:17:35 +02:00
app_manager = Manager(app)
app_manager.add_command("db", MigrateCommand)
app_manager.add_command("runserver", Server(port=8000))
init_admin(app, db)
# Init login manager
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.anonymous_user = AnonymouseUser
init_login(app)
# Add oauth
zeus = init_oauth(app)
app.zeus = zeus
# Load the bootstrap local cdn
Bootstrap(app)
2019-09-05 03:33:29 +02:00
app.config["BOOTSTRAP_SERVE_LOCAL"] = True
# use our own bootstrap theme
2019-09-05 03:33:29 +02:00
app.extensions["bootstrap"]["cdns"]["bootstrap"] = StaticCDN()
# Load the flask debug toolbar
toolbar = DebugToolbarExtension(app)
2019-09-11 16:43:42 +02:00
# Make cookies more secure
app.config.update(
2020-07-17 11:40:15 +02:00
SESSION_COOKIE_HTTPONLY=True, SESSION_COOKIE_SAMESITE="Lax",
2019-09-11 16:43:42 +02:00
)
if not app.debug:
app.config.update(SESSION_COOKIE_SECURE=True)
2019-09-10 15:17:35 +02:00
return app_manager
2019-09-08 01:58:21 +02:00
def add_handlers(app: Flask) -> None:
2019-09-10 15:17:35 +02:00
"Add handlers for 4xx error codes"
# pylint: disable=W0612,W0613
@app.errorhandler(404)
2019-09-08 01:58:21 +02:00
def handle404(e) -> typing.Tuple[str, int]:
2019-09-05 03:33:29 +02:00
return render_template("errors/404.html"), 404
2015-06-09 16:47:11 +02:00
@app.errorhandler(401)
2019-09-08 01:58:21 +02:00
def handle401(e) -> typing.Tuple[str, int]:
2019-09-05 03:33:29 +02:00
return render_template("errors/401.html"), 401
2015-03-31 20:15:22 +02:00
2019-09-08 01:58:21 +02:00
def add_routes(application: Flask) -> None:
2019-09-10 15:17:35 +02:00
"Add all routes to Haldis"
# import views # TODO convert to blueprint
# import views.stats # TODO convert to blueprint
2015-03-31 20:15:22 +02:00
from views.order import order_bp
from views.general import general_bp
from views.stats import stats_blueprint
from views.debug import debug_bp
from login import auth_bp
from zeus import oauth_bp
2015-03-31 20:15:22 +02:00
2019-09-05 03:33:29 +02:00
application.register_blueprint(general_bp, url_prefix="/")
application.register_blueprint(order_bp, url_prefix="/order")
application.register_blueprint(stats_blueprint, url_prefix="/stats")
application.register_blueprint(auth_bp, url_prefix="/")
application.register_blueprint(oauth_bp, url_prefix="/")
2015-06-09 22:24:48 +02:00
if application.debug:
2019-09-05 03:33:29 +02:00
application.register_blueprint(debug_bp, url_prefix="/debug")
2015-03-31 16:29:28 +02:00
2019-09-08 01:58:21 +02:00
def add_template_filters(app: Flask) -> None:
2019-09-10 15:17:35 +02:00
"Add functions which can be used in the templates"
# pylint: disable=W0612
2019-09-05 03:33:29 +02:00
@app.template_filter("countdown")
2020-07-17 11:40:15 +02:00
def countdown(
value, only_positive: bool = True, show_text: bool = True, reload: bool = True
) -> str:
delta = int(value.timestamp() - datetime.now().timestamp())
if delta < 0 and only_positive:
text = "closed"
else:
carry, seconds = divmod(delta, 60)
carry, minutes = divmod(carry, 60)
2020-07-17 11:40:15 +02:00
days, hours = divmod(carry, 24)
days_text = f"{days} days, " if days else ""
appendix = " left" if show_text else ""
text = f"{days_text}{hours:02d}:{minutes:02d}:{seconds:02d}{appendix}"
reload_str = "yes" if reload else "no"
2020-07-17 11:40:15 +02:00
return Markup(
f"<span class='time' data-seconds='{delta}' data-reload='{reload_str}'>"
+ text
+ "</span>"
)
2015-06-09 16:47:11 +02:00
2019-09-05 03:33:29 +02:00
@app.template_filter("year")
2020-02-29 21:56:04 +01:00
def current_year(_value: typing.Any) -> str:
return str(datetime.now().year)
2015-06-09 16:47:11 +02:00
2020-02-29 21:56:04 +01:00
app.template_filter("euro")(euro_string)
app.template_filter("price_range")(price_range_string)
2020-01-27 03:01:49 +01:00
app.template_filter("any")(any)
2015-06-09 16:47:11 +02:00
app = Flask(__name__)
# Load the config file
app.config.from_object("config.Configuration")
app_manager = register_plugins(app)
add_handlers(app)
add_routes(app)
add_template_filters(app)
# For usage when you directly call the script with python
2019-09-05 03:33:29 +02:00
if __name__ == "__main__":
app_manager.run()