Add flask factory to scope app variable

This commit is contained in:
mcbloch 2022-04-19 23:59:23 +02:00
parent e93460743a
commit ae77adc54e
2 changed files with 50 additions and 43 deletions

View file

@ -23,76 +23,77 @@ from utils import euro_string, price_range_string
from zeus import init_oauth from zeus import init_oauth
def register_plugins(_app: Flask) -> Manager: def register_plugins(app: Flask) -> Manager:
"Register the plugins to the app" """Register the plugins to the app"""
# pylint: disable=W0612 # pylint: disable=W0612
if not _app.debug: if not app.debug:
timedFileHandler = TimedRotatingFileHandler( timedFileHandler = TimedRotatingFileHandler(
_app.config["LOGFILE"], when="midnight", backupCount=100 app.config["LOGFILE"], when="midnight", backupCount=100
) )
timedFileHandler.setLevel(logging.DEBUG) timedFileHandler.setLevel(logging.DEBUG)
loglogger = logging.getLogger("werkzeug") loglogger = logging.getLogger("werkzeug")
loglogger.setLevel(logging.DEBUG) loglogger.setLevel(logging.DEBUG)
loglogger.addHandler(timedFileHandler) loglogger.addHandler(timedFileHandler)
_app.logger.addHandler(timedFileHandler) app.logger.addHandler(timedFileHandler)
# Initialize SQLAlchemy # Initialize SQLAlchemy
db.init_app(_app) db.init_app(app)
# Initialize Flask-Migrate # Initialize Flask-Migrate
migrate = Migrate(_app, db) migrate = Migrate(app, db)
_app_manager = Manager(_app) app_manager = Manager(app)
_app_manager.add_command("db", MigrateCommand) app_manager.add_command("db", MigrateCommand)
_app_manager.add_command("runserver", Server(port=8000)) app_manager.add_command("runserver", Server(port=8000))
init_admin(_app, db) init_admin(app, db)
# Init login manager # Init login manager
login_manager = LoginManager() login_manager = LoginManager()
login_manager.init_app(_app) login_manager.init_app(app)
login_manager.anonymous_user = AnonymouseUser login_manager.anonymous_user = AnonymouseUser
init_login(_app) init_login(app)
# Add oauth # Add oauth
zeus = init_oauth(_app) zeus = init_oauth(app)
_app.zeus = zeus app.zeus = zeus
# Load the bootstrap local cdn # Load the bootstrap local cdn
Bootstrap(_app) Bootstrap(app)
_app.config["BOOTSTRAP_SERVE_LOCAL"] = True app.config["BOOTSTRAP_SERVE_LOCAL"] = True
# use our own bootstrap theme # use our own bootstrap theme
_app.extensions["bootstrap"]["cdns"]["bootstrap"] = StaticCDN() app.extensions["bootstrap"]["cdns"]["bootstrap"] = StaticCDN()
# Load the flask debug toolbar # Load the flask debug toolbar
toolbar = DebugToolbarExtension(_app) toolbar = DebugToolbarExtension(app)
# Make cookies more secure # Make cookies more secure
_app.config.update( app.config.update(
SESSION_COOKIE_HTTPONLY=True, SESSION_COOKIE_HTTPONLY=True,
SESSION_COOKIE_SAMESITE="Lax", SESSION_COOKIE_SAMESITE="Lax",
) )
if not _app.debug: if not app.debug:
_app.config.update(SESSION_COOKIE_SECURE=True) app.config.update(SESSION_COOKIE_SECURE=True)
return _app_manager return app_manager
def add_handlers(_app: Flask) -> None: def add_handlers(app: Flask) -> None:
"Add handlers for 4xx error codes" """Add handlers for 4xx error codes"""
# pylint: disable=W0612,W0613 # pylint: disable=W0612,W0613
@_app.errorhandler(404) @app.errorhandler(404)
def handle404(e) -> typing.Tuple[str, int]: def handle404(e) -> typing.Tuple[str, int]:
return render_template("errors/404.html"), 404 return render_template("errors/404.html"), 404
@_app.errorhandler(401) @app.errorhandler(401)
def handle401(e) -> typing.Tuple[str, int]: def handle401(e) -> typing.Tuple[str, int]:
return render_template("errors/401.html"), 401 return render_template("errors/401.html"), 401
def add_routes(application: Flask) -> None: def add_routes(application: Flask) -> None:
"Add all routes to Haldis" """Add all routes to Haldis"""
# import views # TODO convert to blueprint # import views # TODO convert to blueprint
# import views.stats # TODO convert to blueprint # import views.stats # TODO convert to blueprint
@ -113,8 +114,9 @@ def add_routes(application: Flask) -> None:
application.register_blueprint(debug_bp, url_prefix="/debug") application.register_blueprint(debug_bp, url_prefix="/debug")
def add_template_filters(_app: Flask) -> None: def add_template_filters(app: Flask) -> None:
"Add functions which can be used in the templates" """Add functions which can be used in the templates"""
# pylint: disable=W0612 # pylint: disable=W0612
@app.template_filter("countdown") @app.template_filter("countdown")
def countdown( def countdown(
@ -145,12 +147,14 @@ def add_template_filters(_app: Flask) -> None:
def current_year(_value: typing.Any) -> str: def current_year(_value: typing.Any) -> str:
return str(datetime.now().year) return str(datetime.now().year)
_app.template_filter("euro")(euro_string) app.template_filter("euro")(euro_string)
_app.template_filter("price_range")(price_range_string) app.template_filter("price_range")(price_range_string)
_app.template_filter("any")(any) app.template_filter("any")(any)
_app.template_filter("all")(all) app.template_filter("all")(all)
def create_app():
"""Initializer for the Flask app object"""
app = Flask(__name__) app = Flask(__name__)
# Load the config file # Load the config file
@ -161,7 +165,10 @@ add_handlers(app)
add_routes(app) add_routes(app)
add_template_filters(app) add_template_filters(app)
return app_manager
# For usage when you directly call the script with python # For usage when you directly call the script with python
if __name__ == "__main__": if __name__ == "__main__":
app_manager.run() app_mgr = create_app()
app_mgr.run()

View file

@ -77,7 +77,7 @@ class AnonOrderItemForm(OrderItemForm):
self.user_name.data = session.get("anon_name", None) self.user_name.data = session.get("anon_name", None)
def validate(self) -> bool: def validate(self) -> bool:
"Check if the provided anon_name is not already taken" """Check if the provided anon_name is not already taken"""
rv = OrderForm.validate(self) rv = OrderForm.validate(self)
if not rv: if not rv:
return False return False