Add startup handler

This commit is contained in:
BarsTiger
2023-10-07 22:33:26 +03:00
parent cdf83d2be1
commit e25b61a8ea
17 changed files with 79 additions and 1 deletions

View File

@@ -1 +1,21 @@
pass
from rich import print
async def runner():
from .common import dp, bot
from . import handlers
dp.include_router(
handlers.router
)
await bot.delete_webhook(drop_pending_updates=True)
await dp.start_polling(bot)
def main():
import asyncio
print('Starting...')
asyncio.run(runner())

View File

@@ -0,0 +1,4 @@
from . import main
if __name__ == '__main__':
main()

View File

7
anymusicbot/common.py Normal file
View File

@@ -0,0 +1,7 @@
from aiogram import Bot, Dispatcher
from .utils.config import config
bot = Bot(token=config.telegram.bot_token)
dp = Dispatcher()
__all__ = ['bot', 'dp', 'config']

View File

View File

@@ -0,0 +1,9 @@
from aiogram import Router
from . import initialize
router = Router()
router.include_routers(
initialize.router,
)

View File

@@ -0,0 +1 @@
from .initializer import router

View File

@@ -0,0 +1,10 @@
from aiogram import Router, Bot
from rich import print
router = Router()
@router.startup()
async def startup(bot: Bot):
print(f'[green]Started as[/] @{(await bot.me()).username}')

View File

View File

View File

View File

View File

View File

@@ -0,0 +1,4 @@
from ._config import Config
config = Config()

View File

@@ -0,0 +1,22 @@
import tomllib
class Config(dict):
def __init__(self, _config: dict = None):
try:
if _config is None:
config = tomllib.load(open('config.toml', 'rb'))
super().__init__(**config)
else:
super().__init__(**_config)
except FileNotFoundError:
super().__init__()
def __getattr__(self, item):
return (
self.get(item)
if type(self.get(item)) is not dict
else Config(self.get(item))
)

View File

@@ -8,6 +8,7 @@ readme = "README.md"
[tool.poetry.dependencies]
python = "^3.11"
aiogram = "^3.1.1"
rich = "^13.6.0"
[build-system]