TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-18 13:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small tweaks

Browse Source
This commit is contained in:
Dan 2018-06-18 21:12:04 +02:00
parent e3a667a8fe
commit 09dd715556
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pyrogram/client/dispatcher/dispatcher.py
View File

@ -119,7 +119,6 @@ class Dispatcher:
async def update_worker(self): async def update_worker(self):
while True: while True:
tasks = []
update = await self.updates.get() update = await self.updates.get()
if update is None: if update is None:
@ -130,13 +129,13 @@ class Dispatcher:
chats = {i.id: i for i in update[2]} chats = {i.id: i for i in update[2]}
update = update[0] update = update[0]
tasks.append(self.dispatch(update, users=users, chats=chats, is_raw=True)) await self.dispatch(update, users=users, chats=chats, is_raw=True)
if isinstance(update, Dispatcher.MESSAGE_UPDATES): if isinstance(update, Dispatcher.MESSAGE_UPDATES):
if isinstance(update.message, types.MessageEmpty): if isinstance(update.message, types.MessageEmpty):
continue continue
message = utils.parse_messages( message = await utils.parse_messages(
self.client, self.client,
update.message, update.message,
users, users,
@ -145,7 +144,7 @@ class Dispatcher:
is_edited_message = isinstance(update, Dispatcher.EDIT_MESSAGE_UPDATES) is_edited_message = isinstance(update, Dispatcher.EDIT_MESSAGE_UPDATES)
tasks.append(self.dispatch( await self.dispatch(
pyrogram.Update( pyrogram.Update(
message=((message if message.chat.type != "channel" message=((message if message.chat.type != "channel"
else None) if not is_edited_message else None) if not is_edited_message
@ -160,26 +159,24 @@ class Dispatcher:
else None) if is_edited_message else None) if is_edited_message
else None) else None)
) )
)) )
elif isinstance(update, types.UpdateBotCallbackQuery): elif isinstance(update, types.UpdateBotCallbackQuery):
tasks.append(self.dispatch( await self.dispatch(
pyrogram.Update( pyrogram.Update(
callback_query=utils.parse_callback_query( callback_query=await utils.parse_callback_query(
self.client, update, users self.client, update, users
) )
) )
)) )
elif isinstance(update, types.UpdateInlineBotCallbackQuery): elif isinstance(update, types.UpdateInlineBotCallbackQuery):
tasks.append(self.dispatch( await self.dispatch(
pyrogram.Update( pyrogram.Update(
callback_query=utils.parse_inline_callback_query( callback_query=await utils.parse_inline_callback_query(
update, users update, users
) )
) )
)) )
else: else:
continue continue
await asyncio.gather(*tasks)
except Exception as e: except Exception as e:
log.error(e, exc_info=True) log.error(e, exc_info=True)