Update bot/telegram/media.py

bot/telegram/media.py

@@ -1,38 +1,59 @@
 # PATH: bot/telegram/media.py
 import os
-from typing import Callable, Any, Optional, Tuple
+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 = Optional[Callable[[int, int], Any]]
+
+ProgressCb = Callable[[int, int], Any]
+
 
 async def download_to_temp(
     app: Client,
     m: Message,
-    progress: ProgressCb = None,
+    progress: Optional[ProgressCb] = None,
     progress_args: tuple = (),
-    **_ignore,  # <-- future-proof: ignores unknown kwargs safely
-) -> Tuple[str, int, str]:
+) -> Tuple[str, int, str, float]:
     """
-    Returns: (file_path, file_size, file_name)
+    Returns: (file_path, file_size_bytes, file_name, elapsed_seconds)
+    progress(done:int, total:int) -> awaited by Hydrogram (pyrogram-like)
     """
     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"
-    file_size = int(getattr(media, "file_size", 0) or 0)
+    tg_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)
 
-    # Hydrogram (pyrogram-like) supports progress/progress_args
-    await app.download_media(
-        message=m,
-        file_name=out_path,
-        progress=progress,
-        progress_args=progress_args,
-    )
+    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, file_size, file_name
+    return out_path, int(size), file_name, sec