def create_app():
"""Set up the application."""
flask_app = Flask(__name__.split('.')[0])
local = CONFIG_BROKER['local']
flask_app.config.from_object(__name__)
flask_app.config['LOCAL'] = local
flask_app.debug = CONFIG_SERVICES['debug']
flask_app.env = 'development' if CONFIG_SERVICES['debug'] else 'production'
flask_app.config['SYSTEM_EMAIL'] = CONFIG_BROKER['reply_to_email']
# Make the app not care if there's a trailing slash or not
flask_app.url_map.strict_slashes = False
# Future: Override config w/ environment variable, if set
flask_app.config.from_envvar('BROKER_SETTINGS', silent=True)
# Set parameters
broker_file_path = CONFIG_BROKER['broker_files']
AccountHandler.FRONT_END = CONFIG_BROKER['full_url']
SesEmail.is_local = local
if SesEmail.is_local:
SesEmail.emailLog = os.path.join(broker_file_path, 'email.log')
# If local, make the email directory if needed
if local and not os.path.exists(broker_file_path):
os.makedirs(broker_file_path)
JsonResponse.debugMode = flask_app.debug
if CONFIG_SERVICES['cross_origin_url'] == "*":
CORS(flask_app, supports_credentials=False, allow_headers="*", expose_headers="X-Session-Id")
else:
CORS(flask_app, supports_credentials=False, origins=CONFIG_SERVICES['cross_origin_url'],
allow_headers="*", expose_headers="X-Session-Id")
# Enable DB session table handling
flask_app.session_interface = UserSessionInterface()
# Set up bcrypt
bcrypt = Bcrypt(flask_app)
@flask_app.teardown_appcontext
def teardown_appcontext(exception):
GlobalDB.close()
@flask_app.before_request
def before_request():
# Set global value for local
g.is_local = local
sess = GlobalDB.db().session
# setup user
g.user = None
if session.get('name') is not None:
g.user = sess.query(User).filter_by(user_id=session['name']).one_or_none()
content_type = request.headers.get('Content-Type')
# If the request is a POST we want to log the request body
if request.method == 'POST' and content_type and 'login' not in request.url.lower():
request_body = {}
# If request is json, turn it into a dict
if 'application/json' in content_type:
request_body = json.loads(request.get_data().decode('utf8'))
elif 'multipart/form-data' in content_type:
# If request is a multipart request, get only the form portions of it
for key in request.form.keys():
request_body[key] = request.form[key]
request_dict = {
'message': 'Request body for ' + request.url,
'body': request_body
}
logger.info(request_dict)
# Root will point to index.html
@flask_app.route("/", methods=["GET"])
def root():
return "Broker is running"
@flask_app.errorhandler(ResponseException)
def handle_response_exception(exception):
return JsonResponse.error(exception, exception.status)
@flask_app.errorhandler(Exception)
def handle_exception(exception):
wrapped = ResponseException(str(exception), StatusCode.INTERNAL_ERROR, type(exception))
return JsonResponse.error(wrapped, wrapped.status)
# Add routes for modules here
add_login_routes(flask_app, bcrypt)
add_file_routes(flask_app, local, broker_file_path)
add_generation_routes(flask_app, local, broker_file_path)
add_user_routes(flask_app, flask_app.config['SYSTEM_EMAIL'], bcrypt)
add_dashboard_routes(flask_app)
add_settings_routes(flask_app)
add_domain_routes(flask_app)
add_exception_handlers(flask_app)
return flask_app
def file_app(test_app):
file_routes.add_file_routes(test_app.application, Mock(), Mock())
yield test_app
def create_app():
"""Set up the application."""
flask_app = Flask(__name__.split('.')[0])
local = CONFIG_BROKER['local']
flask_app.config.from_object(__name__)
flask_app.config['LOCAL'] = local
flask_app.debug = CONFIG_SERVICES['debug']
flask_app.env = 'development' if CONFIG_SERVICES['debug'] else 'production'
flask_app.config['SYSTEM_EMAIL'] = CONFIG_BROKER['reply_to_email']
# Future: Override config w/ environment variable, if set
flask_app.config.from_envvar('BROKER_SETTINGS', silent=True)
# Set parameters
broker_file_path = CONFIG_BROKER['broker_files']
AccountHandler.FRONT_END = CONFIG_BROKER['full_url']
SesEmail.is_local = local
if SesEmail.is_local:
SesEmail.emailLog = os.path.join(broker_file_path, 'email.log')
# If local, make the email directory if needed
if local and not os.path.exists(broker_file_path):
os.makedirs(broker_file_path)
JsonResponse.debugMode = flask_app.debug
if CONFIG_SERVICES['cross_origin_url'] == "*":
CORS(flask_app, supports_credentials=False, allow_headers="*", expose_headers="X-Session-Id")
else:
CORS(flask_app, supports_credentials=False, origins=CONFIG_SERVICES['cross_origin_url'],
allow_headers="*", expose_headers="X-Session-Id")
# Enable DB session table handling
flask_app.session_interface = UserSessionInterface()
# Set up bcrypt
bcrypt = Bcrypt(flask_app)
@flask_app.teardown_appcontext
def teardown_appcontext(exception):
GlobalDB.close()
@flask_app.before_request
def before_request():
# Set global value for local
g.is_local = local
sess = GlobalDB.db().session
# setup user
g.user = None
if session.get('name') is not None:
g.user = sess.query(User).filter_by(user_id=session['name']).one_or_none()
# Root will point to index.html
@flask_app.route("/", methods=["GET"])
def root():
return "Broker is running"
@flask_app.errorhandler(ResponseException)
def handle_response_exception(exception):
return JsonResponse.error(exception, exception.status)
@flask_app.errorhandler(Exception)
def handle_exception(exception):
wrapped = ResponseException(str(exception), StatusCode.INTERNAL_ERROR, type(exception))
return JsonResponse.error(wrapped, wrapped.status)
# Add routes for modules here
add_login_routes(flask_app, bcrypt)
add_file_routes(flask_app, local, broker_file_path)
add_generation_routes(flask_app, local, broker_file_path)
add_user_routes(flask_app, flask_app.config['SYSTEM_EMAIL'], bcrypt)
add_domain_routes(flask_app)
add_exception_handlers(flask_app)
return flask_app