Formatting.

This commit is contained in:
retoor 2024-12-16 23:19:22 +01:00
parent b745d310fd
commit a28df0371f
2 changed files with 17 additions and 21 deletions

View File

@ -1,14 +1,13 @@
import asyncio
import os
from nio import AsyncClient, RoomMessageText
class BooeehBot:
def __init__(self, url, username, password):
self.url = url
self.username = username
self.password = password
self.client = AsyncClient(url, username)
async def login(self):
try:
response = await self.client.login(self.password)
@ -17,10 +16,10 @@ class BooeehBot:
except Exception as e:
print(f"Login error: {e}")
return None
async def handle_message(self, room, event):
specific_user_id = '@joewilliams007:matrix.org'
specific_user_id = "@joewilliams007:matrix.org"
if isinstance(event, RoomMessageText):
if event.sender == specific_user_id:
response_text = "booeeeh"
@ -28,26 +27,20 @@ class BooeehBot:
await self.client.room_send(
room.room_id,
message_type="m.room.message",
content={
"msgtype": "m.text",
"body": response_text
}
content={"msgtype": "m.text", "body": response_text},
)
print(f"Response to {event.sender}: " + response_text)
except Exception as e:
print(f"Failed to send message: {e}")
async def start(self):
login_response = await self.login()
if not login_response:
return
self.client.add_event_callback(self.handle_message, RoomMessageText)
await self.client.sync_forever(timeout=30000)
async def stop(self):
await self.client.close()

View File

@ -1,20 +1,23 @@
import asyncio
from boeh import env
from boeh import BooeehBot
from boeh import BooeehBot, env
async def main_async():
url = "https://matrix.org"
username = "@retoor2:matrix.org"
password = env.secret4
bot = BooeehBot(url, username, password)
try:
await bot.start()
except KeyboardInterrupt:
await bot.stop()
def main():
asyncio.run(main_async())
if __name__ == "__main__":
main()
main()