Update bot/telegram/media.py

bot/telegram/media.py

@@ -1,12 +1,22 @@
 # PATH: bot/telegram/media.py
 import os
+import time
+from typing import Callable, Optional, Tuple
 from hydrogram import Client
 from hydrogram.types import Message
 from bot.temp.files import make_temp_file_path
 
-async def download_to_temp(app: Client, m: Message) -> tuple[str, int, str]:
+ProgressCb = Callable[[int, int], None]
+
+async def download_to_temp(
+    app: Client,
+    m: Message,
+    progress: Optional[ProgressCb] = None,
+    progress_args: tuple = (),
+) -> Tuple[str, int, str, float]:
     """
-    Returns: (file_path, file_size, file_name)
+    Returns: (file_path, file_size, file_name, elapsed_sec)
+    progress(done, total) will be called if supported by Hydrogram.
     """
     media = m.video or m.document
     if not media:
@@ -18,6 +28,19 @@ async def download_to_temp(app: Client, m: Message) -> tuple[str, int, str]:
     out_path = make_temp_file_path(file_name)
     os.makedirs(os.path.dirname(out_path), exist_ok=True)
 
-    # Hydrogram download (async)
-    await app.download_media(message=m, file_name=out_path)
-    return out_path, file_size, file_name
+    t0 = time.time()
+
+    # Hydrogram (Pyrogram-like) usually supports progress/progress_args.
+    # If not supported, fall back.
+    try:
+        await app.download_media(
+            message=m,
+            file_name=out_path,
+            progress=progress,
+            progress_args=progress_args,
+        )
+    except TypeError:
+        await app.download_media(message=m, file_name=out_path)
+
+    elapsed = max(0.001, time.time() - t0)
+    return out_path, file_size, file_name, elapsed