55 lines
2.2 KiB
Python
Executable File
55 lines
2.2 KiB
Python
Executable File
import asyncio
|
|
|
|
from nio import MatrixRoom, Event
|
|
from nio.crypto import ENCRYPTION_ENABLED
|
|
|
|
from matrix_bot.bot import MatrixBot
|
|
from matrix_bot.client import MatrixClient
|
|
from matrix_bot.eventparser import MessageEventParser, ignore_when_not_concerned, EventNotConcerned
|
|
|
|
from config import env_config
|
|
from callbacks import properly_fail
|
|
from markov import parle
|
|
|
|
|
|
# le décorateur @properly_fail va permettre à la commande de laisser un message d'erreur si la commande plante et
|
|
# d'envoyer le message que le bot n'est plus en train d'écrire
|
|
# la fonction va être appelée dans tous les cas, le décorateur @ignore_when_not_concerned
|
|
# permet de laisser event_parser gérer le cas où la commande n'est pas concernée
|
|
@properly_fail
|
|
@ignore_when_not_concerned
|
|
async def repond(room: MatrixRoom, message: Event, matrix_client: MatrixClient):
|
|
# on initialise un event_parser pour décider à quel message cette commande va répondre
|
|
event_parser = MessageEventParser(room=room, event=message, matrix_client=matrix_client)
|
|
# il ne va pas répondre à ses propres messages
|
|
event_parser.do_not_accept_own_message()
|
|
|
|
# ne répond qu'en message direct ou dans un salon s'il est interpellé avec @identifiant.du.bot dans le corps du message
|
|
if len(room.users) > 2 and not (message.formatted_body and matrix_client.user_id in message.formatted_body):
|
|
raise EventNotConcerned
|
|
|
|
reponse = parle()
|
|
# il envoie l'information qu'il est en train d'écrire
|
|
await matrix_client.room_typing(room.room_id)
|
|
await asyncio.sleep(len(reponse) / 30)
|
|
# il envoie le message
|
|
await matrix_client.send_text_message(room.room_id, reponse, "m.notice")
|
|
|
|
|
|
tchap_bot = MatrixBot(
|
|
env_config.matrix_home_server,
|
|
env_config.matrix_bot_username,
|
|
env_config.matrix_bot_password,
|
|
proxy=env_config.https_proxy,
|
|
use_functions=True,
|
|
ssl=True
|
|
)
|
|
|
|
tchap_bot.matrix_client.matrix_config.encryption_enabled = True
|
|
tchap_bot.matrix_client.matrix_config.ignore_unverified_devices = True
|
|
tchap_bot.matrix_client.matrix_config.join_on_invite = True
|
|
|
|
tchap_bot.callbacks.register_on_message_event(repond)
|
|
tchap_bot.run()
|
|
|