1#!/usr/bin/python -u 2 3import os 4import sys 5import gobject 6import dbus 7import dbus.service 8import dbus.mainloop.glib 9import cPickle 10import json 11import obmc.dbuslib.propertycacher as PropertyCacher 12from obmc.dbuslib.bindings import get_dbus, DbusProperties, DbusObjectManager 13import obmc_system_config as System 14 15INTF_NAME = 'org.openbmc.InventoryItem' 16DBUS_NAME = 'org.openbmc.Inventory' 17FRUS = System.FRU_INSTANCES 18 19class Inventory(DbusProperties,DbusObjectManager): 20 def __init__(self,bus,name): 21 DbusProperties.__init__(self) 22 DbusObjectManager.__init__(self) 23 dbus.service.Object.__init__(self,bus,name) 24 25 26class InventoryItem(DbusProperties): 27 def __init__(self,bus,name,data): 28 DbusProperties.__init__(self) 29 dbus.service.Object.__init__(self,bus,name) 30 31 self.name = name 32 33 if (data.has_key('present') == False): 34 data['present'] = 'False' 35 if (data.has_key('fault') == False): 36 data['fault'] = 'False' 37 if (data.has_key('version') == False): 38 data['version'] = '' 39 40 self.SetMultiple(INTF_NAME,data) 41 42 ## this will load properties from cache 43 PropertyCacher.load(name, INTF_NAME, self.properties) 44 45 46 @dbus.service.method(INTF_NAME, 47 in_signature='a{sv}', out_signature='') 48 def update(self,data): 49 self.SetMultiple(INTF_NAME,data) 50 PropertyCacher.save(self.name,INTF_NAME,self.properties) 51 52 @dbus.service.method(INTF_NAME, 53 in_signature='s', out_signature='') 54 def setPresent(self,present): 55 self.Set(INTF_NAME,'present',present) 56 57 @dbus.service.method(INTF_NAME, 58 in_signature='s', out_signature='') 59 def setFault(self,fault): 60 self.Set(INTF_NAME,'fault',fault) 61 62 63def getVersion(): 64 version = "Error" 65 with open('/etc/os-release', 'r') as f: 66 for line in f: 67 p = line.rstrip('\n') 68 parts = line.rstrip('\n').split('=') 69 if (parts[0] == "VERSION_ID"): 70 version = parts[1] 71 version = version.strip('"') 72 return version 73 74 75if __name__ == '__main__': 76 dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) 77 bus = get_dbus() 78 mainloop = gobject.MainLoop() 79 obj_parent = Inventory(bus, '/org/openbmc/inventory') 80 81 for f in FRUS.keys(): 82 obj_path=f.replace("<inventory_root>",System.INVENTORY_ROOT) 83 obj = InventoryItem(bus,obj_path,FRUS[f]) 84 obj_parent.add(obj_path,obj) 85 86 ## TODO: this is a hack to update bmc inventory item with version 87 ## should be done by flash object 88 if (FRUS[f]['fru_type'] == "BMC"): 89 version = getVersion() 90 obj.update({'version': version}) 91 92 obj_parent.unmask_signals() 93 name = dbus.service.BusName(DBUS_NAME,bus) 94 print "Running Inventory Manager" 95 mainloop.run() 96 97