1ae7c820fSGeorge Keishing#!/usr/bin/env python 2ae7c820fSGeorge Keishingr""" 3ae7c820fSGeorge KeishingGenerate an inventory variable file containing a list of properties 4ae7c820fSGeorge Keishingfields from the YAML phosphor-dbus-interfaces repository. 5ae7c820fSGeorge Keishing""" 6ae7c820fSGeorge Keishingimport sys 7ae7c820fSGeorge Keishingimport os 8ae7c820fSGeorge Keishingimport yaml 9ae7c820fSGeorge Keishingimport json 10ae7c820fSGeorge Keishing 11ae7c820fSGeorge Keishinglib_path = sys.path[0] + "/../lib" 12ae7c820fSGeorge Keishingsys.path.insert(0, lib_path) 13ae7c820fSGeorge Keishingfrom gen_print import * 14ae7c820fSGeorge Keishing 15ae7c820fSGeorge Keishing# This list will be longer when more development codes are available. 1619305205SGeorge Keishinginventory_items = ['fru', 'core', 'fan', 'gpu'] 17ae7c820fSGeorge Keishingprint_var(inventory_items) 18ae7c820fSGeorge Keishingfru_inventory_file_path = 'inventory.py' 19ae7c820fSGeorge Keishingprint_var(fru_inventory_file_path) 20ae7c820fSGeorge Keishing 21ae7c820fSGeorge Keishing# Properties master list 22ae7c820fSGeorge Keishingyaml_master_list = [] 23ae7c820fSGeorge Keishing 24ae7c820fSGeorge Keishing# Clone the phosphor-dbus-interfaces repository 25ae7c820fSGeorge Keishingcmd_buf = 'git clone https://github.com/openbmc/phosphor-dbus-interfaces' 26ae7c820fSGeorge Keishingos.system(cmd_buf) 27ae7c820fSGeorge Keishing 2836fedf23SGeorge Keishingrepo_subdir_path = '/phosphor-dbus-interfaces/xyz/openbmc_project/' 29ae7c820fSGeorge Keishingbase_dir_path = os.getcwd() + repo_subdir_path 30ae7c820fSGeorge Keishing 319ef1fd47SGeorge Keishing# yaml file paths for FRU 3236fedf23SGeorge Keishingyaml_fru_list = [ 'Inventory/Item.interface.yaml', 3336fedf23SGeorge Keishing 'Inventory/Decorator/Asset.interface.yaml', 3436fedf23SGeorge Keishing 'Inventory/Decorator/Revision.interface.yaml', 3536fedf23SGeorge Keishing 'Inventory/Decorator/Replaceable.interface.yaml', 3636fedf23SGeorge Keishing 'Inventory/Decorator/Cacheable.interface.yaml', 3736fedf23SGeorge Keishing 'State/Decorator/OperationalStatus.interface.yaml', 389ef1fd47SGeorge Keishing ] 39ae7c820fSGeorge Keishing 4036fedf23SGeorge Keishing# yaml file paths for CORE 4136fedf23SGeorge Keishingyaml_core_list = [ 'Inventory/Item.interface.yaml', 4236fedf23SGeorge Keishing 'State/Decorator/OperationalStatus.interface.yaml', 4336fedf23SGeorge Keishing ] 44dc68eff6SGeorge Keishing 45dc68eff6SGeorge Keishing# yaml file paths for fan. 46dc68eff6SGeorge Keishingyaml_fan_list = [ 'Inventory/Item.interface.yaml', 47*ca8c61b7SGeorge Keishing 'Inventory/Decorator/MeetsMinimumShipLevel.interface.yaml', 48dc68eff6SGeorge Keishing 'State/Decorator/OperationalStatus.interface.yaml', 49dc68eff6SGeorge Keishing ] 5019305205SGeorge Keishing# yaml file paths for GPU. 5119305205SGeorge Keishingyaml_gpu_list = [ 'Inventory/Item.interface.yaml', 5219305205SGeorge Keishing 'State/Decorator/OperationalStatus.interface.yaml', 5319305205SGeorge Keishing ] 54ae7c820fSGeorge Keishing# Append to master list 55ae7c820fSGeorge Keishingyaml_master_list.append(yaml_fru_list) 5636fedf23SGeorge Keishingyaml_master_list.append(yaml_core_list) 57dc68eff6SGeorge Keishingyaml_master_list.append(yaml_fan_list) 5819305205SGeorge Keishingyaml_master_list.append(yaml_gpu_list) 59ae7c820fSGeorge Keishing 609ef1fd47SGeorge Keishingprint_var(yaml_master_list) 619ef1fd47SGeorge Keishing 62ae7c820fSGeorge Keishing# Populate Inventory data 63ae7c820fSGeorge Keishinginventory_dict = {} 64ae7c820fSGeorge Keishing 659ef1fd47SGeorge Keishingfor master_index in range(len(yaml_master_list)): 669ef1fd47SGeorge Keishing print_var(master_index) 67ae7c820fSGeorge Keishing inventory_dict[str(inventory_items[master_index])] = [] 689ef1fd47SGeorge Keishing for rel_yaml_file_path in yaml_master_list[master_index]: 699ef1fd47SGeorge Keishing yaml_file_path = base_dir_path + rel_yaml_file_path 70ae7c820fSGeorge Keishing 71ae7c820fSGeorge Keishing # Get the yaml dictionary data 729ef1fd47SGeorge Keishing print_timen("Loading " + yaml_file_path) 739ef1fd47SGeorge Keishing f = open(yaml_file_path) 74ae7c820fSGeorge Keishing yaml_data = yaml.load(f) 75ae7c820fSGeorge Keishing f.close() 76ae7c820fSGeorge Keishing for item in range(0, len(yaml_data['properties'])): 77ae7c820fSGeorge Keishing tmp_data = yaml_data['properties'][item]['name'] 78ae7c820fSGeorge Keishing inventory_dict[str(inventory_items[master_index])].append(tmp_data) 79ae7c820fSGeorge Keishing 80ae7c820fSGeorge Keishing# Pretty print json formatter 819ef1fd47SGeorge Keishingdata = json.dumps(inventory_dict, 829ef1fd47SGeorge Keishing indent=4, 839ef1fd47SGeorge Keishing sort_keys=True, 849ef1fd47SGeorge Keishing default=str, 859ef1fd47SGeorge Keishing separators=(',', ':')) 86ae7c820fSGeorge Keishing 87ae7c820fSGeorge Keishing# Check if there is mismatch in data vs expect list 88ae7c820fSGeorge Keishingif len(inventory_dict) != len(inventory_items): 89ae7c820fSGeorge Keishing print_error("The generated list doesn't match Master Inventory List.\n") 90ae7c820fSGeorge Keishing print data 91ae7c820fSGeorge Keishing print_var(inventory_items) 92ae7c820fSGeorge Keishing sys.exit() 93ae7c820fSGeorge Keishing 94ae7c820fSGeorge Keishing# Write dictionary data to inventory file 95ae7c820fSGeorge Keishingprint "\nGenerated Inventory item json format\n" 96ae7c820fSGeorge Keishingprint data 97ae7c820fSGeorge Keishingout = open(fru_inventory_file_path, 'w') 98ae7c820fSGeorge Keishingout.write('inventory_dict = ') 99ae7c820fSGeorge Keishingout.write(data) 100ae7c820fSGeorge Keishing 101ae7c820fSGeorge Keishingout.close() 102ae7c820fSGeorge Keishingprint "\nGenerated Inventory File: ", fru_inventory_file_path 103