python通过内置模块监控磁盘内存CPU负载

Posted 2021-01-20 mairoot

01. 概述

闲来无事干，说干就干。主要是通过python函数以及python内置模块来实现对磁盘、内存、CPU、负载的数据采集，然后发送到企业号，并到达微信端，方便查看信息。
脚本存放于gitlab上：python监控
原文地址：https://www.mairoot.com/?p=1708

02. 脚本附件

#!/usr/bin/env python3
# coding=utf-8
# Create for mai
 # Copyright https://www.mairoot.com
 # Create date 2018-10-21

# 引入模块
import os, time, socket, requests, json, urllib.request, re
from collections import namedtuple
from collections import OrderedDict

# ------------------------通用信息--------------------------------
# 业务信息
run_project="test-all"
# 获取日期时间
get_date=time.strftime("%Y-%m-%d", time.localtime())
get_date_time=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
# IP地址
def get_host_ip():
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        s.connect((‘8.8.8.8‘, 80))
        ip = s.getsockname()[0]
    finally:
        s.close()
    return ip
inside_net=get_host_ip()
outside_net="1.1.1.1"
# 主机名
host_name=socket.gethostname()
# 发生事件
hint_content="None"

# ------------------------磁盘信息--------------------------------
# 获取磁盘信息
disk_ntuple = namedtuple(‘partition‘, ‘device mountpoint fstype‘)
usage_ntuple = namedtuple(‘usage‘, ‘total used free percent‘)

# 获取所有磁盘设备
def disk_partitions(all=False):
    """Return all mountd partitions as a nametuple.
    If all == False return phyisical partitions only.
    """
    phydevs = []
    f = open("/proc/filesystems", "r")
    for line in f:
        if not line.startswith(‘none‘):
            phydevs.append(line.strip())

        retlist = []
        f = open(‘/etc/mtab‘, "r")
        for line in f: 
            if not all and line.startswith(‘none‘):
                continue
            fields = line.split()
            device = fields[0]
            mountpoint = fields[1]
            fstype = fields[2]
            if not all and fstype not in phydevs:
                continue
            if device == ‘none‘:
                device = ‘‘
            ntuple = disk_ntuple(device, mountpoint, fstype)
            retlist.append(ntuple)
        return retlist

# 统计磁盘使用情况
def disk_state(path):
    """Return disk usage associated with path."""
    st = os.statvfs(path)
    free = (st.f_bavail * st.f_frsize)
    total = (st.f_blocks * st.f_frsize)
    used = (st.f_blocks - st.f_bfree) * st.f_frsize
    try:
        percent = ret = (float(used) / total) * 100
    except ZeroDivisionError:
        percent = 0
    return usage_ntuple(round(total/1024/1024/1024, 3), round(used/1024/1024/1024, 3), round(free/1024/1024/1024, 3), round(percent, 3)) 

# ------------------------内存信息--------------------------------
def get_meminfo():
    ‘‘‘ 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 meminfo

# ------------------------负载信息--------------------------------
def get_load():
    f = open("/proc/loadavg")
    loadstate=f.read().split() 
    return loadstate

# ------------------------CPU信息--------------------------------
def get_cpu_use():
    last_worktime=0
    last_idletime=0
    f=open("/proc/stat","r")
    line=""
    while not "cpu " in line: line=f.readline()
    f.close()
    spl=line.split(" ")
    worktime=int(spl[2])+int(spl[3])+int(spl[4])
    idletime=int(spl[5])
    dworktime=(worktime-last_worktime)
    didletime=(idletime-last_idletime)
    rate=float(dworktime)/(didletime+dworktime)
    cpu_t = rate*100
    last_worktime=worktime
    last_idletime=idletime
    if(last_worktime==0): return 0
    return cpu_t

# ------------------------发送信息--------------------------------
# 定义验证信息
ID="ww4730ead71a1818a6"  
Secret="y4CCI-4LoiWmuqs6A5kYRpEyzhUlCveKeQ_Ik_WeEW4"
UserID = "mai"
PartyID=1
AppID = 1000002

# 获取token
def get_token():
    gurl = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={}&corpsecret={}".format(ID, Secret)
    r=requests.get(gurl)
    dict_result= (r.json())
    return dict_result[‘access_token‘]
def get_media_ID(path):
    Gtoken = get_token()
    img_url = "https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token={}&type=image".format(Gtoken)
    files = {‘image‘: open(path, ‘rb‘)}
    r = requests.post(img_url, files=files)
    re = json.loads(r.text)
    return re[‘media_id‘]

