如何用 Python 发送告警通知到微信?

495 阅读4分钟

最近当上了阿里云的推广大使,带了 200 名读者一起免费领取了阿里云的服务器,每个人都说 『真香』。

PS:有想参与领取免费服务器的同学,可加我v:hello-wbm,我发你操作流程。

组织第一期活动,其实还是有不少的问题,主要流程上的问题。

为了让整个流程更加自动化,操作更加流畅,我利用公众号的开发能力,将整个过程集成到公众号消息的自助查询。

其中有一步就是确认用户的购买资格,只要在我的公众号后台回复对应的阿里云ID 我这边就会去查询阿里云后台的关联数据,但阿里云的 cookie 数个小时就会失效,这样一来就有点尴尬,后台的爬虫失效了参与的读者还在一直查,一直查,查到的都是未关联的数据。

这时候实时告警就显得非常重要,常见的告警方式有:邮件,电话,短信,微信。

短信和电话,通常是收费的(有不收费的,可以分享一下),而邮件又不是那么及时,因此最后我选择微信通知。

这里说的微信,是企业微信,而我之前用注册过个体户的执照,因此可以很轻松就可以注册自己的企业微信。

1. 新建应用

登陆网页版企业微信 (work.weixin.qq.com/),点击 应用管理 -> 应用 -> 创建应用

上传应用的 logo,输入应用名称,再选择可见范围,成功创建一个告警应用

2. 获取Secret

使用 Python 发送告警请求,其实就只使用到两个接口

  • 获取 Tokenhttps://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={corpid}&corpsecret={secret}
  • 发送请求https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token={token}

可以看到,最重要的是 corpid 和 secret:

  • corpid:唯一标识你的企业
  • secret:应用级的密钥,有了程序才知道你要发送该企业的哪个应用

corpid 可以通过 我的企业 -> 企业信息 获取

而 secret 获取相对麻烦一点,点击前面创建应用,点击 查看 secret

然后再点击发送就会发送到你的企业微信上

最后将 corpid 和 secret 填入下面的常量中。

import json
import datetime
import requests

CORP_ID = ""
SECRET = ""

class WeChatPub:
    s = requests.session()

    def __init__(self):
        self.token = self.get_token()

    def get_token(self):
        url = f"https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={CORP_ID}&corpsecret={SECRET}"
        rep = self.s.get(url)
        if rep.status_code != 200:
            print("request failed.")
            return
        return json.loads(rep.content)['access_token']


    def send_msg(self, content):
        url = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=" + self.token
        header = {
            "Content-Type": "application/json"
        }
        form_data = {
            "touser": "@all",
            "toparty": " PartyID1 | PartyID2 ",
            "totag": " TagID1 | TagID2 ",
            "msgtype": "textcard",
            "agentid": 1000002,
            "textcard": {
                "title": "服务异常告警",
                "description": content,
                "url": "URL",
                "btntxt": "更多"
            },
            "safe": 0
        }
        rep = self.s.post(url, data=json.dumps(form_data).encode('utf-8'), headers=header)
        if rep.status_code != 200:
            print("request failed.")
            return
        return json.loads(rep.content)

然后就可以通过 send_msg 函数发送消息了。

wechat = WeChatPub()
now = datetime.datetime.now()
timenow = now.strftime('%Y年%m月%d日 %H:%M:%S')
wechat.send_msg(f"<div class=\"gray\">{timenow}</div> <div class=\"normal\">阿里云 cookie 已失效</div><div class=\"highlight\">请尽快更换新的 cookie</div>")

只要你的企业微信没有关闭通知的权限,那你的手机立马就会弹出这个告警信息。

简单几步就对接了企业微信,实现了手机的实时告警功能,推荐有企业微信的同学使用。

当然一定有更多,更好用的实现方法,我只是我选择了其中一种,大家有不错的思路也可以分享在评论区。

絮叨一下

我在掘金上写过很多的 Python 相关文章,其中包括 Python 实用工具,Python 高效技巧,PyCharm 使用技巧,很高兴得到了很多知乎朋友的认可和支持。

在他们的鼓励之下,我将过往文章分门别类整理成三本 PDF 电子书

PyCharm 中文指南

《PyCharm 中文指南》使用 300 多张 GIF 动态图的形式,详细讲解了最贴合实际开发的 105个 PyCharm 高效使用技巧,内容通俗易懂,适合所有 Python 开发者。

在线体验地址:pycharm.iswbm.com

Python 黑魔法指南

《Python黑魔法指南》目前迎来了 v3.0 的版本,囊集了 100 多个开发小技巧,非常适合在闲时进行碎片阅读。

在线体验地址:magic.iswbm.com

Python 中文指南

学 Python 最好的学习资料永远是 Python 官方文档,可惜现在的官方文档大都是英文,虽然有中文的翻译版了,但是进度实在堪忧。为了照顾英文不好的同学,我自己写了一份 面向零基础的朋友 的在线 Python 文档 -- 《Python中文指南》

在线体验地址:python.iswbm.com

**有帮助的话,记得帮我 点个赞哟~