compact juju status for just the 'units' section

juju-st

#!/usr/bin/python3
import json
import subprocess
from prettytable import PrettyTable, PLAIN_COLUMNS

table = PrettyTable()
table.field_names = ['unit', 'workload', 'agent', 'message']
table.set_style(PLAIN_COLUMNS)
table.align['unit'] = table.align['message'] = 'l'

st = subprocess.check_output(['juju', 'status', '--format', 'json'],
                             universal_newlines=True)

st = json.loads(st)

for app_name, app in st['applications'].items():
    if 'units' not in app:
        continue
    for unit_id, unit in app['units'].items():
        table.add_row(["%s%s" % (unit_id, '*' if unit.get('leader') else ''),
                       unit['workload-status']['current'],
                       unit['juju-status']['current'],
                       unit['workload-status'].get('message', '')])
        if 'subordinates' not in unit:
            continue
        for sub_id, subor in unit['subordinates'].items():
            table.add_row(["  %s%s" % (sub_id,
                                       '*' if subor.get('leader') else ''),
                           subor['workload-status']['current'],
                           subor['juju-status']['current'],
                           subor['workload-status'].get('message', '')])

print(table)