# -*- 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 = "!ddrrTpQmepfgivMxeW: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, e)

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

# if __name__ == '__main__':
# 测试调用
# user = "bot1"
# pw = "aaaAAA111!!!"
# message = "123987456"
#
# bot = MatrixBot(user, pw)
# bot.send_message(message)