Update bot/telegram/media.py

bot/telegram/media.py

@@ -1,59 +1,23 @@
 # PATH: bot/telegram/media.py
 import os
-import time
-from typing import Callable, Optional, Tuple, Any
-
 from hydrogram import Client
 from hydrogram.types import Message
-
 from bot.temp.files import make_temp_file_path
 
-
-ProgressCb = Callable[[int, int], Any]
-
-
-async def download_to_temp(
-    app: Client,
-    m: Message,
-    progress: Optional[ProgressCb] = None,
-    progress_args: tuple = (),
-) -> Tuple[str, int, str, float]:
+async def download_to_temp(app: Client, m: Message) -> tuple[str, int, str]:
     """
-    Returns: (file_path, file_size_bytes, file_name, elapsed_seconds)
-    progress(done:int, total:int) -> awaited by Hydrogram (pyrogram-like)
+    Returns: (file_path, file_size, file_name)
+    NOTE: Always returns exactly 3 values.
     """
     media = m.video or m.document
     if not media:
         raise RuntimeError("No media to download")
 
     file_name = getattr(media, "file_name", None) or "video.bin"
-    tg_size = int(getattr(media, "file_size", 0) or 0)
+    file_size = int(getattr(media, "file_size", 0) or 0)
 
     out_path = make_temp_file_path(file_name)
     os.makedirs(os.path.dirname(out_path), exist_ok=True)
 
-    t0 = time.time()
-
-    # Hydrogram (Pyrogram-like) supports progress + progress_args
-    if progress:
-        await app.download_media(
-            message=m,
-            file_name=out_path,
-            progress=progress,
-            progress_args=progress_args,
-        )
-    else:
-        await app.download_media(message=m, file_name=out_path)
-
-    sec = max(0.001, time.time() - t0)
-
-    # get real size from disk (telegram sometimes gives 0)
-    real_size = 0
-    try:
-        real_size = os.path.getsize(out_path)
-    except Exception:
-        real_size = 0
-
-    size = real_size if real_size > 0 else tg_size
-
-    return out_path, int(size), file_name, sec
+    await app.download_media(message=m, file_name=out_path)
+    return out_path, file_size, file_name