Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / agent / env / basic / collect.py

diff --git a/vstf/vstf/agent/env/basic/collect.py b/vstf/vstf/agent/env/basic/collect.py
new file mode 100755 (executable)
index 0000000..bc4f1ee
--- /dev/null
+++ b/vstf/vstf/agent/env/basic/collect.py
@@ -0,0 +1,101 @@
+import os
+import platform
+import logging
+from collections import OrderedDict
+
+from vstf.agent.env.basic.commandline import CommandLine
+from vstf.common import constants as const
+
+log = logging.getLogger(__name__)
+CMD = CommandLine()
+
+
+class Collect(object):
+    """collect host information such as _cpu, memory and so on"""
+
+    def __init__(self):
+        super(Collect, self).__init__()
+        self._system = self._system()
+        self._cpu = self._cpu()
+
+    def _system(self):
+        """the base _system info
+        {'os info':{'_system':'ubuntu', 'kernel': '3.13.3'}}"""
+        return {const.OS_INFO:
+            {
+                '_system': open('/etc/issue.net').readline().strip(),
+                'kernel': platform.uname()[2]
+            }
+        }
+
+    def _memery(self):
+        """ Return the information in /proc/meminfo
+        as a dictionary """
+        meminfo = OrderedDict()
+        with open('/proc/meminfo') as f:
+            for line in f:
+                meminfo[line.split(':')[0]] = line.split(':')[1].strip()
+
+        return {const.MEMORY_INFO:
+            {
+                "Mem Total": meminfo['MemTotal'],
+                "Mem Swap": meminfo['SwapTotal']
+            }
+        }
+
+    def _lscpu(self):
+        ret = {}
+        with os.popen("lscpu") as f:
+            for line in f:
+                ret[line.split(':')[0].strip()] = line.split(':')[1].strip()
+        return ret
+
+    def _cpu(self):
+        ret = []
+        with open('/proc/cpuinfo') as f:
+            cpuinfo = OrderedDict()
+            for line in f:
+                if not line.strip():
+                    ret.append(cpuinfo)
+                    cpuinfo = OrderedDict()
+                elif len(line.split(':')) == 2:
+                    cpuinfo[line.split(':')[0].strip()] = line.split(':')[1].strip()
+                else:
+                    log.error("_cpu info unknow format <%(c)s>", {'c': line})
+        return {const.CPU_INFO:
+            dict(
+                {
+                    "Model Name": ret[0]['model name'],
+                    "Address sizes": ret[0]['address sizes']
+                },
+                **(self._lscpu())
+            )
+        }
+
+    def _hw_sysinfo(self):
+        cmdline = "dmidecode | grep  -A 2 'System Information' | grep -v 'System Information'"
+        ret, output = CMD.execute(cmdline, shell=True)
+        if ret:
+            result = {}
+            # del the stderr
+            for tmp in output.strip().split('\n'):
+                if tmp is None or tmp is "":
+                    continue
+                # split the items 
+                tmp = tmp.split(":")
+                if len(tmp) >= 2:
+                    # first item as key, and the other as value
+                    result[tmp[0].strip("\t")] = ";".join(tmp[1:])
+            return {const.HW_INFO: result}
+        else:
+            return {const.HW_INFO: "get hw info failed. check the host by cmd: dmidecode"}
+
+    def collect_host_info(self):
+        return [self._system, self._cpu, self._memery(), self._hw_sysinfo()]
+
+
+if __name__ == "__main__":
+    c = Collect()
+    import json
+
+    print json.dumps(c.collect_host_info(), indent=4)