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

Added new on_ registrator for use as decorator when register on_startup, on_shutdown: on_startup_polling, on_startup_webhook, on_shutdown_polling, on_shutdown_webhook

Browse source
This commit is contained in:
Arwichok 2019-09-07 09:36:04 +03:00
parent 4984c2313a
commit 895e24727d
1 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

32
aiogram/utils/executor.py
View file

@ -186,6 +186,22 @@ class Executor:
if webhook:
self._on_startup_webhook.append(callback)
def on_startup_polling(self, callback: callable):
"""
Register a callback for the startup with polling
:param callback:
"""
self.on_startup(callback, webhook=False)
def on_startup_webhook(self, callback: callable):
"""
Register a callback for the startup with webhook
:param callback:
"""
self.on_startup(callback, polling=False)
def on_shutdown(self, callback: callable, polling=True, webhook=True):
"""
Register a callback for the shutdown process
@ -209,6 +225,22 @@ class Executor:
if webhook:
self._on_shutdown_webhook.append(callback)
def on_shutdown_polling(self, callback: callable):
"""
Register a callback for the shutdown with pollign
:param callback:
"""
self.on_shutdown(callback, webhook=False)
def on_shutdown_webhook(self, callback: callable):
"""
Register a callback for the shutdown with webhook
:param callback:
"""
self.on_shutdown(callback, polling=False)
def _check_frozen(self):
if self.frozen:
raise RuntimeError('Executor is frozen!')