diff options
| -rw-r--r-- | cursed.py | 5 | ||||
| -rw-r--r-- | message.py | 8 | ||||
| -rw-r--r-- | tele.py | 3 |
3 files changed, 13 insertions, 3 deletions
@@ -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}" @@ -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__": |