Fabric MongoDB Rolling Upgrade Task

fabric_mongodb_rolling_upgrade_.py

@task
def rolling_upgrade(hostname):
        """ Rolling upgrade of mongodb """
        users.root()
        conn = Connection('%s'%(hostname))
        db = conn.admin
        replSet = db.command("replSetGetStatus")
        for secondary in replSet['members']:
                stateStr = str(secondary['stateStr'])
                host = str(secondary['name'])
                health = str(secondary['health'])
                if stateStr == 'SECONDARY':
                        execute(puppet.puppetd_test, host='%s' %(host))
                        execute(mongod.restart, host='%s' %(host))
                        while health != '1.0':
                                time.sleep(10)
                                print health
        for primary in replSet['members']:
                stateStr = str(primary['stateStr'])
                host = str(primary['name'])
                health = str(primary['health'])
                if stateStr == 'PRIMARY':
                        execute(mongod.stop, host='%s' %(host))
                        execute(puppet.puppetd_test, host='%s' %(host))
                        execute(mongod.start, host='%s' %(host))