#!/usr/bin/env python

import dbus


dbus_objects = {
    'power': {
        'bus_name': 'org.openbmc.control.Power',
        'object_name': '/org/openbmc/control/power0',
        'interface_name': 'org.openbmc.control.Power'
    },
    'occstatus0': {
        'bus_name': 'org.openbmc.Sensors',
        'object_name': '/org/openbmc/sensors/host/cpu0/OccStatus',
        'interface_name': 'org.openbmc.SensorValue'
    },
    'occstatus1': {
        'bus_name': 'org.openbmc.Sensors',
        'object_name': '/org/openbmc/sensors/host/cpu1/OccStatus',
        'interface_name': 'org.openbmc.SensorValue'
    },
    'bootprogress': {
        'bus_name': 'org.openbmc.Sensors',
        'object_name': '/org/openbmc/sensors/host/BootProgress',
        'interface_name': 'org.openbmc.SensorValue'
    },
    'host': {
        'bus_name': 'xyz.openbmc_project.State.Host',
        'object_name': '/xyz/openbmc_project/state/host0',
        'interface_name': 'xyz.openbmc_project.State.Host'
    },
    'settings': {
        'bus_name': 'org.openbmc.settings.Host',
        'object_name': '/org/openbmc/settings/host0',
        'interface_name': 'org.freedesktop.DBus.Properties'
    },
    'system': {
        'bus_name': 'org.openbmc.managers.System',
        'object_name': '/org/openbmc/managers/System',
        'interface_name': 'org.freedesktop.DBus.Properties'
    },
}


def getInterface(bus, objs, key):
    obj = bus.get_object(
        objs[key]['bus_name'], objs[key]['object_name'], introspect=False)
    return dbus.Interface(obj, objs[key]['interface_name'])


def getProperty(bus, objs, key, prop):
    obj = bus.get_object(
        objs[key]['bus_name'], objs[key]['object_name'], introspect=False)
    intf = dbus.Interface(obj, dbus.PROPERTIES_IFACE)
    return intf.Get(objs[key]['interface_name'], prop)

def setProperty(bus, objs, key, prop, prop_value):
    obj = bus.get_object(
        objs[key]['bus_name'], objs[key]['object_name'])
    intf = dbus.Interface(obj, dbus.PROPERTIES_IFACE)
    return intf.Set(objs[key]['interface_name'], prop, prop_value)

bus = dbus.SystemBus()
pgood = getProperty(bus, dbus_objects, 'power', 'pgood')

if (pgood == 1):
    intf = getInterface(bus, dbus_objects, 'bootprogress')
    intf.setValue("FW Progress, Starting OS")
    intf = getInterface(bus, dbus_objects, 'occstatus0')
    intf.setValue("Enabled")
    intf = getInterface(bus, dbus_objects, 'occstatus1')
    intf.setValue("Enabled")
else:
    # Power is off, so check power policy
    settings_intf = getInterface(bus, dbus_objects, 'settings')
    system_intf = getInterface(bus, dbus_objects, 'system')
    system_last_state = system_intf.Get("org.openbmc.managers.System",
                                        "system_last_state")
    power_policy = settings_intf.Get("org.openbmc.settings.Host",
                                     "power_policy")
    print "Last System State:"+system_last_state+"Power Policy:"+power_policy

    if (power_policy == "ALWAYS_POWER_ON" or
       (power_policy == "RESTORE_LAST_STATE" and
            system_last_state == "HOST_POWERED_ON")):
        setProperty(bus, dbus_objects, 'host', 'RequestedHostTransition',
                    'xyz.openbmc_project.State.Host.Transition.On')

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4