message.py

import requests
import json
import os
import time


def dingtalk(msg, dingtalk_token, tries=5):
    dingtalk_url = 'https://oapi.dingtalk.com/robot/send?access_token='+dingtalk_token
    data = {
        "msgtype": "text",
        "text": {
            "content": msg
        },
        "at": {
            "isAtAll": False
        }
    }
    header = {'Content-Type': 'application/json'}

    for _ in range(tries):
        try:
            r = requests.post(dingtalk_url,
                              data=json.dumps(data), headers=header).json()
            print(r)
            if r["errcode"] == 0:
                return True
        except:
            pass
        print('Retrying...')
        time.sleep(5)
    return False


def pushplus(title, content, pushplus_token, tries=5):
    title, content = title[:100], content[:100]
    title = '微信通知服务可能即将下线，请切换到其他通知通道（建议使用钉钉）\n' + title
    pushplus_url = 'http://pushplus.hxtrip.com/customer/push/send'
    data = {
        "token": pushplus_token,
        "title": title,
        "content": content
    }
    headers = {'Content-Type': 'application/json'}

    for _ in range(tries):
        try:
            r = requests.post(pushplus_url, data=json.dumps(data),
                              headers=headers).text
            print(r)
            if '<code>200</code>' in r:
                return True
        except:
            pass
        print('Retrying...')
        time.sleep(5)
    return False


def serverchan(text, desp, serverchan_key, tries=5):
    text, desp = text[:100], desp[:100]
    text = 'Server酱服务即将下线，请切换到其他通知通道（建议使用钉钉）\n' + text
    for _ in range(tries):
        try:
            r = requests.get("https://sc.ftqq.com/" + serverchan_key
                             + ".send?text=" + text + "&desp=" + desp).json()
            print(r)
            if r["errno"] == 0:
                return True
        except:
            pass
        print('Retrying...')
        time.sleep(5)
    return False


if __name__ == "__main__":
    msg = "打卡"*1000
    dingtalk_token = os.environ.get('DINGTALK_TOKEN')
    if dingtalk_token:
        ret = dingtalk(msg, dingtalk_token)
        print('send_dingtalk_message', ret)

    serverchan_key = os.environ.get('SERVERCHAN_KEY')
    if serverchan_key:
        ret = serverchan(msg, '', serverchan_key)
        print('send_serverChan_message', ret)

    pushplus_token = os.environ.get('PUSHPLUS_TOKEN')
    if pushplus_token:
        ret = pushplus(msg, '', pushplus_token)
        print('send_pushplus_message', ret)