discover_iostat.py

#!/usr/bin/env python3
"""Author:      Olivier van der Toorn <oliviervdtoorn@gmail.com>
Description:    Discovery script for key values used by the iostat.py script.
"""
import subprocess
import json
import re

STRIP = re.compile(r'\ .*$')


def blockdevices():
    """Enumerates the blockdevices returned by iostat.
    """
    devices = []
    try:
        output = subprocess.getoutput('iostat -N')
        for line in output.split('\n')[6:]:
            line = STRIP.sub('', line)
            if line:
                devices.append(
                    {
                        '{#BLOCK}': line
                    }
                )

    except subprocess.CalledProcessError:  # pragma: no cover
        pass

    return devices


def main():
    """Main function. Enumerates the devices and formats the output for
    use in Zabbix.
    """
    devices = blockdevices()
    data = {'data': devices}
    return json.dumps(data, sort_keys=True, indent=4)


if __name__ == '__main__':
    print(main())  # pragma: no cover