* new dmidecode code
* move netbox creation server to ServerBase class * HP blades support
This commit is contained in:
Solvik Blum
parent
94e0437427
commit
8960df61dc
5 changed files with 321 additions and 158 deletions
|
|
@ -1,96 +1,211 @@
|
|||
import subprocess
|
||||
import re as _re
|
||||
import subprocess as _subprocess
|
||||
|
||||
class Dmidecode():
|
||||
def __init__(self):
|
||||
self.types = {
|
||||
0: 'bios',
|
||||
1: 'system',
|
||||
2: 'base board',
|
||||
3: 'chassis',
|
||||
4: 'processor',
|
||||
7: 'cache',
|
||||
8: 'port connector',
|
||||
9: 'system slot',
|
||||
10: 'on board device',
|
||||
11: 'OEM strings',
|
||||
#13: 'bios language',
|
||||
15: 'system event log',
|
||||
16: 'physical memory array',
|
||||
17: 'memory device',
|
||||
19: 'memory array mapped address',
|
||||
24: 'hardware security',
|
||||
25: 'system power controls',
|
||||
27: 'cooling device',
|
||||
32: 'system boot',
|
||||
41: 'onboard device',
|
||||
}
|
||||
self.content = self._get_output()
|
||||
self.info = self.parse_dmi()
|
||||
|
||||
def parse_dmi(self):
|
||||
"""
|
||||
Parse the whole dmidecode output.
|
||||
Returns a list of tuples of (type int, value dict).
|
||||
"""
|
||||
self.info = []
|
||||
lines = iter(self.content.strip().splitlines())
|
||||
while True:
|
||||
try:
|
||||
line = next(lines)
|
||||
except StopIteration:
|
||||
_handle_re = _re.compile("^Handle\\s+(.+),\\s+DMI\\s+type\\s+(\\d+),\\s+(\\d+)\\s+bytes$")
|
||||
_in_block_re = _re.compile("^\\t\\t(.+)$")
|
||||
_record_re = _re.compile("\\t(.+):\\s+(.+)$")
|
||||
_record2_re = _re.compile("\\t(.+):$")
|
||||
|
||||
_type2str = {
|
||||
0: 'BIOS',
|
||||
1: 'System',
|
||||
2: 'Baseboard',
|
||||
3: 'Chassis',
|
||||
4: 'Processor',
|
||||
5: 'Memory Controller',
|
||||
6: 'Memory Module',
|
||||
7: 'Cache',
|
||||
8: 'Port Connector',
|
||||
9: 'System Slots',
|
||||
10: ' On Board Devices',
|
||||
11: ' OEM Strings',
|
||||
12: ' System Configuration Options',
|
||||
13: ' BIOS Language',
|
||||
14: ' Group Associations',
|
||||
15: ' System Event Log',
|
||||
16: ' Physical Memory Array',
|
||||
17: ' Memory Device',
|
||||
18: ' 32-bit Memory Error',
|
||||
19: ' Memory Array Mapped Address',
|
||||
20: ' Memory Device Mapped Address',
|
||||
21: ' Built-in Pointing Device',
|
||||
22: ' Portable Battery',
|
||||
23: ' System Reset',
|
||||
24: ' Hardware Security',
|
||||
25: ' System Power Controls',
|
||||
26: ' Voltage Probe',
|
||||
27: ' Cooling Device',
|
||||
28: ' Temperature Probe',
|
||||
29: ' Electrical Current Probe',
|
||||
30: ' Out-of-band Remote Access',
|
||||
31: ' Boot Integrity Services',
|
||||
32: ' System Boot',
|
||||
33: ' 64-bit Memory Error',
|
||||
34: ' Management Device',
|
||||
35: ' Management Device Component',
|
||||
36: ' Management Device Threshold Data',
|
||||
37: ' Memory Channel',
|
||||
38: ' IPMI Device',
|
||||
39: ' Power Supply',
|
||||
40: ' Additional Information',
|
||||
41: ' Onboard Devices Extended Information',
|
||||
42: ' Management Controller Host Interface'
|
||||
}
|
||||
_str2type = {}
|
||||
for type_id, type_str in _type2str.items():
|
||||
_str2type[type_str] = type_id
|
||||
|
||||
|
||||
def parse():
|
||||
"""
|
||||
parse the full output of the dmidecode
|
||||
command and return a dic containing the parsed information
|
||||
"""
|
||||
buffer = _execute_cmd()
|
||||
if isinstance(buffer, bytes):
|
||||
buffer = buffer.decode('utf-8')
|
||||
_data = _parse(buffer)
|
||||
return _data
|
||||
|
||||
|
||||
def get_by_type(type_id):
|
||||
"""
|
||||
filter the output of dmidecode per type
|
||||
0 BIOS
|
||||
1 System
|
||||
2 Baseboard
|
||||
3 Chassis
|
||||
4 Processor
|
||||
5 Memory Controller
|
||||
6 Memory Module
|
||||
7 Cache
|
||||
8 Port Connector
|
||||
9 System Slots
|
||||
10 On Board Devices
|
||||
11 OEM Strings
|
||||
12 System Configuration Options
|
||||
13 BIOS Language
|
||||
14 Group Associations
|
||||
15 System Event Log
|
||||
16 Physical Memory Array
|
||||
17 Memory Device
|
||||
18 32-bit Memory Error
|
||||
19 Memory Array Mapped Address
|
||||
20 Memory Device Mapped Address
|
||||
21 Built-in Pointing Device
|
||||
22 Portable Battery
|
||||
23 System Reset
|
||||
24 Hardware Security
|
||||
25 System Power Controls
|
||||
26 Voltage Probe
|
||||
27 Cooling Device
|
||||
28 Temperature Probe
|
||||
29 Electrical Current Probe
|
||||
30 Out-of-band Remote Access
|
||||
31 Boot Integrity Services
|
||||
32 System Boot
|
||||
33 64-bit Memory Error
|
||||
34 Management Device
|
||||
35 Management Device Component
|
||||
36 Management Device Threshold Data
|
||||
37 Memory Channel
|
||||
38 IPMI Device
|
||||
39 Power Supply
|
||||
40 Additional Information
|
||||
41 Onboard Devices Extended Information
|
||||
42 Management Controller Host Interface
|
||||
"""
|
||||
if isinstance(type_id, str):
|
||||
type_id = _str2type[type_id]
|
||||
|
||||
data = parse()
|
||||
result = []
|
||||
for entry in data.values():
|
||||
if entry['DMIType'] == type_id:
|
||||
result.append(entry)
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def _execute_cmd():
|
||||
return _subprocess.check_output("dmidecode", stderr=_subprocess.PIPE)
|
||||
|
||||
|
||||
def _parse(buffer):
|
||||
output_data = {}
|
||||
# Each record is separated by double newlines
|
||||
split_output = buffer.split('\n\n')
|
||||
|
||||
for record in split_output:
|
||||
record_element = record.splitlines()
|
||||
|
||||
# Entries with less than 3 lines are incomplete / inactive; skip them
|
||||
if len(record_element) < 3:
|
||||
continue
|
||||
|
||||
handle_data = _handle_re.findall(record_element[0])
|
||||
|
||||
if not handle_data:
|
||||
continue
|
||||
handle_data = handle_data[0]
|
||||
|
||||
dmi_handle = handle_data[0]
|
||||
|
||||
output_data[dmi_handle] = {}
|
||||
output_data[dmi_handle]["DMIType"] = int(handle_data[1])
|
||||
output_data[dmi_handle]["DMISize"] = int(handle_data[2])
|
||||
|
||||
# Okay, we know 2nd line == name
|
||||
output_data[dmi_handle]["DMIName"] = record_element[1]
|
||||
|
||||
in_block_elemet = ""
|
||||
in_block_list = ""
|
||||
|
||||
# Loop over the rest of the record, gathering values
|
||||
for i in range(2, len(record_element), 1):
|
||||
if i >= len(record_element):
|
||||
break
|
||||
# Check whether we are inside a \t\t block
|
||||
if in_block_elemet != "":
|
||||
|
||||
if line.startswith('Handle 0x'):
|
||||
typ = int(line.split(',', 2)[1].strip()[len('DMI type'):])
|
||||
if typ in self.types:
|
||||
self.info.append(
|
||||
(self.types[typ], self._parse_handle_section(lines))
|
||||
)
|
||||
return self.info
|
||||
in_block_data = _in_block_re.findall(record_element[1])
|
||||
|
||||
if in_block_data:
|
||||
if not in_block_list:
|
||||
in_block_list = in_block_data[0][0]
|
||||
else:
|
||||
in_block_list = in_block_list + "\t\t" + in_block_data[0][1]
|
||||
|
||||
def _parse_handle_section(self, lines):
|
||||
"""
|
||||
Parse a section of dmidecode output
|
||||
* 1st line contains address, type and size
|
||||
* 2nd line is title
|
||||
* line started with one tab is one option and its value
|
||||
* line started with two tabs is a member of list
|
||||
"""
|
||||
data = {
|
||||
'_title': next(lines).rstrip(),
|
||||
}
|
||||
|
||||
for line in lines:
|
||||
line = line.rstrip()
|
||||
if line.startswith('\t\t'):
|
||||
if isinstance(data[k], list):
|
||||
data[k].append(line.lstrip())
|
||||
elif line.startswith('\t'):
|
||||
k, v = [i.strip() for i in line.lstrip().split(':', 1)]
|
||||
if v:
|
||||
data[k] = v
|
||||
output_data[dmi_handle][in_block_elemet] = in_block_list
|
||||
continue
|
||||
else:
|
||||
data[k] = []
|
||||
else:
|
||||
break
|
||||
# We are out of the \t\t block; reset it again, and let
|
||||
# the parsing continue
|
||||
in_block_elemet = ""
|
||||
|
||||
return data
|
||||
record_data = _record_re.findall(record_element[i])
|
||||
|
||||
# Is this the line containing handle identifier, type, size?
|
||||
if record_data:
|
||||
output_data[dmi_handle][record_data[0][0]] = record_data[0][1]
|
||||
continue
|
||||
|
||||
# Didn't findall regular entry, maybe an array of data?
|
||||
|
||||
record_data2 = _record2_re.findall(record_element[i])
|
||||
|
||||
if record_data2:
|
||||
# This is an array of data - let the loop know we are inside
|
||||
# an array block
|
||||
in_block_elemet = record_data2[0][0]
|
||||
continue
|
||||
|
||||
if not output_data:
|
||||
raise ParseError("Unable to parse 'dmidecode' output")
|
||||
|
||||
return output_data
|
||||
|
||||
|
||||
def _get_output(self):
|
||||
try:
|
||||
output = subprocess.check_output(
|
||||
'PATH=$PATH:/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin '
|
||||
'sudo dmidecode', shell=True)
|
||||
except Exception as e:
|
||||
print(e, file=sys.stderr)
|
||||
if str(e).find("command not found") == -1:
|
||||
print("please install dmidecode", file=sys.stderr)
|
||||
print("e.g. sudo apt install dmidecode",file=sys.stderr)
|
||||
|
||||
sys.exit(1)
|
||||
return output.decode()
|
||||
|
||||
def get(self, i):
|
||||
return [v for j, v in self.info if j == i]
|
||||
class ParseError(Exception):
|
||||
pass
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue