#!/usr/bin/env python3
""" Usage: helm template ... | python thisscript.py """

import sys
import yaml
import subprocess

docs = yaml.load_all(sys.stdin, Loader=yaml.Loader)
# get all objects names of kind Deployment, DaemonSet, StatefulSet
for doc in docs:
    if doc['kind'].lower() in ('deployment', 'daemonset', 'statefulset'):
        # call `kubectl rollout status` for each. It will block util the rollout is finished.
        subprocess.call(['kubectl', 'rollout', 'status', doc['kind'], doc['metadata']['name']])