mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-09-10 11:15:27 +00:00
Code Issues Releases Wiki Activity

Add handler for raw mtproto updates

Browse Source
This commit is contained in:
Dan
2018-04-08 13:20:31 +02:00
parent 66d03848de
commit 1a7ab62ed9
5 changed files with 53 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pyrogram/client/client.py
View File

@@ -189,13 +189,19 @@ class Client:
self.dispatcher = Dispatcher(self, workers)
self.update_handler = None
def on_message(self, group: int = 0):
def on(self, handler, group):
def decorator(f):
self.add_handler(pyrogram.MessageHandler(f), group)
self.add_handler(handler(f), group)
return f
return decorator
def on_message(self, group: int = 0):
return self.on(pyrogram.MessageHandler, group)
def on_raw_update(self, group: int = 0):
return self.on(pyrogram.RawUpdateHandler, group)
def add_handler(self, handler: Handler, group: int = 0):
self.dispatcher.add_handler(handler, group)