def init_app(self, app=None):
project_id = app.config.get('AIRBRAKE_PROJECT_ID', None)
api_key = app.config.get('AIRBRAKE_API_KEY', None)
host = app.config.get('AIRBRAKE_HOST', None)
environment = app.config.get('AIRBRAKE_ENVIRONMENT', None)
if project_id and api_key:
self.ab = Airbrake(project_id,
api_key,
host,
environment=environment)
got_request_exception.connect(self.process_exception, sender=app)
def __init__(self, project_id=None, api_key=None, environment="dev",
notifier=None, use_ssl=True, airbrake=None):
logging.Handler.__init__(self)
if isinstance(airbrake, Airbrake):
airbrake.auto_notify = True
self.airbrake = airbrake
else:
self.airbrake = Airbrake(project_id=project_id, api_key=api_key,
environment=environment, notifier=notifier,
use_ssl=use_ssl, auto_notify=True)
def __init__(self, *args, **kwargs):
super(AirbrakeNotifierMiddleware, self).__init__(*args, **kwargs)
self.enabled = False
if hasattr(settings, 'AIRBRAKE') \
and not settings.AIRBRAKE.get('DISABLE', False):
self.enabled = True
project_id = settings.AIRBRAKE.get('PROJECT_ID', None)
api_key = settings.AIRBRAKE.get('API_KEY', None)
host = settings.AIRBRAKE.get('HOST', None)
timeout = settings.AIRBRAKE.get('TIMEOUT', None)
environment = settings.AIRBRAKE.get('ENVIRONMENT', None)
self.ab = Airbrake(project_id, api_key, host, timeout,
environment=environment)
def __init__(self, settings):
if not settings.get('AIRBRAKE', False) or \
settings['AIRBRAKE'].get('DISABLE', False):
return
self.enabled = True
project_id = settings['AIRBRAKE'].get('PROJECT_ID', None)
api_key = settings['AIRBRAKE'].get('API_KEY', None)
host = settings['AIRBRAKE'].get('HOST', None)
timeout = settings['AIRBRAKE'].get('TIMEOUT', None)
environment = settings['AIRBRAKE'].get('ENVIRONMENT', None)
self.ab = Airbrake(project_id,
api_key,
host,
timeout,
environment=environment,
send_uncaught_exc=False)
def __init__(self,
airbrake=None,
level=logging.ERROR,
project_id=None,
api_key=None,
environment=None):
"""Initialize the Airbrake handler with a default logging level.
Default level of logs handled by this class are >= ERROR,
which includes ERROR and CRITICAL. To change this behavior
supply a different argument for 'level'.
"""
logging.Handler.__init__(self, level=level)
if isinstance(airbrake, Airbrake):
self.airbrake = airbrake
else:
self.airbrake = Airbrake(project_id, api_key, environment)
Bootstrap(app)
# use our own bootstrap theme
app.extensions['bootstrap']['cdns']['bootstrap'] = StaticCDN()
db = SQLAlchemy(app)
toolbar = DebugToolbarExtension(app)
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))
def register_plugins(app: Flask) -> Manager:
"Register all the plugins to Haldis"
# pylint: disable=W0612
# 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( # pylint: disable=W0212
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)
app_manager = Manager(app)
app_manager.add_command("db", MigrateCommand)
app_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)
# Make cookies more secure
app.config.update(
SESSION_COOKIE_HTTPONLY=True,
SESSION_COOKIE_SAMESITE='Lax',
)
if not app.debug:
app.config.update(SESSION_COOKIE_SECURE=True)
return app_manager
lambda text: jinja2.Markup(markdown.markdown(text)))
app.jinja_env.tests['not_equalto'] = lambda value, other: value != other
lm = LoginManager(app)
lm.init_app(app)
csrf = CsrfProtect()
csrf.init_app(app)
if config.PRODUCTION:
log_file_handler = RotatingFileHandler(config.WEB_LOG, maxBytes=10000000)
logging.getLogger().addHandler(log_file_handler)
logging.getLogger('werkzeug').addHandler(log_file_handler)
app.logger.addHandler(log_file_handler)
airbrakelogger = logging.getLogger('airbrake')
# Airbrake
airbrake = Airbrake(project_id=config.AIRBRAKE_ID,
api_key=config.AIRBRAKE_KEY)
# ugly hack to make this work for our errbit
airbrake._api_url = "{}/api/v3/projects/{}/notices".format(
config.AIRBRAKE_BASE_URL, airbrake.project_id)
airbrakelogger.addHandler(AirbrakeHandler(airbrake=airbrake))
app.logger.addHandler(AirbrakeHandler(airbrake=airbrake))
app.before_request(lambda: Session.remove())
from battlebots.web import views # NOQA
from battlebots.web.views import bots # NOQA
