|
|
|
|
import attr
|
|
|
|
|
import typing
|
|
|
|
|
import jinja2
|
|
|
|
|
import emoji_data_python as edp
|
|
|
|
|
from utils import print_user_link
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
Template format:
|
|
|
|
|
|
|
|
|
|
||<| template_name |>||
|
|
|
|
|
//| locale_1 |//
|
|
|
|
|
Text for locale 1
|
|
|
|
|
>>| Reply Keyboard: row 1, button 1;; Row 1, button 2 |<<
|
|
|
|
|
>>| Row 2, button 1 :-: callback_data_for_inline_keyboard |<<
|
|
|
|
|
//| locale_2 |//
|
|
|
|
|
Text for locale 2
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
KeyboardMarkup = typing.List[typing.List[typing.Union[str, typing.Tuple[str, str]]]]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@attr.s(auto_attribs=True)
|
|
|
|
|
class TemplateProvider:
|
|
|
|
|
# {template: {locale: jinja template}}
|
|
|
|
|
templates: typing.Dict[str, typing.Dict[str, jinja2.Template]] = attr.Factory(dict)
|
|
|
|
|
locales: typing.Tuple[str] = ('', 'both', 'en', 'ru')
|
|
|
|
|
|
|
|
|
|
def get_template(self, template_name: str, locale: str) -> jinja2.Template:
|
|
|
|
|
locales = [locale] + list(self.locales)
|
|
|
|
|
template = self.templates[template_name]
|
|
|
|
|
for locale in locales:
|
|
|
|
|
if locale in template.keys():
|
|
|
|
|
return template[locale]
|
|
|
|
|
return next(template.values())
|
|
|
|
|
|
|
|
|
|
def render_template(self, template_name: str, locale: str, **kwargs) -> str:
|
|
|
|
|
return edp.replace_colons(
|
|
|
|
|
self.get_template(template_name, locale)
|
|
|
|
|
.render(print_user_link=print_user_link, **kwargs, **__builtins__)).strip()
|
|
|
|
|
|
|
|
|
|
def add_template(self, template_string: str):
|
|
|
|
|
template_name = template_string.split('||<|')[1].split('|>||')[0].strip()
|
|
|
|
|
template: typing.Dict[str, jinja2.Template] = dict()
|
|
|
|
|
for locale_string in template_string.split('//|')[1:]:
|
|
|
|
|
locale_name: str = locale_string.split('|//')[0].strip()
|
|
|
|
|
locale_text: str = '|//'.join(locale_string.split('|//')[1:]).strip()
|
|
|
|
|
template[locale_name] = jinja2.Environment(loader=jinja2.FileSystemLoader('./')).from_string(locale_text)
|
|
|
|
|
self.templates[template_name] = template
|
|
|
|
|
|
|
|
|
|
def load_file(self, filename: str):
|
|
|
|
|
file_content = open(filename).read()
|
|
|
|
|
for template_string in file_content.split('||<|')[1:]:
|
|
|
|
|
template_string = '||<|' + template_string
|
|
|
|
|
self.add_template(edp.replace_colons(template_string))
|
|
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
|
def separate_text_and_keyboards(text: str) -> typing.Tuple[str, typing.Optional[KeyboardMarkup]]:
|
|
|
|
|
pure_text = text.split('>>|')[0].strip()
|
|
|
|
|
if len(text.split('>>|')) <= 1:
|
|
|
|
|
return pure_text, None
|
|
|
|
|
keyboard_rows = [row.split('|<<')[0].strip() for row in text.split('>>|')[1:]]
|
|
|
|
|
keyboard: KeyboardMarkup = list()
|
|
|
|
|
for row_raw in keyboard_rows:
|
|
|
|
|
row = list()
|
|
|
|
|
for btn in row_raw.split(';;'):
|
|
|
|
|
if ':-:' in btn:
|
|
|
|
|
row.append(tuple(btn.split(':-:')))
|
|
|
|
|
else:
|
|
|
|
|
row.append(btn)
|
|
|
|
|
keyboard.append(row)
|
|
|
|
|
return pure_text, keyboard
|
