1#!/usr/bin/env python
2r"""
3Generate an inventory variable file containing a list of properties
4fields from the YAML phosphor-dbus-interfaces repository.
5"""
6import sys
7import os
8import yaml
9import json
10
11lib_path = sys.path[0] + "/../lib"
12sys.path.insert(0, lib_path)
13from gen_print import *
14
15# This list will be longer when more development codes are available.
16inventory_items = ['fru', 'core', 'fan', 'gpu']
17print_var(inventory_items)
18fru_inventory_file_path = 'inventory.py'
19print_var(fru_inventory_file_path)
20
21# Properties master list
22yaml_master_list = []
23
24# Clone the phosphor-dbus-interfaces repository
25cmd_buf = 'git clone https://github.com/openbmc/phosphor-dbus-interfaces'
26os.system(cmd_buf)
27
28repo_subdir_path = '/phosphor-dbus-interfaces/xyz/openbmc_project/'
29base_dir_path = os.getcwd() + repo_subdir_path
30
31# yaml file paths for FRU
32yaml_fru_list = [ 'Inventory/Item.interface.yaml',
33                  'Inventory/Decorator/Asset.interface.yaml',
34                  'Inventory/Decorator/Revision.interface.yaml',
35                  'Inventory/Decorator/Replaceable.interface.yaml',
36                  'Inventory/Decorator/Cacheable.interface.yaml',
37                  'State/Decorator/OperationalStatus.interface.yaml',
38                 ]
39
40# yaml file paths for CORE
41yaml_core_list = [ 'Inventory/Item.interface.yaml',
42                   'State/Decorator/OperationalStatus.interface.yaml',
43                 ]
44
45# yaml file paths for fan.
46yaml_fan_list = [ 'Inventory/Item.interface.yaml',
47                  'Inventory/Decorator/MeetsMinimumShipLevel.interface.yaml',
48                  'State/Decorator/OperationalStatus.interface.yaml',
49                ]
50# yaml file paths for GPU.
51yaml_gpu_list = [ 'Inventory/Item.interface.yaml',
52                   'State/Decorator/OperationalStatus.interface.yaml',
53                ]
54# Append to master list
55yaml_master_list.append(yaml_fru_list)
56yaml_master_list.append(yaml_core_list)
57yaml_master_list.append(yaml_fan_list)
58yaml_master_list.append(yaml_gpu_list)
59
60print_var(yaml_master_list)
61
62# Populate Inventory data
63inventory_dict = {}
64
65for master_index in range(len(yaml_master_list)):
66    print_var(master_index)
67    inventory_dict[str(inventory_items[master_index])] = []
68    for rel_yaml_file_path in yaml_master_list[master_index]:
69        yaml_file_path = base_dir_path + rel_yaml_file_path
70
71        # Get the yaml dictionary data
72        print_timen("Loading " + yaml_file_path)
73        f = open(yaml_file_path)
74        yaml_data = yaml.load(f)
75        f.close()
76        for item in range(0, len(yaml_data['properties'])):
77            tmp_data = yaml_data['properties'][item]['name']
78            inventory_dict[str(inventory_items[master_index])].append(tmp_data)
79
80# Pretty print json formatter
81data = json.dumps(inventory_dict,
82                  indent=4,
83                  sort_keys=True,
84                  default=str,
85                  separators=(',', ':'))
86
87# Check if there is mismatch in data vs expect list
88if len(inventory_dict) != len(inventory_items):
89    print_error("The generated list doesn't match Master Inventory List.\n")
90    print data
91    print_var(inventory_items)
92    sys.exit()
93
94# Write dictionary data to inventory file
95print "\nGenerated Inventory item json format\n"
96print data
97out = open(fru_inventory_file_path, 'w')
98out.write('inventory_dict = ')
99out.write(data)
100
101out.close()
102print "\nGenerated Inventory File: ", fru_inventory_file_path
103