EmojiSaver/main.py at main · bohd4nx/EmojiSaver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import asyncio

from aiogram import Bot, Dispatcher
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode
from aiogram.types import BotCommand
from aiogram_i18n import I18nMiddleware
from aiogram_i18n.cores.fluent_compile_core import FluentCompileCore

from bot.commands import help_router, start_router
from bot.core import config, logger, setup_logging
from bot.core.constants import DEFAULT_LOCALE
from bot.database import close_db, init_db
from bot.handlers import emoji, errors, packs, stickers
from bot.handlers.errors import setup_error_handlers
from bot.middlewares import (
    DatabaseMiddleware,
    LocaleMiddleware,
    RateLimitMiddleware,
)


async def main() -> None:
    setup_logging()
    await init_db()

    bot = Bot(
        token=config.BOT_TOKEN,
        default=DefaultBotProperties(
            parse_mode=ParseMode.HTML,
            link_preview_is_disabled=True,
        ),
    )

    _commands = {
        "ru": [
            BotCommand(command="start", description="🏠 Главное меню"),
            BotCommand(command="help", description="❓ Помощь"),
        ],
        "en": [
            BotCommand(command="start", description="🏠 Main menu"),
            BotCommand(command="help", description="❓ Help"),
        ],
    }
    for lang, cmds in _commands.items():
        await bot.set_my_commands(cmds, language_code=lang)

    _descriptions = {
        "ru": (
            "Скачивайте и конвертируйте стикеры и анимированные эмодзи Telegram в форматы TGS, JSON, Lottie и PNG.\n\n"
            "Извлекайте кастомные эмодзи, конвертируйте стикеры и скачивайте целые паки."
        ),
        "en": (
            "Download and convert Telegram animated emoji & stickers to TGS, JSON, Lottie, and PNG formats.\n\n"
            "Extract custom emoji, convert stickers, and download entire packs."
        ),
    }
    for lang, desc in _descriptions.items():
        await bot.set_my_description(desc, language_code=lang)

    _short_descriptions = {
        "ru": "🔄 Конвертируйте и скачивайте анимированные эмодзи и стикеры Telegram",
        "en": "🔄  Download and convert Telegram animated emoji & stickers to editable formats",
    }
    for lang, short_desc in _short_descriptions.items():
        await bot.set_my_short_description(short_desc, language_code=lang)

    i18n_core = FluentCompileCore(path="locales/{locale}")
    await i18n_core.startup()

    dp = Dispatcher()
    dp.include_routers(
        start_router,
        help_router,
        packs.router,
        emoji.router,
        stickers.router,
        errors.router,
    )

    i18n = I18nMiddleware(core=i18n_core, default_locale=DEFAULT_LOCALE)
    i18n.setup(dispatcher=dp)

    dp.update.middleware(DatabaseMiddleware())
    dp.update.middleware(LocaleMiddleware())
    dp.update.middleware(RateLimitMiddleware())

    setup_error_handlers(dp)

    try:
        await dp.start_polling(
            bot,
            polling_timeout=30,
            handle_as_tasks=True,
            tasks_concurrency_limit=100,
            close_bot_session=True,
        )
    finally:
        await i18n.core.shutdown()
        await close_db()


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except (KeyboardInterrupt, SystemExit):
        pass
    except Exception as exc:
        logger.exception("Unexpected error: %s", exc)