python小脚本,实时监测服务器是否宕机状态,并发送到指定Telegram群组

✍️Auth:运维笔记       Date:2024/09/11       Cat:python相关       👁️:127 次浏览

一,前言

众所周知,市面上监控软件很多,有Zabbix,Prometheus等,但对于相对简单的功能,需要第一时间发现问题,如服务器宕机,zabbix和Prometheus都需要等几分钟才会报警。

想到最原始的方法,也是最简单的方法,就是ping 服务器,ping断开就是宕机。所以用一个小脚本即可。

二,脚本内容

1,脚本说明:

  • 每个服务器都有对应的名称,保存在字典 servers 中。
  • 如果某个服务器连续 10 秒无法 ping 通,则向 TG群发送服务器的名称和 IP(根据自己需求添加发送指定的媒体)。
  • 可以根据需求调整 time.sleep(10) 的时间间隔。
  • 状态跟踪: server_status 字典保存了每个服务器的当前状态,True 表示服务器在线,False 表示服务器不在线。
  • 状态变化检测:
    如果服务器从“在线”变为“不在线”,就会被标记为挂掉,发送“服务器挂掉”通知。
    如果服务器从“不在线”变为“在线”,就会被标记为恢复,发送“服务器恢复”通知。
  • 状态更新: 状态在每次检测后更新为当前状态。
    这样就能在服务器挂掉时收到通知,在恢复后也会及时得到通知。
  • 每次检测时挂掉服务器都会发送通知: 即使之前已经挂掉过,只要服务器继续挂掉,脚本会在每次循环中发送挂掉信息,如果只发送一次,会误发或者忽略,所以重复发送可以肯定服务器状态。
  • 恢复的服务器只发送一次通知: 通过跟踪状态,只在服务器从“不在线”状态切换到“在线”状态时发送“恢复”通知。重复发的挂掉信息则会停止。

2,脚本内容

import os
import time
import requests

# 服务器列表,格式为 { '服务器名称': '服务器IP' }
servers = {
    '测试服务器1': '192.168.203.132',
    'servername2': '192.168.1.2',
    'servername3': '192.168.1.3'
}

# 记录每个服务器的状态,True 表示在线,False 表示不在线
server_status = {name: True for name, ip in servers.items()}

# TG 机器人 API Token 和群 ID
telegram_token = '5249774617:asdasdasd.....'
chat_id = '-459853...'

# 向 Telegram 群发送通知的函数
def send_telegram_message(message):
    url = f'https://api.telegram.org/bot{telegram_token}/sendMessage'
    data = {
        'chat_id': chat_id,
        'text': message
    }
    try:
        requests.post(url, data=data)
    except Exception as e:
        print(f"发送消息失败: {e}")

# 检查服务器状态的函数
def check_servers():
    down_servers = []
    recovered_servers = []

    for name, ip in servers.items():
        response = os.system(f"ping -c 5 {ip} > /dev/null 2>&1")
        is_online = (response == 0)

        # 如果服务器挂掉(即 ping 不通)
        if not is_online:
            # 如果之前是在线的,刚刚挂掉,标记为挂掉并发送通知
            if server_status[name]:
                server_status[name] = False
            # 每次服务器挂掉时都发送通知
            down_servers.append((name, ip))

        # 如果服务器恢复(即 ping 通),只发送一次恢复通知
        elif not server_status[name] and is_online:
            recovered_servers.append((name, ip))
            server_status[name] = True

    return down_servers, recovered_servers

# 主循环
while True:
    down_servers, recovered_servers = check_servers()

    # 处理挂掉的服务器
    for name, ip in down_servers:
        send_telegram_message(f"❌{name} -Server DOWN,IP:({ip})")

    # 处理恢复的服务器
    for name, ip in recovered_servers:
        send_telegram_message(f"✅{name} -Server UP,IP:({ip})")

    time.sleep(10)  # 每 10 秒检测一次

三,脚本运行测试

原脚本由于是持续运行,所以是死循环,只要后台运行一次即可。

nohup python3 monitor.py >dev/null 2>&1 &

模拟测试结果,完成。


后续可以继续完善,可以写个检测此脚本的小shell程序,定时检查此脚本,万一机器重启,可以重启自动重启。

参考我以前文章:grep在shell脚本中使用,判断进程是否存在

打赏作者

发表评论