使用telegram机器人发送通知

文章目录

  • 背景
  • 1 创建机器人
  • 2 与机器人的会话
  • 3 调用API让机器人发送消息

背景

在训练深度学习模型时,除了粗略估计外,很难预测训练何时结束。此外,我们可能还想随时随地查看训练情况,如果每次都需要登录回服务器的话并不方便。因此,为我们的训练设置自动通知会大有裨益。

相关的插件和库其实很多,例如而knockknock、dinglingling,但是存在着以下缺点:

  • 依赖第三方服务,dinglingling依赖于ServerChan在微信端发送通知,是需要付费订阅会员的。
  • 需要添加装饰器,代码可读性下降。
  • 自定义能力差,仅仅起到通知作用。

基于此,个人认为比较好的方式是直接能使用免费的服务给自己发送消息,我比较推荐使用telegram(电报),因为飞书、钉钉、企业微信等都或多或少在逐步限缩个人版的机器人使用,以后可能会成为付费专属功能。

可以正常使用telegram的情况下,可以按照下面的步骤操作。

1 创建机器人

电报的机器人是我们可以自己创建的一个聊天对象,表现形式和添加的好友一样,只是名字需要以 bot 结尾,不区分大小写。

要使用机器人发消息给自己或群组,先得有一个机器人,创建一个属于自己的机器人很简单,添加BotFather这个账号,依次发送以下内容即可:

/start

/newbot

YourNameBot (给机器人取名字,以 bot 结尾,不区分大小写,直接发送过去即可,如果重名了会提示重新输入)

创建成功后BotFather会返回一个token,这是用来调用API接口的,后面会用到。

2 与机器人的会话

上面我们创建了“YourNameBot”这个机器人,要与之对话,在任意聊天窗口发送 “@YourNameBot” 然后点击这条消息即可打开与YourNameBot的聊天。

如果你需要机器人发送消息给群组,还需要把机器人拉入到群组中。

特别注意:我们需要先与机器人之间创建会话,主动给它发一次消息,机器人才能给我们发消息,否则机器人无法主动发送消息。

通过发送第一条消息,我们可以访问

https://api.telegram.org/bot/getUpdates

这个链接,把token替换成刚刚我们获得的机器人的token,就可以查看机器人的会话信息,因为我们向机器人主动发了消息,所以我们可以获取以message为key下面的[‘chat’][‘id’],这就是我们需要的chat_id(消息室的标识)。

3 调用API让机器人发送消息

现在,我们有了token、chat_id,接下来我们只需要组建我们的message就可以发送了,这个接口是:

https://api.telegram.org/bot/sendMessage

我们可以使用python的requests库来发送post请求。写了个简易的代码,大家可以根据自己的需求进行修改。

import requests
import json

def telegram_notification(epoch, avg_test_scalars):
    """
    Send notification to telegram

    params
    epoch: int
    avg_test_scalars: dict

    return: None
    """
    # chat id of user
    CHAT_ID: int = 1234567
    # telegram bot token
    TOKEN="1234567"

    url = 'https://api.telegram.org/bot{}/sendMessage'.format(TOKEN)
    headers = {
        'Content-Type': 'application/json'
    }
    # main text
    text = "Epoch: {} has completed!\n".format(epoch)
    text += json.dumps(avg_test_scalars)
    # pack data
    data = {
        'chat_id': '{}'.format(CHAT_ID),
        'text': text
    }
    # send request
    response = requests.post(url=url, headers=headers, json=data)
    
    return


if __name__ == "__main__":
    telegram_notification(1, {'a': '1', 'b': '2'})

效果如图:
使用telegram机器人发送通知_第1张图片

你可能感兴趣的:(运维,运维开发,运维)