Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 1 | #!/usr/bin/env python |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 2 | |
Brad Bishop | 0f608de | 2016-09-08 22:48:22 -0400 | [diff] [blame] | 3 | import os |
| 4 | import sys |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 5 | import gobject |
Brad Bishop | f327ddd | 2016-09-13 22:37:35 -0400 | [diff] [blame] | 6 | import json |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 7 | import dbus |
| 8 | import dbus.service |
| 9 | import dbus.mainloop.glib |
Brad Bishop | ee1b154 | 2016-05-12 16:55:00 -0400 | [diff] [blame] | 10 | import obmc.dbuslib.propertycacher as PropertyCacher |
Brad Bishop | 84e73b5 | 2016-05-12 15:57:52 -0400 | [diff] [blame] | 11 | from obmc.dbuslib.bindings import get_dbus, DbusProperties, DbusObjectManager |
Brad Bishop | 985b2a5 | 2016-08-30 20:06:23 -0400 | [diff] [blame] | 12 | |
| 13 | try: |
| 14 | import obmc_system_config as System |
| 15 | have_system = True |
| 16 | except ImportError: |
| 17 | have_system = False |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 18 | |
| 19 | INTF_NAME = 'org.openbmc.InventoryItem' |
Norman James | 323ed97 | 2015-12-09 09:06:37 -0600 | [diff] [blame] | 20 | DBUS_NAME = 'org.openbmc.Inventory' |
Brad Bishop | 985b2a5 | 2016-08-30 20:06:23 -0400 | [diff] [blame] | 21 | |
| 22 | if have_system: |
| 23 | FRUS = System.FRU_INSTANCES |
| 24 | else: |
| 25 | FRUS = {} |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 26 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 27 | |
| 28 | class Inventory(DbusProperties, DbusObjectManager): |
| 29 | def __init__(self, bus, name): |
Brad Bishop | f47f5fa | 2016-09-08 22:29:01 -0400 | [diff] [blame] | 30 | super(Inventory, self).__init__( |
| 31 | conn=bus, |
| 32 | object_path=name) |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 33 | |
| 34 | |
Brad Bishop | 84e73b5 | 2016-05-12 15:57:52 -0400 | [diff] [blame] | 35 | class InventoryItem(DbusProperties): |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 36 | def __init__(self, bus, name, data): |
Brad Bishop | f47f5fa | 2016-09-08 22:29:01 -0400 | [diff] [blame] | 37 | super(InventoryItem, self).__init__( |
| 38 | conn=bus, |
| 39 | object_path=name) |
Norman James | b714eb2 | 2015-10-26 17:12:57 -0500 | [diff] [blame] | 40 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 41 | self.name = name |
Adriana Kobylak | 9baab4e | 2016-08-03 11:03:45 -0500 | [diff] [blame] | 42 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 43 | if 'present' not in data: |
| 44 | data['present'] = 'False' |
| 45 | if 'fault' not in data: |
| 46 | data['fault'] = 'False' |
| 47 | if 'version' not in data: |
| 48 | data['version'] = '' |
Norman James | 0a5fbf8 | 2015-12-03 17:56:55 -0600 | [diff] [blame] | 49 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 50 | self.SetMultiple(INTF_NAME, data) |
Adriana Kobylak | 9baab4e | 2016-08-03 11:03:45 -0500 | [diff] [blame] | 51 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 52 | ## this will load properties from cache |
| 53 | PropertyCacher.load(name, INTF_NAME, self.properties) |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 54 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 55 | @dbus.service.method( |
| 56 | INTF_NAME, in_signature='a{sv}', out_signature='') |
| 57 | def update(self, data): |
| 58 | self.SetMultiple(INTF_NAME, data) |
| 59 | PropertyCacher.save(self.name, INTF_NAME, self.properties) |
Norman James | 96da5c2 | 2015-10-15 10:17:06 -0500 | [diff] [blame] | 60 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 61 | @dbus.service.method( |
| 62 | INTF_NAME, in_signature='s', out_signature='') |
| 63 | def setPresent(self, present): |
| 64 | self.Set(INTF_NAME, 'present', present) |
| 65 | PropertyCacher.save(self.name, INTF_NAME, self.properties) |
| 66 | |
| 67 | @dbus.service.method( |
| 68 | INTF_NAME, in_signature='s', out_signature='') |
| 69 | def setFault(self, fault): |
| 70 | self.Set(INTF_NAME, 'fault', fault) |
| 71 | PropertyCacher.save(self.name, INTF_NAME, self.properties) |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 72 | |
| 73 | |
Norman James | c9c92dc | 2015-11-17 08:59:07 -0600 | [diff] [blame] | 74 | def getVersion(): |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 75 | version = "Error" |
| 76 | with open('/etc/os-release', 'r') as f: |
| 77 | for line in f: |
| 78 | p = line.rstrip('\n') |
| 79 | parts = line.rstrip('\n').split('=') |
| 80 | if (parts[0] == "VERSION_ID"): |
| 81 | version = parts[1] |
| 82 | version = version.strip('"') |
| 83 | return version |
Norman James | c9c92dc | 2015-11-17 08:59:07 -0600 | [diff] [blame] | 84 | |
| 85 | |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 86 | if __name__ == '__main__': |
| 87 | dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) |
Brad Bishop | 84e73b5 | 2016-05-12 15:57:52 -0400 | [diff] [blame] | 88 | bus = get_dbus() |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 89 | mainloop = gobject.MainLoop() |
Norman James | a3e47c4 | 2015-10-18 14:43:10 -0500 | [diff] [blame] | 90 | obj_parent = Inventory(bus, '/org/openbmc/inventory') |
Matt Spinler | 14c8286 | 2016-08-15 13:31:10 -0500 | [diff] [blame] | 91 | INVENTORY_FILE = os.path.join(sys.prefix, 'share', |
| 92 | 'inventory', 'inventory.json') |
| 93 | |
| 94 | if os.path.exists(INVENTORY_FILE): |
| 95 | with open(INVENTORY_FILE, 'r') as f: |
| 96 | try: |
| 97 | inv = json.load(f) |
| 98 | except ValueError: |
| 99 | print "Invalid JSON detected in " + INVENTORY_FILE |
| 100 | else: |
| 101 | FRUS = inv |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 102 | |
| 103 | for f in FRUS.keys(): |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 104 | obj_path = f.replace("<inventory_root>", System.INVENTORY_ROOT) |
| 105 | obj = InventoryItem(bus, obj_path, FRUS[f]) |
| 106 | obj_parent.add(obj_path, obj) |
Norman James | c9c92dc | 2015-11-17 08:59:07 -0600 | [diff] [blame] | 107 | |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 108 | ## TODO: this is a hack to update bmc inventory item with version |
| 109 | ## should be done by flash object |
| 110 | if (FRUS[f]['fru_type'] == "BMC"): |
| 111 | version = getVersion() |
| 112 | obj.update({'version': version}) |
Norman James | c9c92dc | 2015-11-17 08:59:07 -0600 | [diff] [blame] | 113 | |
Brad Bishop | f0f3efe | 2016-06-29 23:20:24 -0400 | [diff] [blame] | 114 | obj_parent.unmask_signals() |
Brad Bishop | c91b0bf | 2016-08-30 20:03:48 -0400 | [diff] [blame] | 115 | name = dbus.service.BusName(DBUS_NAME, bus) |
Norman James | 19e4591 | 2015-10-04 20:19:41 -0500 | [diff] [blame] | 116 | print "Running Inventory Manager" |
| 117 | mainloop.run() |