Refuse gemini-only media early, cache build version

lumen_commands.py

@@ -552,6 +552,10 @@ async def cmd_stats(message: Message) -> None:
     # Счётчики — по датам Google (сброс в _reset_quota_if_new_day проверяем перед отрисовкой: фоновый тик мог не успеть, а used копились через рестарты).
     bot._reset_quota_if_new_day()
 
+    # Вебхук опрашиваем параллельно со сборкой текста: иначе висящий API
+    # держал бы всю команду до 10с уже после готовой статистики.
+    _webhook_task = asyncio.ensure_future(_webhook_info_text())
+
     total_chats = len(bot.chat_state)
     # "Активные" — с живой активностью за сутки по настенным часам (monotonic
     # сбрасывался рестартом и считал активными всех). Без метки — неактивен.
@@ -608,7 +612,7 @@ async def cmd_stats(message: Message) -> None:
     proxy_line = bot._tg_proxy_breaker.status_text()
 
     quota_day = bot.GLOBAL_QUOTA.get("quota_day") or "—"
-    webhook_text = await _webhook_info_text()
+    webhook_text = await _webhook_task
 
     text = (
         f"<b>Статистика Lumen</b>\n"
@@ -644,12 +648,20 @@ async def cmd_stats(message: Message) -> None:
     await bot._tg_call(message.reply, text, parse_mode=ParseMode.HTML)
 
 
+_BUILD_VERSION_CACHED: str | None = None
+
+
 def _build_version() -> str:
     """Короткий хеш сборки для /stats: env сборки, иначе git, иначе unknown."""
+    global _BUILD_VERSION_CACHED
+    # git-вызов блокирует loop до 5с — считаем один раз за жизнь процесса.
+    if _BUILD_VERSION_CACHED is not None:
+        return _BUILD_VERSION_CACHED
     for env_name in ("LUMEN_BUILD_VERSION", "BUILD_VERSION", "GIT_COMMIT", "COMMIT_SHA"):
         raw = os.getenv(env_name, "").strip()
         if raw:
-            return raw[:12]
+            _BUILD_VERSION_CACHED = raw[:12]
+            return _BUILD_VERSION_CACHED
     try:
         proc = subprocess.run(
             ["git", "rev-parse", "--short", "HEAD"],
@@ -657,9 +669,11 @@ def _build_version() -> str:
         )
         sha = (proc.stdout or "").strip()
         if proc.returncode == 0 and sha:
-            return sha[:12]
+            _BUILD_VERSION_CACHED = sha[:12]
+            return _BUILD_VERSION_CACHED
     except Exception:
         pass
+    _BUILD_VERSION_CACHED = "unknown"
     return "unknown"
 
 

lumen_message_core.py

@@ -304,6 +304,31 @@ def _strip_trigger_content(clean_prompt: str, trigger: str, message: Message) ->
     return content
 
 
+def _direct_attachment_needs_gemini(message: Message) -> bool:
+    """Видео/аудио/документ (не картинка) во вложении или реплае — такое поздно
+    отдавать только Gemini, скачивание заранее бесполезно при исчерпанном лимите."""
+    for msg in (message, getattr(message, "reply_to_message", None)):
+        if msg is None:
+            continue
+        # Имя атрибута уже говорит о категории без скачивания (и без опоры на
+        # имя класса — в тестах лежат даблы, у aiogram свои типы).
+        for attr in ("video", "video_note", "animation", "voice", "audio"):
+            if getattr(msg, attr, None):
+                return True
+        src = _msg_media_source(msg)
+        if src is None:
+            continue
+        if type(src).__name__ in ("Video", "VideoNote", "Animation", "Voice", "Audio"):
+            return True
+        _, mime, _ = _media_file_id_and_mime(src)
+        m = (mime or "").lower()
+        if m.startswith(("video/", "audio/")):
+            return True
+        if type(src).__name__ == "Document" and not m.startswith("image/"):
+            return True
+    return False
+
+
 async def _handle_message_core(message: Message, extra_media: list[tuple[bytes, str]] | None = None) -> None:
     import bot
     t = message.text or message.caption or ""
@@ -336,6 +361,17 @@ async def _handle_message_core(message: Message, extra_media: list[tuple[bytes,
         ))
         return
 
+    # Видео/аудио при исчерпанном Gemini отказываем до скачивания: поздняя
+    # проверка после маршрута уже скачала бы файл и пожгла транскрибацию.
+    # Картинки пропускаем — их отдаст Groq/OpenRouter через поздний фолбэк.
+    if bot._user_daily_gemini_exhausted(uid_early, _early_entry) and _direct_attachment_needs_gemini(message):
+        bot._record_stats_event("daily_limit_denials")
+        await bot._safe_reply(message, bot._t(
+            message.chat.id, "user_daily_gemini",
+            used=(_early_entry or {}).get("gemini", 0), limit=bot._user_daily_limit(uid_early, "gemini", _early_entry),
+        ))
+        return
+
     state = bot.get_state(message.chat.id)
 
     # Проверка на ссылки загрузки (TikTok — сразу всегда, даже в группах без упоминания)

tests/test_bot_commands.py

@@ -286,8 +286,31 @@ def test_cmd_stats_truncation_keeps_html_valid(monkeypatch):
         lcs.chat_state.pop(chat_id, None)
 
 
-def test_match_trigger_prefix_finds_draw_trigger():
-    assert bot._match_trigger_prefix("нарисуй кота на пляже", bot.DRAW_TRIGGER_PREFIXES) == "нарисуй"
+def test_build_version_caches_git_call(monkeypatch):
+    # Защищает кеш версии: git-вызов блокирует loop до 5с, без кеша каждый
+    # /stats стопает бота. Регрессия — прямой subprocess при каждом вызове.
+    import lumen_commands as lc
+    monkeypatch.setattr(lc, "_BUILD_VERSION_CACHED", None)
+    calls = []
+
+    class _Proc:
+        returncode = 0
+        stdout = "abc1234\n"
+
+    def fake_run(*args, **kwargs):
+        calls.append(1)
+        return _Proc()
+
+    monkeypatch.setattr(lc.subprocess, "run", fake_run)
+    try:
+        assert lc._build_version() == "abc1234"
+        assert lc._build_version() == "abc1234"
+        assert len(calls) == 1
+    finally:
+        monkeypatch.setattr(lc, "_BUILD_VERSION_CACHED", None)
+
+
+def test_match_trigger_prefix_finds_draw_trigger():    assert bot._match_trigger_prefix("нарисуй кота на пляже", bot.DRAW_TRIGGER_PREFIXES) == "нарисуй"
 
 
 def test_match_trigger_prefix_finds_tts_trigger():

tests/test_bot_state.py

@@ -2874,3 +2874,36 @@ def test_core_refuses_youtube_when_gemini_exhausted(rate_guard_setup, monkeypatc
     finally:
         bot.GLOBAL_QUOTA.get("user_daily", {}).pop("456", None)
         bot.chat_state.pop(123, None)
+
+
+def test_core_refuses_video_before_download_when_gemini_exhausted(rate_guard_setup, monkeypatch):
+    # Защищает ранний отказ для видео/аудио: поздняя проверка скачала бы файл
+    # и пожгла транскрибацию перед отказом. Регрессия — возврат проверки под
+    # скачивание. Старые тесты гоняют только текстовые/YouTube-отказы.
+    from types import SimpleNamespace as _NS
+    message = rate_guard_setup()
+    message.text = "смотри"
+    message.video = _NS(file_id="vid123")
+    monkeypatch.setattr(bot, "DAILY_USER_GEMINI_LIMIT", 1)
+    bot._record_user_daily(456, gemini=True)
+    replies = []
+
+    async def fake_safe_reply(msg, text, **kwargs):
+        replies.append(text)
+
+    async def fail_download(*args, **kwargs):
+        raise AssertionError("exhausted user media must not be downloaded")
+
+    async def fail_route(*args, **kwargs):
+        raise AssertionError("gemini-only request must not reach models")
+
+    monkeypatch.setattr(bot, "_safe_reply", fake_safe_reply)
+    monkeypatch.setattr(bot, "_run_route", fail_route)
+    monkeypatch.setattr(bot, "_download_message_attachment_to_tmp", fail_download)
+    monkeypatch.setattr(bot, "_fetch_media", fail_download)
+    try:
+        asyncio.run(bot._handle_message_core(message))
+        assert len(replies) == 1 and "1/1" in replies[0]
+    finally:
+        bot.GLOBAL_QUOTA.get("user_daily", {}).pop("456", None)
+        bot.chat_state.pop(123, None)