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

✨ Ability to update data dict from middlewares returns

Browse source
This commit is contained in:
Mykola Solodukha 2020-10-16 20:00:19 +03:00
parent 4e82865f45
commit 0d3b0c4e43
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
aiogram/dispatcher/handler.py
View file

@ -98,7 +98,11 @@ class Handler:
if self.middleware_key:
try:
await self.dispatcher.middleware.trigger(f"pre_process_{self.middleware_key}", args + (data,))
data.update(
await self.dispatcher.middleware.trigger(
f"pre_process_{self.middleware_key}", args + (data,)
) or {}
)
except CancelHandler: # Allow to cancel current event
return results
@ -112,7 +116,11 @@ class Handler:
ctx_token = current_handler.set(handler_obj.handler)
try:
if self.middleware_key:
await self.dispatcher.middleware.trigger(f"process_{self.middleware_key}", args + (data,))
data.update(
await self.dispatcher.middleware.trigger(
f"process_{self.middleware_key}", args + (data,)
) or {}
)
partial_data = _check_spec(handler_obj.spec, data)
response = await handler_obj.handler(*args, **partial_data)
if response is not None: