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

Allow to get current handler

Browse source
This commit is contained in:
Alex Root Junior 2018-09-21 22:34:59 +03:00
parent 5aff7f7d35
commit 5be3f9d00b
2 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aiogram/dispatcher/handler.py
View file

@ -2,13 +2,14 @@ import inspect
from contextvars import ContextVar
ctx_data = ContextVar('ctx_handler_data')
current_handler = ContextVar('current_handler')
class SkipHandler(BaseException):
class SkipHandler(Exception):
pass
class CancelHandler(BaseException):
class CancelHandler(Exception):
pass
@ -87,9 +88,9 @@ class Handler:
except FilterNotPassed:
continue
else:
ctx_token = current_handler.set(handler)
try:
if self.middleware_key:
# context.set_value('handler', handler)
await self.dispatcher.middleware.trigger(f"process_{self.middleware_key}", args + (data,))
partial_data = _check_spec(handler, data)
response = await handler(*args, **partial_data)
@ -101,6 +102,8 @@ class Handler:
continue
except CancelHandler:
break
finally:
current_handler.reset(ctx_token)
finally:
if self.middleware_key:
await self.dispatcher.middleware.trigger(f"post_process_{self.middleware_key}",