message class and simple sent/received icon

3 files changed, 13 insertions, 3 deletions

diff --git a/cursed.py b/cursed.py
index 9ff0bec..a4c71bb 100644
--- a/cursed.py
+++ b/cursed.py
@@ -1,6 +1,7 @@
 import curses
 import asyncio
 import sys
+from message import Message
 
 class Console():
     def __init__(self, messages):
@@ -23,7 +24,7 @@ class Console():
     def redraw(self):
         self.stdscr.clear()
         for msg in self.messages:
-            self.stdscr.addstr(msg + '\n')
+            self.stdscr.addstr(str(msg) + '\n')
         if self.typed_message != "":
             self.stdscr.addstr(self.typed_message)
         self.stdscr.refresh()
@@ -36,7 +37,7 @@ class Console():
                 if char == '\n':
                     if self.typed_message != "":
                         await on_str(self.typed_message)
-                        self.messages.append(self.typed_message)
+                        self.messages.append(Message(self.typed_message, -1, ">"))
                         self.typed_message = ""
                 elif char == '\t':
                     await on_tab()
diff --git a/message.py b/message.py
new file mode 100644
index 0000000..ee80ed7
--- /dev/null
+++ b/message.py
@@ -0,0 +1,8 @@
+class Message():
+    def __init__(self, message, id, status):
+        self.message = message
+        self.id = id
+        self.status = status
+
+    def __str__(self):
+        return f"{self.status} {self.message}"
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__":