From 1264edac359a357c1daaadd79a4bcb41757ce6de Mon Sep 17 00:00:00 2001
From: Andrew Nemzek <andrewnemzek@invenshure.com>
Date: Tue, 15 Dec 2015 14:49:18 -0600
Subject: [PATCH] Better support mongo replicas and variable retry params

---
 api/config.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/api/config.py b/api/config.py
index 6f5130a2..f2b0ec26 100644
--- a/api/config.py
+++ b/api/config.py
@@ -44,6 +44,8 @@ DEFAULT_CONFIG = {
     },
     'persistent': {
         'db_uri': 'mongodb://localhost:9001/scitran',
+        'db_connect_timeout': 2000,
+        'db_server_selection_timeout': 3000,
         'data_path': os.path.join(os.path.dirname(__file__), '../persistent/data'),
     },
 }
@@ -72,7 +74,12 @@ for outer_key, scoped_config in __config.iteritems():
 if not os.path.exists(__config['persistent']['data_path']):
     os.makedirs(__config['persistent']['data_path'])
 
-db = pymongo.MongoClient(__config['persistent']['db_uri'], j=True, connectTimeoutMS=2000, serverSelectionTimeoutMS=3000).get_default_database()
+db = pymongo.MongoClient(
+    __config['persistent']['db_uri'],
+    j=True,
+    connectTimeoutMS=__config['persistent']['db_connect_timeout'],
+    serverSelectionTimeoutMS=__config['persistent']['db_server_selection_timeout']
+).get_default_database()
 
 
 def initialize_db():
-- 
GitLab