From fd35a269555146d53fa4ee109fe36cb7610d9bff Mon Sep 17 00:00:00 2001
From: Curtis Castrapel <ccastrapel@netflix.com>
Date: Tue, 28 May 2019 12:45:39 -0700
Subject: [PATCH] Support read replicas

---
 lemur/factory.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/lemur/factory.py b/lemur/factory.py
index e0cf5505..0563d873 100644
--- a/lemur/factory.py
+++ b/lemur/factory.py
@@ -19,6 +19,7 @@ from logging import Formatter, StreamHandler
 from logging.handlers import RotatingFileHandler
 
 from flask import Flask
+from flask_replicated import FlaskReplicated
 import logmatic
 
 from lemur.certificates.hooks import activate_debug_dump
@@ -53,6 +54,7 @@ def create_app(app_name=None, blueprints=None, config=None):
     configure_blueprints(app, blueprints)
     configure_extensions(app)
     configure_logging(app)
+    configure_database(app)
     install_plugins(app)
 
     @app.teardown_appcontext
@@ -158,6 +160,11 @@ def configure_blueprints(app, blueprints):
         app.register_blueprint(blueprint, url_prefix="/api/{0}".format(API_VERSION))
 
 
+def configure_database(app):
+    if app.config.get("SQLALCHEMY_ENABLE_FLASK_REPLICATED"):
+        FlaskReplicated(app)
+
+
 def configure_logging(app):
     """
     Sets up application wide logging.