mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 13:27:47 +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.dispatcher = Dispatcher(self, workers)
self.update_handler = None self.update_handler = None
def on(self, handler, group, filters=None): def on_message(self, filters=None, group: int = 0):
def decorator(func): def decorator(func):
self.add_handler(handler(func, filters), group) self.add_handler(pyrogram.MessageHandler(func, filters), group)
return func return func
return decorator 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): 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): def add_handler(self, handler: Handler, group: int = 0):
self.dispatcher.add_handler(handler, group) self.dispatcher.add_handler(handler, group)