Centos7下用Python写一个磁盘空间自动预警脚本(以PCB制前工程InCAM/InCAMPro资料库及server目录为例)

思路分析:
(1).查看InCAM/InCAMPro资料库路径;

# 进入InCAM/InCAMPro资料库配置文件(dblist.xml)路径,查看配置文件资料库配置路径
# vim /incam/server/site_data/dblist.xml

如下图获取资料库路径为"/incam/camdb":

(2).获取资料库路径挂载点,在终端执行"df -h"指令;

# df -h

如下图获取资料库挂载分区及详细的使用状况:

(3).获取server实际路径:

# ls -al /incam/server

如下图获取server实际路径为"/backup/share/server":

(4).获取server实际路径挂载点,在终端执行"df -h"指令;

# df -h

如下图获取server挂载分区及详细的使用状况:

编写监控代码(监控CAM资料库及server空间状况):

#!/usr/bin/env py3k
import re
import sys
import os
import time

sys.path.append("/incam/server/site_data/scripts/py3_package")
import sendMail

# 开发日志
_header = {
    '脚本名称': '磁盘空间预警程式',
    '开发人员': 'Twei Tang',
    '开发时间': '2022年11月18日',
    '版本信息': 'A.1.0',
    '联系方式': (
        '邮箱地址<>',
        '微信号码<358143105>',
        '手机号码<13627441202>'
    ),
    '开发信息': (
        '无'
    ),
    '修改信息': (
        '初始版本(A.1.0)，首次开发测试，暂无版本变更信息'
    ),
    '沟通记录': (
        '无'
    )
}

if __name__ == "__main__":
    _back_time = time.strftime('%Y-%m-%d %H:%M:%S')
    _log_time = time.strftime('%Y-%m-%d-%H')
    _dict = {}
    _ser_dict = {}
    # _db_size_path = os.environ.get("INCAM_DIR") + "/tmp/db_size.213"    #  自动任务懒得写环境变量,直接使用绝对路径
    _db_size_path = "/incam/tmp/db_size.213"
    _server_size_path = "/incam/tmp/server_size.213"

    try:
        os.system(f"rm -rf {_db_size_path}")
        os.system(f"rm -rf {_server_size_path}")
    except BaseException as e:
        print(e)

    # 某某服务器server路径
    _ser_grep = "/backup"
    try:
        os.system(f"df -h |grep {_ser_grep} > {_server_size_path}")
    except BaseException as e:
        print(e)

    # 某某服务器CAM资料库存储路径
    _grep = "/incam/camdb"
    try:
        os.system(f"df -h |grep {_grep} > {_db_size_path}")
    except BaseException as e:
        print(e)

    _ser_cut = 0
    _ser_cut_size = 80		# server空间预警值
    if os.path.exists(_server_size_path):
        with open(_server_size_path, "r") as f:
            _lines = f.readlines()
        _size = re.sub(" +", " ", _lines[0])
        _tmp_list = _size.split()
        if _ser_grep == "/backup":
            _ser_dict["磁盘路径:"] = _tmp_list[0]
            _ser_dict["磁盘容量:"] = _tmp_list[1]
            _ser_dict["已用空间:"] = _tmp_list[2]
            _ser_dict["剩余空间:"] = _tmp_list[3]
            _ser_dict["使用比率:"] = _tmp_list[4]
            _ser_dict["映射路径:"] = _tmp_list[5]
            _ser_cut = float(_tmp_list[4].strip('%'))

    _cut = 0
    _cut_size = 90		# CAM资料库空间预警值	
    if os.path.exists(_db_size_path):
        with open(_db_size_path, "r") as f:
            _lines = f.readlines()
        _size = re.sub(" +", " ", _lines[0])
        _tmp_list = _size.split()
        if _grep == "/incam/camdb":
            _dict["磁盘路径:"] = _tmp_list[0]
            _dict["磁盘容量:"] = _tmp_list[1]
            _dict["已用空间:"] = _tmp_list[2]
            _dict["剩余空间:"] = _tmp_list[3]
            _dict["使用比率:"] = _tmp_list[4]
            _dict["映射路径:"] = _tmp_list[5]
            _cut = float(_tmp_list[4].strip('%'))
        else:
            _dict["磁盘路径:"] = "你的服务器IP地址:/incam/camdb"
            _dict["磁盘容量:"] = _tmp_list[0]
            _dict["已用空间:"] = _tmp_list[1]
            _dict["剩余空间:"] = _tmp_list[2]
            _dict["使用比率:"] = _tmp_list[3]
            _dict["映射路径:"] = _tmp_list[4]
            _cut = float(_tmp_list[3].strip('%'))

    if len(_dict.keys()) > 0 and _cut >= _cut_size or len(_ser_dict.keys()) > 0 and _ser_cut >= _ser_cut_size:
        if len(_dict.keys()) > 0 and _cut >= _cut_size:
            _string = "<br>".join([k + v for k, v in _dict.items()])
            _string += f'<br>存储空间超预警值({_cut_size}%)!'

        if len(_ser_dict.keys()) > 0 and _ser_cut >= _ser_cut_size:
            _string = "<br>".join([k + v for k, v in _ser_dict.items()])
            _string += f'<br>存储空间超预警值({_ser_cut_size}%)!'

        att_list = []
        _log = "某某服务器CAM资料库存储空间告警!<br>" + _string
        mess_list = [f'<tr bgcolor=#DDDDDD><th>disk_space_warning.py</th><th>{_back_time}</th><th align=left>{_log}</th></tr>']
        mess_text = "\n".join(mess_list)
        html_text = f'''
                    <p>某某服务器存储空间预警</p>
                     <table border=2 bordercolor=#1212FC>
                     <tr border=2 bgcolor=#FF6666><th>程序</th><th>时间</th><th>预警信息</th></tr>
                     {mess_text}
                     </table>
                    '''
        _to_list = ["用户邮箱1", "用户邮箱2"]
        try:
            SM = sendMail.MailMain("某某服务器(disk_space_warning.py)",
                                   "[email protected]",
                                   _to_list,
                                   f"{html_text}",
                                   f"某某服务器存储空间预警")
            if len(att_list) > 0:
                SM.SendMail(text_type="html", att_type=att_list)
            else:
                SM.SendMail(text_type="html")
        except BaseException as e:
        	# 错误日志记录
            _error_log = f"/incam/server/logs/存储空间预警邮件发送失败日志/{_log_time}.log"
            if not os.path.exists(f"/incam/server/logs/存储空间预警邮件发送失败日志"):
                os.mkdir(f"/incam/server/logs/存储空间预警邮件发送失败日志")
            if not os.path.exists(_error_log):
                os.system(f"touch {_error_log}")
            with open(_error_log, "a", encoding="utf-8") as f:
                f.write(f"Error: 无法发送邮件, 详情:{e}\n")

    try:
        os.system(f"rm -rf {_db_size_path}")
        os.system(f"rm -rf {_server_size_path}")
    except BaseException as e:
        print(e)

    sys.exit()

配置脚本自动执行: