message class and simple sent/received icon

1 files changed, 2 insertions, 1 deletions

diff --git a/tele.py b/tele.py
index c7d980b..5e014b3 100644
--- a/tele.py
+++ b/tele.py
@@ -3,6 +3,7 @@ import logging
 import telethon
 import credentials
 import cursed
+from message import Message
 
 logging.basicConfig(
         format='[%(levelname) 5s/%(asctime)s] %(name)s: %(message)s',
@@ -16,7 +17,7 @@ client = telethon.TelegramClient(
 
 @client.on(telethon.events.NewMessage)
 async def new_message(event):
-    messages.append(event.raw_text)
+    messages.append(Message(event.raw_text, event.id, "<"))
     redraw()
 
 if __name__ == "__main__":