#!/usr/bin/env python
#coding: utf8

import sys
import json
import requests
import time
reload(sys)
sys.setdefaultencoding('utf-8')


def robot(data):
    webhook="群机器人webhoon地址"
    headers = {'content-type': 'application/json'}
    r = requests.post(webhook, headers=headers, data=json.dumps(data))
    r.encoding = 'utf-8'
    return (r.text)

if __name__ == "__main__":
    myfile = sys.stdin
    alert_data = json.load(myfile)
    content = "### Eights-CDH测试环境集群告警
"
    for i in range(0,len(alert_data)):
        alert=alert_data[i]["body"]["alert"]
        timestamp= time.localtime(alert["timestamp"]["epochMs"]/1000)
        content = content +"> 时间：**"+time.strftime("%Y-%m-%d %H:%M:%S",timestamp)+"**
"
        content = content +"> 内容："+alert["content"] +"
"
        content = content +"> 详情："+alert["source"]+"
"
        content = content +"> ---------------------------------------
"
    print(content)
    data={"msgtype": "markdown","markdown": {"content": content}}
    res = robot(data)
    print(res)

cat $1 > /opt/cloudera/alert/alert.log
cat $1 | python /opt/cloudera/alert/alert.py