Python利用Telegram机器人搭建消息提醒

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

对开发者来说，Telgram最吸引人的地方估计是 Telgram bot，你可以在建个机器人来做些事情。不过目前对我来说，最有用的一个功能是通过机器人来做消息提醒。作为开发者，我想实时知道网站的运行情况，系统报错时能即时提醒是一个痛点问题。

当然，在国内有些一些现成的解决方案，比如接入企业微信或者钉钉之类的聊天工具中，就是接入麻烦，你还的先弄个企业。也有一些第三方平台通过公众号来做提醒的。

我之前的做法是通过发短信的方式进行提醒，但是我根本就没有去看短信的习惯。短信用完了这种方案就放弃了。后来，我又想到一个新的办法，就是用邮件方式，sendcloud每天有一定免费限额的邮件发送量。将邮件提醒集成到系统，给QQ邮箱发送消息，因为QQ邮箱有一个好处是可以关联到微信，这样每当有系统有报错时都可以在第一时间收到微信的消息提醒。

近来发现Telgram的机器人也是一个备选方案，10分钟就可以快速的搭建一个出来。

申请机器人

搞一个机器人也很简单。

直接向【机器人爸爸】申请一个机器人，搜 "BotFather" ，然后给他发送一条消息 /newbot

这时候他会提示你给机器人设置一个用户名，必须以 _bot 结尾

如果遇到重复的名字会提示你换个名字。申请成功后，他会给你下发一个token。

这个token就是你后面调用api的凭证，所以要保管好，因为任何人只要拿到这个token就可以利用你的机器人操作api，如果token泄漏了可以更新token。

验证token

可以用下面的接口来验证该token是否可用

https://api.telegram.org/bot{token}/getMe

注意，token前面有个固定的 bot 单词，如果你不写会提示404错误的。

正常会返回成功消息

{
  "ok": true,
  "result": {
    "id": 12345678,
    "is_bot": true,
    "first_name": "xxxx",
    "username": "xxxxxx",
    "can_join_groups": true,
    "can_read_all_group_messages": false,
    "supports_inline_queries": false
  }
}

发消息

发消息前我们先创建一个频道专门用来接收消息
给频道设置唯一频道帐号
把机器人拉到频道里面，机器人才可以在里面发消息

调用发送消息接口

https://api.telegram.org/bot{token}/sendMessage?text=hello&chat_id=@频道id

text 是你要发送的消息
chat_id 是频道ID。

返回数据：

{
  "ok": true,
  "result": {
    "message_id": 4,
    "sender_chat": {
      "id": -110201250852,
      "title": "日志消息",
      "username": "频道id",
      "type": "channel"
    },
    "chat": {
      "id": -110201250852,
      "title": "日志消息",
      "username": "频道id",
      "type": "channel"
    },
    "date": 1654791886,
    "text": "hello"
  }
}

接口验证没问题你就可以用相应的库集成进你的系统啦

我们以flask为例，以下为核心代码

# view.py
@api.route("/error")
def exception_test():
    s = 1/0
    return success()


# app.py
@app.errorhandler(Exception)
def server_error(e):
    app.logger.error(f"内部错误{str(e)}", exc_info=True)
    if app.config.get("ENV") in ("production", 'development', 'local'):
        tb = traceback.format_exc()
        telegram.send_message(f"错误信息：{str(e)} \n堆栈信息：{tb}", chat_id=app.config.get("TELEGRAM_CHAT_ID"))
    return error(code=500, http_code=500, msg="内部错误")


# telgram.py
class Telegram:

    def __init__(self, app=None):
        self.app = app
        self._session = requests.session()
        self.token = None
        if app is not None:
            self.init_app(app)

    def init_app(self, app):
        self.app = app
        self.token = app.config.get("TELEGRAM_BOT_TOKEN")

    def send_message(self, text, chat_id):
        response = self._session.get(
            f"https://api.telegram.org/bot{self.token}/sendMessage?text={text}&chat_id=@{chat_id}")

启动程序，访问 localhost:5000/error，这时候telgram就会收到消息提醒