# 发送文本 
def  send_text(text): 
    post_data = {}
    msg_content = {}
    msg_content[‘content‘] = text
    post_data[‘touser‘] = UserID
    post_data[‘toparty‘] = PartyID
    post_data[‘msgtype‘] = ‘text‘
    post_data[‘agentid‘] = AppID
    post_data[‘text‘] = msg_content
    post_data[‘safe‘] = ‘0‘
    Gtoken = get_token()
    purl1="https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token={}".format(Gtoken)
    json_post_data = json.dumps(post_data,False,False)
    request_post = urllib.request.urlopen(purl1,json_post_data.encode(encoding=‘UTF8‘))
    return request_post
    
# 发送图片 
def  send_tu(path):
    img_id = get_media_ID(path)
    post_data1 = {}
    msg_content1 = {}
    msg_content1[‘media_id‘] = img_id
    post_data1[‘touser‘] = UserID
    post_data1[‘toparty‘] = PartyID
    post_data1[‘msgtype‘] = ‘image‘
    post_data1[‘agentid‘] = AppID
    post_data1[‘image‘] = msg_content1
    post_data1[‘safe‘] = ‘0‘
    Gtoken = get_token()
    purl2="https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token={}".format(Gtoken)
    json_post_data1 = json.dumps(post_data1,False,False)
    request_post = urllib.request.urlopen(purl2,json_post_data1.encode(encoding=‘UTF8‘))
    return request_post
    
# ------------------------修改内容--------------------------------
# 磁盘 -------------------
def disk_send_mgs():
    disk_paths=[‘/‘, ‘/boot‘, ‘/data‘]
    for disk_path in disk_paths:
        disk_new_state=disk_state(disk_path)
        disk_percent=float(disk_state(disk_path)[3])
        #print(disk_new_state)
        #print(disk_percent)
        if disk_percent > 80:
            run_events="目录{}使用率为{}%".format(disk_path, disk_new_state[3])
            disk_send_state="获取时间:{};
运行业务:{};
远程地址:{};
内网地址:{};
主机名称:{};
发生事件:{};
提示内容:{}.".format(get_date_time, run_project, outside_net, inside_net, host_name, run_events, hint_content)
            #print (disk_new_state)
            send_text(disk_send_state)
# 内存 -------------------
def mem_send_mgs():
    get_run_meminfo=get_meminfo() #数据转换为list
    memtotal=format(get_run_meminfo[‘MemTotal‘])
    memtotal=float(memtotal.split()[0])
    memfree=format(get_run_meminfo[‘MemFree‘])
    memfree=float(memfree.split()[0])
    memswap=format(get_run_meminfo[‘SwapTotal‘])
    memswap=float(memswap.split()[0])
    memswapfree=format(get_run_meminfo[‘SwapFree‘])
    memswapfree=float(memswapfree.split()[0])
    mem_percent=round(((memtotal - memfree) / memtotal)*100, 1)
    memswap_percent=round(((memswap - memswapfree) / memswap)*100, 1)
    if mem_percent > 80:
        run_events="内存使用率超过{}%".format(mem_percent)
        mem_send_state="获取时间:{};
运行业务:{};
远程地址:{};
内网地址:{};
主机名称:{};
发生事件:{};
提示内容:{}.".format(get_date_time, run_project, outside_net, inside_net, host_name, run_events, hint_content)
        send_text(mem_send_state)
        #print (mem_percent)
    if memswap_percent > 80:
        run_events="交换内存使用率超过{}%".format(memswap_percent)
        memswap_send_state="获取时间:{};
运行业务:{};
远程地址:{};
内网地址:{};
主机名称:{};
发生事件:{};
提示内容:{}.".format(get_date_time, run_project, outside_net, inside_net, host_name, run_events, hint_content)
        send_text(memswap_send_state)
        #print (memswap_percent)
# 负载 -------------------
def load_send_mgs():
    loadstate=get_load()
    loadstate1=loadstate[0]
    loadstate15=float(loadstate[2])
    if loadstate15 > 5:
        run_events="系统负载过高{}%".format(loadstate15)
        load_send_state="获取时间:{};
运行业务:{};
远程地址:{};
内网地址:{};
主机名称:{};
发生事件:{};
提示内容:{}.".format(get_date_time, run_project, outside_net, inside_net, host_name, run_events, hint_content)
        send_text(load_send_state)
        #print(loadstate15)
# CPU -------------------
def cpu_send_mgs():
    cpuuse_percent=round(get_cpu_use(),3)
    if cpuuse_percent > 80:
        run_events="CPU使用过高{}%".format(cpuuse_percent)
        cpu_send_state="获取时间:{};
运行业务:{};
远程地址:{};
内网地址:{};
主机名称:{};
发生事件:{};
提示内容:{}.".format(get_date_time, run_project, outside_net, inside_net, host_name, run_events, hint_content)
        send_text(cpu_send_state)

# ------------------------执行内容--------------------------------
# 运行main函数
if __name__==‘__main__‘:
    disk_send_mgs()
    mem_send_mgs()
    load_send_mgs()
    cpu_send_mgs()

03. 微信端接收信息