156 lines
4.7 KiB
Python
156 lines
4.7 KiB
Python
import logging
|
|
import typing
|
|
from datetime import datetime
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
|
|
from airbrake import Airbrake, AirbrakeHandler
|
|
from flask import Flask, render_template
|
|
from flask_bootstrap import Bootstrap, StaticCDN
|
|
from flask_debugtoolbar import DebugToolbarExtension
|
|
from flask_login import LoginManager
|
|
from flask_migrate import Migrate, MigrateCommand
|
|
from flask_oauthlib.client import OAuth, OAuthException
|
|
from flask_script import Manager, Server
|
|
|
|
from admin import init_admin
|
|
from login import init_login
|
|
from models import db
|
|
from models.anonymous_user import AnonymouseUser
|
|
from utils import euro_string
|
|
from zeus import init_oauth
|
|
|
|
|
|
def create_app() -> Manager:
|
|
app = Flask(__name__)
|
|
|
|
# Load the config file
|
|
app.config.from_object("config.Configuration")
|
|
|
|
manager = register_plugins(app, debug=app.debug)
|
|
add_handlers(app)
|
|
add_routes(app)
|
|
add_template_filters(app)
|
|
|
|
# TODO do we need to return and then run the manager?
|
|
return manager
|
|
|
|
|
|
def register_plugins(app: Flask, debug: bool) -> Manager:
|
|
# Register Airbrake and enable the logrotation
|
|
if not app.debug:
|
|
timedFileHandler = TimedRotatingFileHandler(
|
|
app.config["LOGFILE"], when="midnight", backupCount=100
|
|
)
|
|
timedFileHandler.setLevel(logging.DEBUG)
|
|
|
|
loglogger = logging.getLogger("werkzeug")
|
|
loglogger.setLevel(logging.DEBUG)
|
|
loglogger.addHandler(timedFileHandler)
|
|
app.logger.addHandler(timedFileHandler)
|
|
|
|
airbrakelogger = logging.getLogger("airbrake")
|
|
|
|
# Airbrake
|
|
airbrake = Airbrake(
|
|
project_id=app.config["AIRBRAKE_ID"], api_key=app.config["AIRBRAKE_KEY"]
|
|
)
|
|
# ugly hack to make this work for out errbit
|
|
airbrake._api_url = "http://errbit.awesomepeople.tv/api/v3/projects/{}/notices".format(
|
|
airbrake.project_id
|
|
)
|
|
|
|
airbrakelogger.addHandler(AirbrakeHandler(airbrake=airbrake))
|
|
app.logger.addHandler(AirbrakeHandler(airbrake=airbrake))
|
|
|
|
# Initialize SQLAlchemy
|
|
db.init_app(app)
|
|
|
|
# Initialize Flask-Migrate
|
|
migrate = Migrate(app, db)
|
|
manager = Manager(app)
|
|
manager.add_command("db", MigrateCommand)
|
|
manager.add_command("runserver", Server(port=8000))
|
|
|
|
# Add admin interface
|
|
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)
|
|
app.config["BOOTSTRAP_SERVE_LOCAL"] = True
|
|
|
|
# use our own bootstrap theme
|
|
app.extensions["bootstrap"]["cdns"]["bootstrap"] = StaticCDN()
|
|
|
|
# Load the flask debug toolbar
|
|
toolbar = DebugToolbarExtension(app)
|
|
|
|
return manager
|
|
|
|
|
|
def add_handlers(app: Flask) -> None:
|
|
@app.errorhandler(404)
|
|
def handle404(e) -> typing.Tuple[str, int]:
|
|
return render_template("errors/404.html"), 404
|
|
|
|
@app.errorhandler(401)
|
|
def handle401(e) -> typing.Tuple[str, int]:
|
|
return render_template("errors/401.html"), 401
|
|
|
|
|
|
def add_routes(application: Flask) -> None:
|
|
# import views # TODO convert to blueprint
|
|
# import views.stats # TODO convert to blueprint
|
|
|
|
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
|
|
|
|
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="/")
|
|
|
|
if application.debug:
|
|
application.register_blueprint(debug_bp, url_prefix="/debug")
|
|
|
|
|
|
def add_template_filters(app: Flask) -> None:
|
|
@app.template_filter("countdown")
|
|
def countdown(value, only_positive: bool = True, show_text: bool = True) -> str:
|
|
delta = value - datetime.now()
|
|
if delta.total_seconds() < 0 and only_positive:
|
|
return "closed"
|
|
hours, remainder = divmod(delta.seconds, 3600)
|
|
minutes, seconds = divmod(remainder, 60)
|
|
time = "%02d:%02d:%02d" % (hours, minutes, seconds)
|
|
if show_text:
|
|
return "closes in " + time
|
|
return time
|
|
|
|
@app.template_filter("year")
|
|
def current_year(value: typing.Any) -> str:
|
|
return str(datetime.now().year)
|
|
|
|
@app.template_filter("euro")
|
|
def euro(value: int) -> None:
|
|
euro_string(value)
|
|
|
|
|
|
# For usage when you directly call the script with python
|
|
if __name__ == "__main__":
|
|
manager = create_app()
|
|
manager.run()
|