template_extensions.py

""" Регистрация дополнительных тегов и настройки шаблонизатора """
from __future__ import annotations

import typing

import aiogram.types

import progressbar
import resources
import settings
from template.dev import *
from template_for_aiogram.types import *
from template_for_aiogram.scopes import *

if typing.TYPE_CHECKING:
    from apps.botpiska.models.subscription import Subscription


PROGRESSBAR_DEFAULT_STYLE = progressbar.styles.basic_green
PROGRESSBAR_STYLES = {
    'red': progressbar.styles.basic_red,
    'orange': progressbar.styles.basic_orange,
    'yellow': progressbar.styles.basic_yellow,
    'green': progressbar.styles.basic_green,
    'blue': progressbar.styles.basic_blue,
    'purple': progressbar.styles.basic_purple,
    'brown': progressbar.styles.basic_brown,
    'black': progressbar.styles.basic_black,
    'rainbow': progressbar.styles.basic_rainbow,
}

# TEMPLATE GLOBALS --------------------------------------------------

set_global_context({
    'support': settings.SUPPORT_CHAT_ID,
    'tech_support': settings.TECH_SUPPORT_CHAT_ID,
    'tg': aiogram.types,
    'botpiska_bot': settings.BOT_NAME
})


# SPECIFIERS --------------------------------------------------------

def res_extract_func(value, context, _) -> aiogram.types.InputFile | str:
    """ Воспринимает переданное значение в качестве индекса файла.
        Производит поиск cache->database->filesystem """
    return resources.resource(value.format_map(context))


specifiers.update(rs=res_extract_func)


# CONVERTERS --------------------------------------------------------

def to_style(name: str) -> progressbar.Style:
    return PROGRESSBAR_STYLES.get(name, PROGRESSBAR_DEFAULT_STYLE)


# TAGS --------------------------------------------------------------

@register([MESSAGE, ELEMENT], name='progressbar')
def _progressbar(_, *, steps: int, of: int, width: int = None, style: ConvertBy[to_style] = None) -> str:
    """ Progressbar """
    return progressbar.progressbar(steps / of, width, style)


# Python sometimes messes up annotations and turns them into strings,
# fixed by providing annotations explicitly
@register([MESSAGE, ELEMENT], annotations={'user': int, 'display': str})
def chat(_, *, user=None, display='Пользователь') -> Text:
    """ Ссылка на чат с пользователем, оформленная как его имя """

    if user is not None:
        return f'<a href="tg://user?id={user}">{display}</a>'
    return display


@register([MESSAGE, ELEMENT], name='chat-current')
def chat_current(_) -> Text:
    """ Ссылка на чат с текущим пользователем, оформленная как его имя """

    if user := typing.cast(typing.Optional[aiogram.types.User], aiogram.types.User.get_current()):
        return chat(_, user=user.id, display=user.first_name)
    return chat(_)


@register([INLINE_KEYBOARD, INLINE_KEYBOARD_ROW], name='buy-button')
def buy_button(_, *, subscription: 'Subscription', discount: int = None) -> aiogram.types.InlineKeyboardButton:
    """ Кнопка для покупки подписки """

    text = f'{subscription.short_title} — {subscription.monthly_price:.2f}₽/мес'

    if discount is not None:
        text += f' ( скидка {discount}% )'

    if subscription.is_featured:
        text = f'🔶 {text} 🔶'

    return aiogram.types.InlineKeyboardButton(
        callback_data=f'buy:{subscription.id}',
        text=text
    )