mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +00:00
Code Issues Releases Wiki Activity

Rework decorators impl

Browse Source
This commit is contained in:
Dan 2018-04-10 14:54:39 +02:00
parent 7537a27620
commit 9165c7f591
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pyrogram/client/client.py
View File

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