mirror of
https://github.com/alerta/alerta.git
synced 2025-01-24 17:29:39 +00:00
122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
from typing import Any, Dict
|
|
|
|
import sentry_sdk
|
|
from flask import Flask
|
|
from flask_compress import Compress
|
|
from flask_cors import CORS
|
|
from sentry_sdk.integrations.flask import FlaskIntegration
|
|
from werkzeug.middleware.proxy_fix import ProxyFix
|
|
|
|
from alerta.database.base import Database, QueryBuilder
|
|
from alerta.exceptions import ExceptionHandlers
|
|
from alerta.models.alarms import AlarmModel
|
|
from alerta.models.enums import Scope
|
|
from alerta.utils.audit import AuditTrail
|
|
from alerta.utils.config import Config
|
|
from alerta.utils.hooks import HookTrigger
|
|
from alerta.utils.key import ApiKeyHelper
|
|
from alerta.utils.logging import Logger
|
|
from alerta.utils.mailer import Mailer
|
|
from alerta.utils.plugin import Plugins
|
|
from alerta.utils.tracing import Tracing
|
|
from alerta.utils.webhook import CustomWebhooks
|
|
from alerta.version import __version__
|
|
|
|
# Sentry will read the DSN from SENTRY_DSN environment variable.
|
|
sentry_sdk.init(integrations=[FlaskIntegration()], release=__version__)
|
|
|
|
config = Config()
|
|
tracing = Tracing()
|
|
logger = Logger()
|
|
hooks = HookTrigger()
|
|
audit = AuditTrail()
|
|
alarm_model = AlarmModel()
|
|
|
|
cors = CORS()
|
|
compress = Compress()
|
|
handlers = ExceptionHandlers()
|
|
key_helper = ApiKeyHelper()
|
|
|
|
db = Database()
|
|
qb = QueryBuilder()
|
|
|
|
mailer = Mailer()
|
|
plugins = Plugins()
|
|
custom_webhooks = CustomWebhooks()
|
|
|
|
|
|
def create_app(config_override: Dict[str, Any] = None, environment: str = None) -> Flask:
|
|
app = Flask(__name__)
|
|
app.config['ENVIRONMENT'] = environment
|
|
config.init_app(app)
|
|
app.config.update(config_override or {})
|
|
|
|
tracing.setup_tracing(app)
|
|
logger.setup_logging(app)
|
|
|
|
if app.config['USE_PROXYFIX']:
|
|
app.wsgi_app = ProxyFix(app.wsgi_app) # type: ignore
|
|
|
|
hooks.init_app(app)
|
|
audit.init_app(app)
|
|
alarm_model.init_app(app)
|
|
Scope.init_app(app)
|
|
|
|
cors.init_app(app)
|
|
compress.init_app(app)
|
|
handlers.register(app)
|
|
key_helper.init_app(app)
|
|
|
|
db.init_db(app)
|
|
qb.init_app(app)
|
|
|
|
mailer.register(app)
|
|
plugins.register(app)
|
|
custom_webhooks.register(app)
|
|
|
|
from alerta.utils.format import CustomJSONEncoder
|
|
app.json_encoder = CustomJSONEncoder
|
|
|
|
from alerta.views import api
|
|
app.register_blueprint(api)
|
|
|
|
from alerta.webhooks import webhooks
|
|
app.register_blueprint(webhooks)
|
|
|
|
from alerta.auth import auth as auth_blueprint
|
|
app.register_blueprint(auth_blueprint)
|
|
|
|
from alerta.management import mgmt
|
|
app.register_blueprint(mgmt)
|
|
|
|
return app
|
|
|
|
|
|
try:
|
|
from celery import Celery
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def create_celery_app(app: Flask = None) -> 'Celery':
|
|
from alerta.utils.format import register_custom_serializer
|
|
register_custom_serializer()
|
|
|
|
app = app or create_app()
|
|
celery = Celery(
|
|
app.name,
|
|
backend=app.config['CELERY_RESULT_BACKEND'],
|
|
broker=app.config['CELERY_BROKER_URL']
|
|
)
|
|
celery.conf.update(app.config)
|
|
TaskBase = celery.Task
|
|
|
|
class ContextTask(TaskBase): # type: ignore
|
|
abstract = True
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
with app.app_context():
|
|
return TaskBase.__call__(self, *args, **kwargs)
|
|
|
|
celery.Task = ContextTask
|
|
return celery
|