fallback for Mongodb 2.4 which has no 'replSetGetStatus' admin cmd

Author: Tim Vaillancourt

MongoBackup/DB.py

@@ -36,7 +36,7 @@ def connect(self):
     def auth_if_required(self):
         if self.username is not None and self.password is not None:
             try:
-                logging.debug("Authenticating MongoDB connection")
+                logging.debug("Authenticating connection with username: %s" % self.username)
                 self._conn[self.authdb].authenticate(self.username, self.password)
             except Exception, e:
                 logging.fatal("Unable to authenticate with host %s:%s: %s" % (self.host, self.port, e))
@@ -60,6 +60,14 @@ def admin_command(self, admin_command):
             raise Exception, "Could not get output from command: '%s' after %i retries!" % (admin_command, self.retries), None
         return status
 
+    def server_version(self):
+        status  = self.admin_command('serverStatus')
+        if 'version' in status:
+            version = status['version'].split('-')[0]
+            return tuple(version.split('.'))
+        else:
+            raise Exception, "Could not get server version using admin command 'serverStatus'! Error: %s"  % e, None
+
     def connection(self):
         return self._conn
 

MongoBackup/ReplsetHandler.py

@@ -27,10 +27,20 @@ def close(self):
         return self.connection.close()
 
     def get_rs_status(self):
-        return self.db.admin_command('replSetGetStatus')
+        try:
+            return self.db.admin_command('replSetGetStatus')
+        except Exception, e:
+            raise Exception, "Error getting replica set status! Error: %s" % e, None
 
     def get_rs_config(self):
-        return self.db.admin_command('replSetGetConfig')
+        try:
+            if self.db.server_version() > tuple("2.4.0".split(".")):
+                output = self.db.admin_command('replSetGetConfig')
+                return output['config']
+            else:
+                return self.connection['local'].system.replset.find_one()
+        except Exception, e:
+            raise Exception, "Error getting replica set config! Error: %s" % e, None
 
     def find_desirable_secondary(self):
         rs_status    = self.get_rs_status()
@@ -65,7 +75,7 @@ def find_desirable_secondary(self):
                 log_data    = {}
 
                 hidden_weight = 0.20
-                for member_config in rs_config['config']['members']:
+                for member_config in rs_config['members']:
                     if member_config['host'] == member['name']:
                         if 'hidden' in member_config and member_config['hidden'] == True:
                             score += (score * hidden_weight)