yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-01 05:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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)