Elasticsearch Rolling Upgrade with Python -- 2

I have a script that almost completes the task of doing a rolling upgrade with Python. There are few more tasks that needs to be done in the ticket which will involve understading Python, Elasticsearch(intermediate knowledge), ansible and terraform.
If there are changes there will be back and forth communications and improvisations can be made accordingly.