Python_basic/parsing_introspection_data.py at master · ervikrant06/Python_basic · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
from __future__ import print_function
import json
import sys
import enum
from prettytable import PrettyTable
x=PrettyTable()

### ====================================================== ###

# sub-sub-keys for which value needs to be print from extra dictionary sub-key.

cpu_list=['physicalcpu','clock','cores','enabled_cores','frequency','physid','product','threads','vendor','version']
ipmi_list=['802.1q-vlan-id','802.1q-vlan-priority','default-gateway-ip','ip-address','ip-address-source','mac-address','subnet-mask','rmcp+-cipher-suites']
network_list=['iface','driver','firmware','serial','link','product','duplex','speed','vendor']
memory_list=['description','product','size','slot','vendor','clock']
disk_list=['disk_name','model','size','vendor','SMART/serial_number','wwn-id']

### ======================================================= ###

with open(sys.argv[1]) as data_file:
    '''
    Reading introspection json file which we are going to pass as first and
    only argument to this python script.
    '''
    data=json.load(data_file)
    options=dict()
    for index,value in enumerate(sorted(data.keys()),1):
        options[index]=value
        print("{0}: {1}".format(index,value))

def display(inputlistarguments):
    '''
    If chosen option doesn't require sub-menu then display the results for
    options chosen from main-menu.
    '''
    if type(data[options[inputlistarguments]]) == dict:
        x.field_names=data[options[inputlistarguments]].keys()
        x.add_row([eachvalue for eachvalue in data[options[inputlistarguments]].values()])
        print(x.get_string())
    else:
        print(data[options[inputlistarguments]])

def extra_options_print(input_main_key,input_sub_key):
    '''
    If "extra" is choosen from main menu then that key is input_main_key
    and sub-key choosen is input_sub_key. Depending upon sub-key values
    corresponding to sub-sub-keys are displayed.
    '''
    if input_sub_key == 'cpu':
        x.field_names=cpu_list
        for keys in sorted(data[input_main_key][input_sub_key].keys()):
            if keys.startswith('physical_'):
                physical_cpu_list=[data[input_main_key][input_sub_key][keys][eachparameter]
                                   for eachparameter in cpu_list[1:len(cpu_list)]]
                physical_cpu_list.insert(0,keys)
                x.add_row(physical_cpu_list)
        print(x.get_string())
    elif input_sub_key == 'ipmi':
        x.field_names=ipmi_list
        x.add_row([data[input_main_key][input_sub_key]['lan'][eachparameter] for eachparameter in ipmi_list])
        print(x.get_string())
    elif input_sub_key == 'network':
        x.field_names=network_list
        for keys in sorted(data[input_main_key][input_sub_key].keys()):
                network_iface_list=[data[input_main_key][input_sub_key][keys].get(eachparameter)
                                    for eachparameter in network_list[1:len(network_list)]]
                network_iface_list.insert(0, keys)
                x.add_row(network_iface_list) # list1.insert(0,keys))
        print(x.get_string())
    elif input_sub_key == 'memory':
        x.field_names=memory_list
        for keys in sorted(data[input_main_key][input_sub_key].keys()):
            if len(memory_list) == len(data[input_main_key][input_sub_key][keys]):
                x.add_row([data[input_main_key][input_sub_key][keys][eachparameter]
                           for eachparameter in memory_list])
        print(x.get_string())
    elif input_sub_key == 'disk':
        x.field_names=disk_list
        for keys in sorted(data[input_main_key][input_sub_key].keys()):
            if not keys.startswith('logical'):
                physical_disk_list=[data[input_main_key][input_sub_key][keys].get(eachparameter)
                                    for eachparameter in disk_list[1:len(disk_list)]]
                physical_disk_list.insert(0,keys)
                x.add_row(physical_disk_list)
        print(x.get_string())
    elif input_sub_key == 'firmware':
        x.field_names=sorted(data[input_main_key][input_sub_key]['bios'].keys())
        x.add_row([eachvalue for eachvalue in
                   data[input_main_key][input_sub_key]['bios'].values()])
        print(x.get_string())
    elif input_sub_key == 'system':
        x.field_names=sorted(data[input_main_key][input_sub_key]['kernel'].keys())
        x.add_row([eachvalue for eachvalue in
                   data[input_main_key][input_sub_key]['kernel'].values()])
        print(x.get_string())

def inventory_options_print(input_main_key,input_sub_key):
    '''
    input_main_key is inventory and input_sub_key is chosen from sub-menu of
    inventory. Depending up-on type of sub-key display the results.
    '''
    if type(data[input_main_key][input_sub_key]) == list:
        x.field_names=data[input_main_key][input_sub_key][0].keys()
        for eachelement in data[input_main_key][input_sub_key]:
            x.add_row(eachelement.values())
        print(x.get_string())
    elif type(data[input_main_key][input_sub_key]) == dict:
        x.field_names=data[input_main_key][input_sub_key].keys()
        x.add_row([eachvalue for eachvalue in
                   (data[input_main_key][input_sub_key].values())])
        print(x.get_string())
    else:
        print(data[input_main_key][input_sub_key])

def another_menu(input_values1):
    '''
    Display the sub-keys of chosen menu option. Call separate function once
    sub-option is choose.
    '''
    extra_options=dict()
    for index_extra,value_extra in enumerate(sorted(data[input_values1].keys()),1):
        extra_options[index_extra]=value_extra
        print("{0}: {1}".format(index_extra,value_extra))
    if input_values1 == 'inventory':
        inventory_options_print(input_values1,extra_options[int(input("Enter one numeric value: "))])
    else:
        extra_options_print(input_values1,extra_options[int(input("Enter one numeric value: "))])

def check(input_values):
    '''
    If chosen option is extra or inventory then display the sub-menu
    accordingly.
    '''
    if (options[input_values] == 'extra' or options[input_values] == 'inventory'):
        another_menu(options[input_values])
    else:
        display(input_values)

def input_validation():
    '''
    Main menu input from user and check whether it requires nested menu or
    not.
    '''
    inputvalues=str(input("Enter one numeric value: "))
    check(int(inputvalues))

input_validation()