diff --git a/api/config.py b/api/config.py index 6f5130a2f62638f59a42f13de808ae843b04fb1d..116349c780e0ae6b5cea92fdb8215c86adb48a12 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': 5000, + '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():