import os import re import dateutil from flask import Flask, session, Markup, escape, render_template from flask._compat import string_types from flask.ext.sqlalchemy import SQLAlchemy from flask_login import LoginManager from flask_wtf import CsrfProtect from werkzeug.exceptions import abort from app.notify_client.api_client import NotificationsAdminAPIClient from app.notify_client.api_key_api_client import ApiKeyApiClient from app.notify_client.user_api_client import UserApiClient from app.its_dangerous_session import ItsdangerousSessionInterface import app.proxy_fix from config import configs from utils import logging login_manager = LoginManager() csrf = CsrfProtect() notifications_api_client = NotificationsAdminAPIClient() user_api_client = UserApiClient() api_key_api_client = ApiKeyApiClient() def create_app(config_name, config_overrides=None): application = Flask(__name__) application.config['NOTIFY_ADMIN_ENVIRONMENT'] = config_name application.config.from_object(configs[config_name]) init_app(application, config_overrides) logging.init_app(application) init_csrf(application) notifications_api_client.init_app(application) user_api_client.init_app(application) api_key_api_client.init_app(application) login_manager.init_app(application) login_manager.login_view = 'main.sign_in' from app.main import main as main_blueprint application.register_blueprint(main_blueprint) from .status import status as status_blueprint application.register_blueprint(status_blueprint) proxy_fix.init_app(application) application.session_interface = ItsdangerousSessionInterface() application.add_template_filter(placeholders) application.add_template_filter(replace_placeholders) application.add_template_filter(nl2br) application.add_template_filter(format_datetime) application.after_request(useful_headers_after_request) register_errorhandlers(application) return application def init_csrf(application): csrf.init_app(application) @csrf.error_handler def csrf_handler(reason): if 'user_id' not in session: application.logger.info( u'csrf.session_expired: Redirecting user to log in page' ) return application.login_manager.unauthorized() application.logger.info( u'csrf.invalid_token: Aborting request, user_id: {user_id}', extra={'user_id': session['user_id']}) abort(400, reason) def init_app(app, config_overrides): if config_overrides: for key in app.config.keys(): if key in config_overrides: app.config[key] = config_overrides[key] for key, value in app.config.items(): if key in os.environ: app.config[key] = convert_to_boolean(os.environ[key]) @app.context_processor def inject_global_template_variables(): return {'asset_path': '/static/'} def convert_to_boolean(value): if isinstance(value, string_types): if value.lower() in ['t', 'true', 'on', 'yes', '1']: return True elif value.lower() in ['f', 'false', 'off', 'no', '0']: return False return value def placeholders(value): if not value: return value return Markup(re.sub( r"\(\(([^\)]+)\)\)", # anything that looks like ((registration number)) lambda match: "{}".format(match.group(1)), value )) def nl2br(value): _paragraph_re = re.compile(r'(?:\r\n|\r|\n){2,}') result = u'\n\n'.join(u'
%s
' % p.replace('\n', Markup('