Add new example usage code for auto-replies (#249)

2024-09-21 03:08:49 +03:00 · 2017-09-18 19:09:58 +01:00 · 2017-09-18 19:09:58 +01:00 · 879621ab70
commit 879621ab70
parent aea95a398b
1 changed files with 113 additions and 0 deletions
--- a/telethon_examples/auto_reply.py
+++ b/telethon_examples/auto_reply.py
@ -0,0 +1,113 @@
+#!/usr/bin/env python3
+# disclaimer: you should not actually use this. it can be quite spammy.
+from telethon import TelegramClient
+from telethon.errors import SessionPasswordNeededError
+from getpass import getpass
+from telethon.tl.types import InputPeerUser,InputPeerChannel
+from telethon.tl.types import Updates
+from telethon.tl.types import UpdateNewChannelMessage,UpdateNewMessage
+from telethon.tl.functions.messages import SendMessageRequest,EditMessageRequest
+from telethon.tl.types import MessageService
+from nltk.tokenize import word_tokenize
+from os import environ
+from time import sleep
+
+CHANNELS = {}
+CHANNELNAMES = {}
+USERS = {}
+EMACS_BLACKLIST = [1058260578, # si @linux_group
+        123456789]
+REACTS = {'emacs':'Needs more vim.',
+        'chrome':'Needs more firefox.',
+}
+
+class NeedsMore(TelegramClient):
+    def __init__(self):
+        settings = {'api_id':int(environ['TG_API_ID']),
+                'api_hash':environ['TG_API_HASH'],
+                'user_phone':environ['TG_PHONE'],
+                'session_name':'needsmore'}
+        super().__init__(
+            settings.get('session_name','session1'),
+            settings['api_id'],
+            settings['api_hash'],
+            proxy=None,
+            process_updates=True)
+
+        user_phone = settings['user_phone']
+
+        print('INFO: Connecting to Telegram Servers...', end='', flush=True)
+        self.connect()
+        print('Done!')
+
+        if not self.is_user_authorized():
+            print('INFO: Unauthorized user')
+            self.send_code_request(user_phone)
+            code_ok = False
+            while not code_ok:
+                code = input('Enter the auth code: ')
+                try:
+                    code_ok = self.sign_in(user_phone, code)
+                except SessionPasswordNeededError:
+                    pw = getpass('Two step verification enabled. Please enter your password: ')
+                    self.sign_in(password=pw)
+            print('INFO: Client initialized succesfully!')
+
+    def run(self):
+        # Listen for updates
+        while True:
+            update = self.updates.poll() # This will block until an update is available
+            triggers = []
+            if isinstance(update, Updates):
+                for x in update.updates:
+                    if not isinstance(x,UpdateNewChannelMessage): continue
+                    if isinstance(x.message,MessageService): continue
+                    # We're only interested in messages to supergroups
+                    words = word_tokenize(x.message.message.lower())
+                    # Avoid matching 'emacs' in 'spacemacs' and similar
+                    if 'emacs' in words and x.message.to_id.channel_id not in EMACS_BLACKLIST:
+                        triggers.append(('emacs',x.message))
+                    if 'chrome' in words:
+                        triggers.append(('chrome',x.message))
+                    if 'x files theme' == x.message.message.lower() and x.message.out:
+                        # Automatically reply to yourself saying 'x files theme' with the audio
+                        msg = x.message
+                        chan = InputPeerChannel(msg.to_id.channel_id,CHANNELS[msg.to_id.channel_id])
+                        self.send_voice_note(chan,'xfiles.m4a',reply_to=msg.id)
+                        sleep(1)
+                    if '.shrug' in x.message.message.lower() and x.message.out:
+                        # Automatically replace '.shrug' in any message you
+                        # send to a supergroup with the shrug emoticon
+                        msg = x.message
+                        chan = InputPeerChannel(msg.to_id.channel_id,CHANNELS[msg.to_id.channel_id])
+                        self(EditMessageRequest(chan,msg.id,
+                            message=msg.message.replace('.shrug','¯\_(ツ)_/¯')))
+                        sleep(1)
+
+f               for trigger in triggers:
+                    msg = trigger[1]
+                    chan = InputPeerChannel(msg.to_id.channel_id,CHANNELS[msg.to_id.channel_id])
+                    log_chat = InputPeerUser(user_id=123456789,access_hash=987654321234567890)
+                    self.send_message(log_chat,"{} said {} in {}. Sending react {}".format(
+                        msg.from_id,msg.message,CHANNELNAMES[msg.to_id.channel_id],REACTS[trigger[0]][:20]))
+                    react = '>{}\n{}'.format(trigger[0],REACTS[trigger[0]])
+                    self.invoke(SendMessageRequest(chan,react,reply_to_msg_id=msg.id))
+                    sleep(1)
+
+if __name__ == "__main__":
+    #TODO: this block could be moved to __init__
+    # You can create these text files using https://github.com/LonamiWebs/Telethon/wiki/Retrieving-all-dialogs
+    with open('channels.txt','r') as f:
+        # Format: channel_id access_hash #Channel Name
+        lines = f.readlines()
+        chans = [l.split(' #',1)[0].split(' ') for l in lines]
+        CHANNELS = {int(c[0]):int(c[1]) for c in chans} # id:hash
+        CHANNELNAMES = {int(l.split()[0]):l.split('#',1)[1].strip() for l in lines} #id:name
+    with open('users','r') as f:
+        # Format: [user_id, access_hash, 'username', 'Firstname Lastname']
+        lines = f.readlines()
+        uss = [l.strip()[1:-1].split(',') for l in lines]
+        USERS = {int(user[0]):int(user[1]) for user in uss} # id:hash
+
+    needsmore = NeedsMore()
+    needsmore.run()