diff --git a/server.py b/server.py index f70d4d0d..d0771d5d 100644 --- a/server.py +++ b/server.py @@ -6,8 +6,8 @@ import sentry_sdk from flask import Flask, redirect, url_for, render_template, request, jsonify from flask_admin import Admin from flask_cors import cross_origin -from flask_debugtoolbar import DebugToolbarExtension from flask_login import current_user +import flask_profiler from sentry_sdk.integrations.flask import FlaskIntegration from app import paddle_utils @@ -22,6 +22,8 @@ from app.config import ( SHA1, PADDLE_MONTHLY_PRODUCT_ID, RESET_DB, + FLASK_PROFILER_PATH, + FLASK_PROFILER_PASSWORD, ) from app.dashboard.base import dashboard_bp from app.developer.base import developer_bp @@ -77,6 +79,20 @@ def create_app() -> Flask: init_admin(app) setup_paddle_callback(app) + if FLASK_PROFILER_PATH: + LOG.d("Enable flask-profiler") + app.config["flask_profiler"] = { + "enabled": True, + "storage": {"engine": "sqlite", "FILE": FLASK_PROFILER_PATH}, + "basicAuth": { + "enabled": True, + "username": "admin", + "password": FLASK_PROFILER_PASSWORD, + }, + "ignore": ["^/static/.*"], + } + flask_profiler.init_app(app) + return app