python調用釘釘機器人發起通知

有時候需要做個某些服務的狀態監控，用釘釘機器人發通知挺方便的。可以用shell腳本配合crontab檢測狀態，檢測到異常就調用python腳本發起告警。

python內容

此處用的python3，需要先安裝requests模塊。pip install requests -i https://mirrors.aliyun.com/pypi/simple

# filename: dingtalk.py
import requests
import json
import sys

def gaojing(data):
    # 將消息提交給釘釘機器人
    headers = {'Content-Type': 'application/json;charset=utf-8'}
    # 注意替換釘釘群的機器人webhook
    webhook = "https://oapi.dingtalk.com/robot/send?access_token=xxxx"
    requests.post(url=webhook,data=json.dumps(data),headers=headers)

def get_data(text_content):
    # 返回釘釘機器人所需的文本格式
    text = {
        "msgtype": "text",
        "text": {
            "content": text_content
        },
    }
    # print(json.dumps(text))
    return text

if __name__ == "__main__":
    # 命令行第一個參數為告警內容
    text_content = sys.argv[1]
    data = get_data(text_content)
    gaojing(data)

配合shell和crontab

假設有個java進程名為test.jar，監聽1234端口

#!/usr/bin/env bash
# filename: get_java_status.sh

netstat -anp | grep -v LISTENING | grep LISTEN | grep 1234
if [ $(echo $?) -eq 0 ];then 
	exit 1
else
	# 調用python腳本
	python3 /home/scripts/dingtalk.py "test.jar服務異常，1234端口不存在"
fi

crontab每5分鍾檢測一次：

*/5 * * * * /home/scripts/get_java_status.sh