aboutsummaryrefslogtreecommitdiffstats
path: root/tools/systeminfo.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/systeminfo.py')
-rw-r--r--tools/systeminfo.py139
1 files changed, 139 insertions, 0 deletions
diff --git a/tools/systeminfo.py b/tools/systeminfo.py
new file mode 100644
index 00000000..19c5d16e
--- /dev/null
+++ b/tools/systeminfo.py
@@ -0,0 +1,139 @@
+# Copyright 2015 Intel Corporation.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Tools for access to OS details
+"""
+
+import os
+import platform
+import subprocess
+import locale
+
+from conf import settings
+
+def get_os():
+ """Get distro name.
+
+ :returns: Return distro name as a string
+ """
+ return ' '.join(platform.dist())
+
+def get_kernel():
+ """Get kernel version.
+
+ :returns: Return kernel version as a string
+ """
+ return platform.release()
+
+def get_cpu():
+ """Get CPU information.
+
+ :returns: Return CPU information as a string
+ """
+ with open('/proc/cpuinfo') as file_:
+ for line in file_:
+ if not line.strip():
+ continue
+ if not line.rstrip('\n').startswith('model name'):
+ continue
+
+ return line.rstrip('\n').split(':')[1]
+
+def get_nic():
+ """Get NIC(s) information.
+
+ :returns: Return NIC(s) information as a string
+ """
+ nics = []
+ output = subprocess.check_output('lspci', shell=True)
+ output = output.decode(locale.getdefaultlocale()[1])
+ for line in output.split('\n'):
+ for nic_pciid in settings.getValue('WHITELIST_NICS'):
+ if line.startswith(nic_pciid):
+ nics.append(''.join(line.split(':')[2:]).strip())
+ return ', '.join(nics).strip()
+
+def get_platform():
+ """Get platform information.
+
+ Currently this is the motherboard vendor, name and socket
+ count.
+
+ :returns: Return platform information as a string
+ """
+ output = []
+
+ with open('/sys/class/dmi/id/board_vendor', 'r') as file_:
+ output.append(file_.readline().rstrip())
+
+ with open('/sys/class/dmi/id/board_name', 'r') as file_:
+ output.append(file_.readline().rstrip())
+
+ num_nodes = len([name for name in os.listdir(
+ '/sys/devices/system/node/') if name.startswith('node')])
+ output.append(''.join(['[', str(num_nodes), ' sockets]']))
+
+ return ' '.join(output).strip()
+
+def get_cpu_cores():
+ """Get number of CPU cores.
+
+ :returns: Return number of CPU cores
+ """
+ cores = 0
+ with open('/proc/cpuinfo') as file_:
+ for line in file_:
+ if line.rstrip('\n').startswith('processor'):
+ cores += 1
+ continue
+
+ # this code must be executed by at leat one core...
+ if cores < 1:
+ cores = 1
+ return cores
+
+def get_memory():
+ """Get memory information.
+
+ :returns: amount of system memory as string together with unit
+ """
+ with open('/proc/meminfo') as file_:
+ for line in file_:
+ if not line.strip():
+ continue
+ if not line.rstrip('\n').startswith('MemTotal'):
+ continue
+
+ return line.rstrip('\n').split(':')[1].strip()
+
+def get_memory_bytes():
+ """Get memory information in bytes
+
+ :returns: amount of system memory
+ """
+ mem_list = get_memory().split(' ')
+ mem = float(mem_list[0].strip())
+ if mem_list.__len__() > 1:
+ unit = mem_list[1].strip().lower()
+ if unit == 'kb':
+ mem *= 1024
+ elif unit == 'mb':
+ mem *= 1024 ** 2
+ elif unit == 'gb':
+ mem *= 1024 ** 3
+ elif unit == 'tb':
+ mem *= 1024 ** 4
+
+ return int(mem)
+