forked from Github-Mirrors/canaille
143 lines
4 KiB
Python
143 lines
4 KiB
Python
import datetime
|
|
import sys
|
|
|
|
from flask import Flask
|
|
from flask import request
|
|
from flask import session
|
|
from flask_wtf.csrf import CSRFProtect
|
|
|
|
from canaille.app.forms import password_strength_calculator
|
|
|
|
csrf = CSRFProtect()
|
|
|
|
|
|
def setup_sentry(app): # pragma: no cover
|
|
if not app.config["CANAILLE"]["SENTRY_DSN"]:
|
|
return None
|
|
|
|
try:
|
|
import sentry_sdk
|
|
from sentry_sdk.integrations.flask import FlaskIntegration
|
|
|
|
except Exception:
|
|
return None
|
|
|
|
sentry_sdk.init(
|
|
dsn=app.config["CANAILLE"]["SENTRY_DSN"], integrations=[FlaskIntegration()]
|
|
)
|
|
return sentry_sdk
|
|
|
|
|
|
def setup_jinja(app):
|
|
app.jinja_env.filters["len"] = len
|
|
app.jinja_env.filters["password_strength"] = password_strength_calculator
|
|
app.jinja_env.policies["ext.i18n.trimmed"] = True
|
|
|
|
|
|
def setup_blueprints(app):
|
|
import canaille.core.endpoints
|
|
|
|
app.url_map.strict_slashes = False
|
|
|
|
app.register_blueprint(canaille.core.endpoints.bp)
|
|
|
|
if app.features.has_oidc:
|
|
import canaille.oidc.endpoints
|
|
|
|
app.register_blueprint(canaille.oidc.endpoints.bp)
|
|
|
|
if "CANAILLE_SCIM" in app.config and app.config["CANAILLE_SCIM"]["ENABLE_SERVER"]:
|
|
import canaille.scim.endpoints
|
|
|
|
app.register_blueprint(canaille.scim.endpoints.bp)
|
|
|
|
|
|
def setup_flask(app):
|
|
csrf.init_app(app)
|
|
|
|
@app.before_request
|
|
def make_session_permanent():
|
|
session.permanent = True
|
|
app.permanent_session_lifetime = datetime.timedelta(days=365)
|
|
|
|
@app.context_processor
|
|
def global_processor():
|
|
from canaille.app.session import current_user
|
|
|
|
return {
|
|
"debug": app.debug or app.config.get("TESTING", False),
|
|
"logo_url": app.config["CANAILLE"]["LOGO"],
|
|
"favicon_url": app.config["CANAILLE"]["FAVICON"]
|
|
or app.config["CANAILLE"]["LOGO"],
|
|
"website_name": app.config["CANAILLE"]["NAME"],
|
|
"user": current_user(),
|
|
"menu": True,
|
|
"is_boosted": request.headers.get("HX-Boosted", False),
|
|
"features": app.features,
|
|
}
|
|
|
|
|
|
def setup_flask_converters(app):
|
|
from canaille.app import models
|
|
from canaille.app.flask import model_converter
|
|
|
|
for model_name, model_class in models.MODELS.items():
|
|
app.url_map.converters[model_name.lower()] = model_converter(model_class)
|
|
|
|
|
|
def create_app(
|
|
config: dict = None,
|
|
validate: bool = True,
|
|
backend=None,
|
|
env_file: str = None,
|
|
env_prefix: str = "",
|
|
):
|
|
"""Application entry point.
|
|
|
|
:param config: A configuration dict. This will take priority over any other configuration method.
|
|
:param validate: Whether to validate or not the configuration.
|
|
:param backend: An optional backend to force. If unset backend will be initialized according to the configuration.
|
|
:param env_file: The path to an environment var file in which configuration can be loaded.
|
|
:param env_prefix: The prefix to configuration environment vars.
|
|
"""
|
|
from .app.configuration import setup_config
|
|
from .app.features import setup_features
|
|
from .app.i18n import setup_i18n
|
|
from .app.logging import setup_logging
|
|
from .app.themes import setup_themer
|
|
from .backends import setup_backend
|
|
|
|
app = Flask(__name__)
|
|
with app.app_context():
|
|
if not setup_config(
|
|
app=app,
|
|
config=config,
|
|
test_config=validate,
|
|
env_file=env_file,
|
|
env_prefix=env_prefix,
|
|
): # pragma: no cover
|
|
sys.exit(1)
|
|
|
|
sentry_sdk = setup_sentry(app)
|
|
try:
|
|
setup_logging(app)
|
|
backend = setup_backend(app, backend)
|
|
setup_features(app)
|
|
setup_flask_converters(app)
|
|
setup_blueprints(app)
|
|
setup_jinja(app)
|
|
setup_i18n(app)
|
|
setup_themer(app)
|
|
setup_flask(app)
|
|
|
|
if app.features.has_oidc:
|
|
from .oidc.oauth import setup_oauth
|
|
|
|
setup_oauth(app)
|
|
|
|
except Exception as exc: # pragma: no cover
|
|
if sentry_sdk:
|
|
sentry_sdk.capture_exception(exc)
|
|
raise
|
|
|
|
return app
|