#!/usr/bin/env python2
#
# This should be both valid Python 2 and 3, but currently the collectd
# Python plugin is only Python 2.

import dbus

def get_property(dbus_iface, propname):
    return dbus_iface.proxy_object.Get(dbus_iface.dbus_interface, propname, dbus_interface='org.freedesktop.DBus.Properties')

def init():
    global system_manager
    system_manager = dbus.Interface(dbus.SystemBus().get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1"), "org.freedesktop.systemd1.Manager")

def read():
    global system_manager

    dat = {
        # These are the values of the
        # basic/unit-name.h:UnitActiveStates enum.
        'NActiveUnits': 0,
        'NReloadingUnits': 0,
        'NInactiveUnits': 0,
        'NFailedUnits': 0,
        'NActivatingUnits': 0,
        'NDeactivatingUnits': 0
    }
    for unit in system_manager.ListUnits():
        key = str('N%sUnits' % unit[3].capitalize())
        dat[key] = dat[key] + 1

    for prop in ['NJobs', 'NInstalledJobs', 'NFailedJobs']:
        dat[prop] = int(get_property(system_manager, prop))

    return dat

def dump(dat):
    collectd.info("---- dump() ----")
    for key, val in read().iteritems():
        if __name__ == '__main__':
            print("%s => %s" % (key, val))
        else:
            collectd.info("%s => %s" % (key, val))
            collectd.Values(
                plugin='systemd',
                type='count',
                type_instance=key
            ).dispatch(
                values=[val]
            )


if __name__ == '__main__':
    init()
    dump(read())
else:
    import collectd
    def _config(conf):
        pass
    def _read():
        dump(read())
    collectd.register_config(_config)
    collectd.register_init(init)
    collectd.register_read(_read)