summaryrefslogtreecommitdiffstats
path: root/examples/dmidump.py
blob: 977504fd6b97e4acf44175f6e247fa1fce0b8600 (plain)
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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
#!/usr/bin/env python
#
#   Examples which makes use of the different python-dmidecode features
#   This script should be run as root, or else expect permission warnings
#
#   Copyright 2008-2009 Nima Talebi <nima@autonomy.net.au>
#   Copyright 2010      David Sommerseth <davids@redhat.com>
#
#   This program is free software; you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation; either version 2 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program; if not, write to the Free Software
#   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
#
#   For the avoidance of doubt the "preferred form" of this code is one which
#   is in an open unpatent encumbered format. Where cryptographic key signing
#   forms part of the process of creating an executable the information
#   including keys needed to generate an equivalently functional executable
#   are deemed to be part of the source code.
#

import dmidecode
import sys, os
from pprint import pprint

def print_warnings():
        "Simple function, dumping out warnings with a prefix if warnings are found and clearing warning buffer"
        warn = dmidecode.get_warnings()
        if warn:
              print "### WARNING: %s" % warn
              dmidecode.clear_warnings()


# Check if running as root .... provide a warning if not
root_user = (os.getuid() == 0 and True or False)
if not root_user:
        print "####"
        print "####  NOT RUNNING AS ROOT"
        print "####"
        print "#### The first run must always be done as root for this example to work."
        print "#### When not run as root, quite some permission errors might appear"
        print "####"
        print "#### If this script is first run as root, it should be possible to run this script"
        print "#### as an unprivileged user afterwards, with less warnings."
        print "####"
        print
        print


#. Test all functions using /dev/mem...  Using the legacy API
if root_user:
        print "*** bios ***\n";      dmidecode.bios()
        print_warnings()
        print "*** system ***\n";    dmidecode.system()
        print_warnings()
        print "*** system ***\n";    dmidecode.system()
        print_warnings()
        print "*** baseboard ***\n"; dmidecode.baseboard()
        print_warnings()
        print "*** chassis ***\n";   dmidecode.chassis()
        print_warnings()
        print "*** processor ***\n"; dmidecode.processor()
        print_warnings()
        print "*** memory ***\n";    dmidecode.memory()
        print_warnings()
        print "*** cache ***\n";     dmidecode.cache()
        print_warnings()
        print "*** connector ***\n"; dmidecode.connector()
        print_warnings()
        print "*** slot ***\n";      dmidecode.slot()
        print_warnings()


#. Now test get/set of memory device file...
print "*** get_dev()"
print dmidecode.get_dev()
print_warnings()
print "*** set_dev('dmidata.dump')"
print dmidecode.set_dev("dmidata.dump");
print_warnings()
print "*** get_dev()"
print dmidecode.get_dev()
print_warnings()

#. Test taking a dump...
if root_user:
        print "*** Dumping DMI data to dump file"
        print dmidecode.dump()
        print_warnings()

#. Test reading the dump...  Using the preferred API
print "*** bios ***\n";      pprint(dmidecode.QuerySection('bios'))
print_warnings()
print "*** system ***\n";    pprint(dmidecode.QuerySection('system'))
print_warnings()
print "*** system ***\n";    pprint(dmidecode.QuerySection('system'))
print_warnings()
print "*** baseboard ***\n"; pprint(dmidecode.QuerySection('baseboard'))
print_warnings()
print "*** chassis ***\n";   pprint(dmidecode.QuerySection('chassis'))
print_warnings()
print "*** processor ***\n"; pprint(dmidecode.QuerySection('processor'))
print_warnings()
print "*** memory ***\n";    pprint(dmidecode.QuerySection('memory'))
print_warnings()
print "*** cache ***\n";     pprint(dmidecode.QuerySection('cache'))
print_warnings()
print "*** connector ***\n"; pprint(dmidecode.QuerySection('connector'))
print_warnings()
print "*** slot ***\n";      pprint(dmidecode.QuerySection('slot'))
print_warnings()

print "*** Extracting memory information"
for v in dmidecode.memory().values():
  if type(v) == dict and v['dmi_type'] == 17:
    pprint(v['data']['Size']),

print "*** Querying for DMI type 3 and 7"
pprint(dmidecode.type(3))        # <-- Legacy API
pprint(dmidecode.QueryTypeId(7)) # <-- preferred API
print_warnings()

print "*** Querying for the BIOS section"
pprint(dmidecode.QuerySection('bios'))
print_warnings()

#
# Test XML stuff
#
print
print
print
print "---------------------------------------"
print "*** *** *** Testing XML API *** *** ***"
print "---------------------------------------"
print
print
dmixml = dmidecode.dmidecodeXML()

# Fetch all DMI data into a libxml2.xmlDoc object
print "*** Getting all DMI data into a XML document variable"
dmixml.SetResultType(dmidecode.DMIXML_DOC)  # Valid values: dmidecode.DMIXML_DOC, dmidecode.DMIXML_NODE
xmldoc = dmixml.QuerySection('all')

# Dump the XML to dmidump.xml - formated in UTF-8 decoding
print "*** Dumping XML document to dmidump.xml"
xmldoc.saveFormatFileEnc('dmidump.xml','UTF-8',1)

# Do some XPath queries on the XML document
print "*** Doing some XPath queries against the XML document"
dmixp = xmldoc.xpathNewContext()

# What to look for - XPath expressions
keys = ['/dmidecode/SystemInfo/Manufacturer',
	'/dmidecode/SystemInfo/ProductName',
	'/dmidecode/SystemInfo/SerialNumber',
	'/dmidecode/SystemInfo/SystemUUID']

# Extract data and print it
for k in keys:
	data = dmixp.xpathEval(k)
	for d in data:
		print "%s: %s" % (k, d.get_content())

del dmixp
del xmldoc

# Query for only a particular DMI TypeID - 0x04 - Processor
print "*** Quering for Type ID 0x04 - Processor - dumping XML document to stdout"
dmixml.QueryTypeId(0x04).saveFormatFileEnc('-','UTF-8',1)
print_warnings()