AI_NEWS/send_matrix.py

# -*- coding: utf-8 -*-
# pip install matrix_client

from matrix_client.client import MatrixClient
from matrix_client.api import MatrixHttpApi

class MatrixBot:
    def __init__(self, user, password):
        self.base_url = "https://matrix.erhe.top"
        self.user = user
        self.password = password
        self.client = MatrixClient("https://matrix.erhe.top")
        self.token = self.login()
        self.to = "!CgWvWEnLbKYvhXLvil:chat.abeginner.cn"

    def login(self):
        self.token = self.client.login(username=self.user, password=self.password)
        return self.token

    def send_message(self, message):
        if self.token:
            try:
                api = MatrixHttpApi(self.base_url, token=self.token)
                api.send_message(self.to, message)
            except Exception as e:
                print(e)
                api = MatrixHttpApi(self.base_url, token=self.token)
                api.send_message(self.to, str(e))

        else:
            print("Bot is not logged in. Please login first.")

# if __name__ == '__main__':
#     # 测试调用
#     user = "message-bot"
#     pw = "aaaAAA111!!!"
#     message = "123987456"

#     bot = MatrixBot(user, pw)
#     bot.send_message(message)