Merge pull request #300 from ffmemes/codex/localize-source-admin-ru

Localize source admin controls

Author: ohld

src/tgbot/handlers/moderator/meme_source.py

@@ -1,11 +1,14 @@
 import re
 from dataclasses import dataclass
+from html import escape as html_escape
+from typing import Any
 
 from telegram import Message, Update
 from telegram.ext import (
     ContextTypes,
 )
 
+from src import localizer
 from src.flows.parsers.tg import parse_telegram_source
 from src.flows.parsers.vk import parse_vk_source
 from src.storage.constants import MemeSourceStatus, MemeSourceType
@@ -33,6 +36,11 @@
 _MEME_SOURCE_LINK_RE = re.compile(MEME_SOURCE_LINK_REGEXP)
 
 
+def _t(key: str, lang: str | None, **kwargs: object) -> str:
+    text = localizer.t(key, lang)
+    return text.format(**kwargs) if kwargs else text
+
+
 @dataclass(frozen=True)
 class MemeSourceLink:
     url: str
@@ -78,13 +86,15 @@ def parse_meme_source_status_callback_data(data: str) -> tuple[int, str]:
 
 
 async def handle_meme_source_link(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
-    if await get_moderator_user_info(update.effective_user.id) is None:
-        await update.message.reply_text("Only moderators can manage meme sources.")
+    moderator = await get_moderator_user_info(update.effective_user.id)
+    lang = _get_moderator_lang(update, moderator)
+    if moderator is None:
+        await update.message.reply_text(_t("moderator.meme_source.only_moderators_manage", lang))
         return
 
     link = parse_meme_source_link(update.message.text)
     if link is None:
-        await update.message.reply_text("Unsupported meme source")
+        await update.message.reply_text(_t("moderator.meme_source.unsupported_source", lang))
         return
 
     meme_source = await get_or_create_meme_source(
@@ -94,17 +104,19 @@ async def handle_meme_source_link(update: Update, context: ContextTypes.DEFAULT_
         added_by=update.effective_user.id,
     )
 
-    await meme_source_admin_pipeline(meme_source, update)
+    await meme_source_admin_pipeline(meme_source, update, lang)
 
 
 async def handle_meme_source_language_selection(
     update: Update, context: ContextTypes.DEFAULT_TYPE
 ) -> None:
     user_id = update.effective_user.id
-    if await get_moderator_user_info(user_id) is None:
+    moderator = await get_moderator_user_info(user_id)
+    lang = _get_moderator_lang(update, moderator)
+    if moderator is None:
         await update.callback_query.answer(
-            "🤷‍♀️ Only moderators can change meme source language 🤷‍♂️"
-        )  # noqa: E501
+            _t("moderator.meme_source.only_moderators_language", lang)
+        )
         return
 
     args = update.callback_query.data.split(":")
@@ -118,30 +130,34 @@ async def handle_meme_source_language_selection(
             trigger_parse=False,
         )
     except MemeSourceNotFoundError:
-        await update.callback_query.answer("Meme source not found")
+        await update.callback_query.answer(_t("moderator.meme_source.not_found", lang))
         return
 
     await log(
         f"ℹ️ MemeSource ${meme_source_id}: set_lang={lang_code} (by {user_id})",
         context.bot,
     )
 
-    await update.callback_query.answer(f"Meme source lang is {lang_code} now")
-    await meme_source_admin_pipeline(result["source"], update)
+    await update.callback_query.answer(
+        _t("moderator.meme_source.language_updated", lang, language=lang_code)
+    )
+    await meme_source_admin_pipeline(result["source"], update, lang)
 
 
 async def handle_meme_source_change_status(
     update: Update, context: ContextTypes.DEFAULT_TYPE
 ) -> None:
     user_id = update.effective_user.id
-    if await get_moderator_user_info(user_id) is None:
-        await update.callback_query.answer("🤷‍♀️ Only moderators can change meme source status 🤷‍♂️")  # noqa: E501
+    moderator = await get_moderator_user_info(user_id)
+    lang = _get_moderator_lang(update, moderator)
+    if moderator is None:
+        await update.callback_query.answer(_t("moderator.meme_source.only_moderators_status", lang))
         return
 
     try:
         meme_source_id, status = parse_meme_source_status_callback_data(update.callback_query.data)
     except (IndexError, KeyError, ValueError):
-        await update.callback_query.answer("Invalid meme source status action")
+        await update.callback_query.answer(_t("moderator.meme_source.invalid_status_action", lang))
         return
 
     try:
@@ -154,24 +170,37 @@ async def handle_meme_source_change_status(
             trigger_parse=False,
         )
     except MemeSourceNotFoundError:
-        await update.callback_query.answer(f"Meme source {meme_source_id} not found")
+        await update.callback_query.answer(
+            _t("moderator.meme_source.not_found_by_id", lang, source_id=meme_source_id)
+        )
         return
     except ValueError as e:
         await update.callback_query.answer(str(e)[:180])
         return
 
     if result["unsnoozed_count"]:
         await update.effective_chat.send_message(
-            f"Unsnoozed {result['unsnoozed_count']} memes of {meme_source_id}"
+            _t(
+                "moderator.meme_source.unsnoozed_memes",
+                lang,
+                count=result["unsnoozed_count"],
+                source_id=meme_source_id,
+            )
         )
 
     await log(
         f"ℹ️ MemeSource ${meme_source_id}: set_status={status} (by {user_id})",
         context.bot,
     )
 
-    await update.callback_query.answer(f"Meme source status is {status} now")
-    await meme_source_admin_pipeline(result["source"], update)
+    await update.callback_query.answer(
+        _t(
+            "moderator.meme_source.status_updated",
+            lang,
+            status=_status_label(status, lang),
+        )
+    )
+    await meme_source_admin_pipeline(result["source"], update, lang)
 
     meme_source = result["source"]
     if status == MemeSourceStatus.PARSING_ENABLED:  # trigger parsing
@@ -183,57 +212,127 @@ async def handle_meme_source_change_status(
 
     if result["snoozed_count"]:
         await update.effective_chat.send_message(
-            f"Snoozed {result['snoozed_count']} memes of {meme_source_id}"
+            _t(
+                "moderator.meme_source.snoozed_memes",
+                lang,
+                count=result["snoozed_count"],
+                source_id=meme_source_id,
+            )
         )
 
 
-def _get_meme_source_info(meme_source: dict) -> str:
-    return f"""
-id: {meme_source["id"]}
-url: {meme_source["url"]}
-type: {meme_source["type"]}
-language: {meme_source["language_code"]}
-added by: {meme_source["added_by"]}
-<b>status</b>: {meme_source["status"]}
-    """
+def _get_moderator_lang(
+    update: Update,
+    moderator_info: dict[str, Any] | None = None,
+) -> str | None:
+    if moderator_info and moderator_info.get("interface_lang"):
+        return str(moderator_info["interface_lang"])
+    if update.effective_user and getattr(update.effective_user, "language_code", None):
+        return update.effective_user.language_code
+    return "ru"
+
+
+def _html(value: object) -> str:
+    if value is None:
+        return "—"
+    return html_escape(str(value), quote=False)
+
+
+def _source_type_label(source_type: object) -> str:
+    value = getattr(source_type, "value", source_type)
+    if value == MemeSourceType.TELEGRAM.value:
+        return "Telegram"
+    if value == MemeSourceType.INSTAGRAM.value:
+        return "Instagram"
+    if value == MemeSourceType.VK.value:
+        return "VK"
+    return _html(value)
+
+
+def _status_label(status: object, lang: str | None) -> str:
+    value = getattr(status, "value", status)
+    try:
+        return localizer.t(f"moderator.meme_source.status.{value}", lang)
+    except KeyError:
+        return _html(value)
+
 
-    # Column("nlikes", Integer, nullable=False, server_default="0"),
-    # Column("ndislikes", Integer, nullable=False, server_default="0"),
-    # Column("nmemes_sent_events", Integer, nullable=False, server_default="0"),
-    # Column("nmemes_parsed", Integer, nullable=False, server_default="0"),
-    # Column("nmemes_sent", Integer, nullable=False, server_default="0"),
-    # Column("latest_meme_age", Integer, nullable=False, server_default="0"),
+def _format_int(value: object) -> str:
+    try:
+        return f"{int(value):,}".replace(",", " ")
+    except (TypeError, ValueError):
+        return "—"
 
 
-def _get_meme_source_stats_info(meme_source_stats: dict) -> str:
-    return f"""
-likes: {meme_source_stats["nlikes"]}
-dislikes: {meme_source_stats["ndislikes"]}
-memes sent events: {meme_source_stats["nmemes_sent_events"]}
-memes parsed: {meme_source_stats["nmemes_parsed"]}
-memes sent: {meme_source_stats["nmemes_sent"]}
-latest meme age: {meme_source_stats["latest_meme_age"]}
-    """
+def _format_latest_age(value: object, lang: str | None) -> str:
+    try:
+        days = int(value)
+    except (TypeError, ValueError):
+        return "—"
+
+    if days <= 0:
+        return localizer.t("moderator.meme_source.today", lang)
+
+    return _t("moderator.meme_source.days_short", lang, days=days)
+
+
+def _get_meme_source_info(meme_source: dict, lang: str | None) -> str:
+    source_type = _source_type_label(meme_source["type"])
+    language = _html(meme_source["language_code"])
+    added_by = _html(meme_source["added_by"])
+    status = _status_label(meme_source["status"], lang)
+
+    return _t(
+        "moderator.meme_source.card",
+        lang,
+        id=_html(meme_source["id"]),
+        type=source_type,
+        language=language,
+        url=_html(meme_source["url"]),
+        status=status,
+        added_by=added_by,
+    )
+
+
+def _get_meme_source_stats_info(meme_source_stats: dict, lang: str | None) -> str:
+    return _t(
+        "moderator.meme_source.stats",
+        lang,
+        likes=_format_int(meme_source_stats["nlikes"]),
+        dislikes=_format_int(meme_source_stats["ndislikes"]),
+        sent_events=_format_int(meme_source_stats["nmemes_sent_events"]),
+        parsed=_format_int(meme_source_stats["nmemes_parsed"]),
+        sent=_format_int(meme_source_stats["nmemes_sent"]),
+        latest_age=_format_latest_age(meme_source_stats["latest_meme_age"], lang),
+    )
 
 
 async def meme_source_admin_pipeline(
     meme_source: dict,
     update: Update,
+    lang: str | None = None,
 ) -> Message:
-    ms_info = _get_meme_source_info(meme_source)
+    if lang is None:
+        lang = _get_moderator_lang(update)
+
+    ms_info = _get_meme_source_info(meme_source, lang)
     ms_stats = await get_meme_source_stats_by_id(meme_source["id"])
     if ms_stats:
-        ms_info += _get_meme_source_stats_info(ms_stats)
+        ms_info += "\n\n" + _get_meme_source_stats_info(ms_stats, lang)
 
     if meme_source["language_code"] is None:
         return await send_or_edit(
             update,
-            text=f"""{ms_info}\nPlease select a language for {meme_source["url"]}""",
+            text=f"{ms_info}\n\n{localizer.t('moderator.meme_source.select_language', lang)}",
             reply_markup=meme_source_language_selection_keyboard(meme_source_id=meme_source["id"]),
         )
 
     return await send_or_edit(
         update,
         text=ms_info,
-        reply_markup=meme_source_change_status_keyboard(meme_source["id"], meme_source["status"]),
+        reply_markup=meme_source_change_status_keyboard(
+            meme_source["id"],
+            meme_source["status"],
+            lang,
+        ),
     )

src/tgbot/senders/keyboards.py

@@ -2,6 +2,7 @@
 
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup
 
+from src import localizer
 from src.storage.constants import (
     MemeSourceStatus,
 )
@@ -126,12 +127,13 @@ def meme_source_language_selection_keyboard(meme_source_id: int):
 def meme_source_change_status_keyboard(
     meme_source_id: int,
     current_status: MemeSourceStatus | str | None = None,
+    lang: str | None = "ru",
 ):
     return InlineKeyboardMarkup(
         [
             [
                 InlineKeyboardButton(
-                    f"➡️ {status.value}",
+                    localizer.t(f"moderator.meme_source.action.{status.value}", lang),
                     callback_data=MEME_SOURCE_SET_STATUS_PATTERN.format(
                         meme_source_id=meme_source_id,
                         status=status.value,