python-sdk/monitor_lib.py at master · ucloud-web/python-sdk · 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
#!/usr/bin/python

import os
import sys
import time
import socket
import re
import socket
import fcntl
import struct
import subprocess


def f_cpu():
    arr = {}

    f_stat = open("/proc/stat", "r")
    # proc.stat
    f_stat.seek(0)
    ts = int(time.time())
    for line in f_stat:
        m = re.match("(\w+)\s+(.*)", line)
        if not m:
            continue
        if m.group(1) == "cpu":
            fields = m.group(2).split()
            arr["proc.cpu.user"] =  fields[0];
            arr["proc.cpu.nice"] =  fields[1];
            arr["proc.cpu.system"] =  fields[2];
            arr["proc.cpu.idle"] =  fields[3];
            arr["proc.cpu.iowait"] =  fields[4];
            arr["proc.cpu.irq"] =  fields[5];
            arr["proc.cpu.softirq"] =  fields[6];

    return arr


def f_meminfo():
    # proc.meminfo

    arr = {}
    f_meminfo = open("/proc/meminfo", "r")
    f_meminfo.seek(0)
    ts = int(time.time())
    for line in f_meminfo:
        m = re.match("(\w+):\s+(\d+)", line)
        if m:
            key = "proc.meminfo.%s" % m.group(1).lower()
            arr[key] = m.group(2)
    return arr

def f_diskinfo():
    # 1kblocks
    arr = {}

    df_proc = subprocess.Popen(["df", "-PlTk"], stdout=subprocess.PIPE)
    stdout, _ = df_proc.communicate()
    if df_proc.returncode == 0:
        for line in stdout.split("\n"): # pylint: disable=E1103
            fields = line.split()
            # skip header/blank lines
            if not line or not fields[2].isdigit():
                continue
            # Skip mounts/types we don't care about.
            # Most of this stuff is of type tmpfs, but we don't
            # want to blacklist all tmpfs since sometimes it's
            # used for active filesystems (/var/run, /tmp)
            # that we do want to track.
            if fields[1] in ("debugfs", "devtmpfs"):
                continue
            if fields[6] == "/dev":
                continue
            # /dev/shm, /lib/init_rw, /lib/modules, etc
            #if fields[6].startswith(("/lib/", "/dev/")):  # python2.5+
            if fields[6].startswith("/lib/"):
                continue
            if fields[6].startswith("/dev/"):
                continue

            mount = fields[6]

            arr[mount] = {}
            arr[mount]["df.1kblocks.total"] = fields[2];
            arr[mount]["df.1kblocks.used"] = fields[3];
            arr[mount]["df.1kblocks.free"] = fields[4];

    return arr


'''
@summary: Get the host ip
'''
def get_ip_address(ifname='eth0'):
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    return socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915,
            struct.pack('256s',ifname[:15]))[20:24])


'''
@summary: Detect if there are readonly disks
@return: list of readonly disks; None = something error
'''
def f_detect_disk_readonly():

    arr = []
    f_logfile = open("/proc/mounts", "r")
    ts = int(time.time())
    for line in f_logfile:
        m = re.match("^\/dev\/sd(\w+) [\w\W]* ro,[\w\W]*", line)
        if m:
            device = line.split(" ")[0]
            arr.append(device)
    return arr

def main():
    print f_cpu()
    print f_meminfo()
    print f_diskinfo()

if __name__ == "__main__":
    main()