python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor: fixed the typo and bug with a poll_answer

Browse source
This commit is contained in:
NiKuma0 2022-08-23 18:02:36 +03:00
parent 1582d697bd
commit 21c82b27a0
1 changed files with 14 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

24
aiogram/dispatcher/middlewares/user_context.py
View file

@ -43,21 +43,25 @@ class UserContextMiddleware(BaseMiddleware):
"""
Resolve chat and user instance from Update object
"""
if event.inline_query:
if event.inlinee_query.message:
return event.callback_query.message.chat, event.callback_query.from_user
return None, event.callback_query.from_user
parametrs = (
parameters = (
'message', 'edited_message',
'channel_post', 'edited_channel_post',
'chosen_inline_result', 'callback_query',
'shipping_query', 'pre_checkout_query',
'poll_answer', 'my_chat_member',
'chat_member', 'chat_join_request',
'inline_query',
)
for parametr in parametrs:
parametr = getattr(event, parametr, None)
if parametr:
for parameter in parameters:
telegram_obj = getattr(event, parameter, None)
if telegram_obj:
break
return getattr(parametr, 'chat', None), getattr(parametr, 'from_user', None)
if not telegram_obj:
return None, None
if parameter == 'inline_query' and telegram_obj.message:
return parameter.message.chat, parameter.message.from_user
if parameter == 'poll_answer':
return None, parameter.user
return getattr(parameter, 'chat', None), getattr(parameter, 'from_user', None)