Merge pull request #97 from timvaillancourt/MCB_1.0-bugfix-v8

MCB_1.0: Fix connection stacktrace

Author: timvaillancourt

mongodb_consistent_backup/Common/DB.py

@@ -4,7 +4,7 @@
 from pymongo.errors import ConnectionFailure, OperationFailure, ServerSelectionTimeoutError
 from time import sleep
 
-from mongodb_consistent_backup.Errors import DBAuthenticationError, DBConnectionError, DBOperationError, Error
+from mongodb_consistent_backup.Errors import DBAuthenticationError, DBConnectionError, DBOperationError, Error, OperationError
 
 
 class DB:

mongodb_consistent_backup/Main.py

@@ -45,14 +45,17 @@ def __init__(self, prog_name="mongodb-consistent-backup"):
         self.timer                    = Timer(self.manager)
         self.timer_name               = "mongodb_consistent_backup.%s" % self.__class__.__name__
 
-        self.setup_config()
-        self.setup_logger()
-        self.setup_signal_handlers()
-        self.get_lock()
-        self.init()
-        self.set_backup_dirs()
-        self.get_db_conn()
-        self.setup_state()
+        try:
+            self.setup_config()
+            self.setup_logger()
+            self.setup_signal_handlers()
+            self.get_lock()
+            self.init()
+            self.set_backup_dirs()
+            self.get_db_conn()
+            self.setup_state()
+        except OperationError, e:
+            self.exception("Error setting up %s: %s" % (self.program_name, e), e)
 
     def setup_config(self):
         try:
@@ -91,11 +94,14 @@ def setup_state(self):
 
     def get_db_conn(self):
         self.uri = MongoUri(self.config.host, self.config.port)
-        self.db  = DB(self.uri, self.config, True, 'secondaryPreferred')
+        try:
+            self.db = DB(self.uri, self.config, True, 'secondaryPreferred')
+        except OperationError, e:
+            return self.exception("Cannot connect to seed host(s): %s" % self.uri, e)
         self.is_sharded = self.db.is_mongos()
         if not self.is_sharded:
             self.is_sharded = self.db.is_configsvr()
-        if not self.is_sharded and not self.db.is_replset():
+        if not self.is_sharded and not self.db.is_replset() and not self.db.is_configsvr():
             raise OperationError("Host %s is not part of a replset and is not a sharding config/mongos server!" % self.uri.get())
 
     def get_lock(self):