initial commit

pgbotlib/misc.py

@@ -0,0 +1,42 @@
+import telethon
+import pgbotlib.dbstuff
+import pgbotlib.response
+
+
+class NameGenerator:
+    def __init__(self, config: dict, db: pgbotlib.dbstuff.DBConn) -> None:
+        self.db = db
+
+    def get_name(self, sender: telethon.tl.types.User) -> str:
+        query = 'SELECT name FROM names WHERE tg_id = %s'
+        result = self.db.query_random(query, (sender.id,))
+        if result:
+            return result
+        return self.get_tg_name(sender)
+
+    @staticmethod
+    def get_tg_name(sender: telethon.tl.types.User) -> str:
+        result = [sender.first_name, sender.last_name]
+        while None in result:
+            result.remove(None)
+        if result:
+            return ' '.join(result)
+        if sender.username:
+            return sender.username
+        return sender.id
+
+
+class MiscReactor:
+    def __init__(self, config: dict, db: pgbotlib.dbstuff.DBConn) -> None:
+        self.namegen = NameGenerator(config, db)
+        self.db = db
+
+    def spawn_edited_handler(self,
+                             client: telethon.TelegramClient,
+                             trigger: telethon.events.common.EventBuilder) -> None:
+        @client.on(trigger)
+        async def handle_edited_message(event):
+            sender = await event.get_sender()
+            sender_name = self.namegen.get_name(sender)
+            chat_id = event.message.peer_id
+            await client.send_message(chat_id, f'Я всё видел! {sender_name} совсем охуел, сообщения правит!')