Fix thumbnail

Signed-off-by: anasty17 <e.anastayyar@gmail.com>
rwcats · Nov 8, 2023 · d00f87c · d00f87c
1 parent fdb2b6c
commit d00f87c
Show file tree

Hide file tree

Showing 6 changed files with 518 additions and 414 deletions.
diff --git a/bot/helper/common.py b/bot/helper/common.py
@@ -276,7 +276,7 @@ async def beforeStart(self):
                 )
 
             if is_telegram_link(self.thumb):
-                msg = await get_tg_link_message(self.thumb)
+                msg = (await get_tg_link_message(self.thumb))[0]
                 self.thumb = await createThumb(msg) if msg.photo or msg.document else ""
 
     async def getTag(self, text: list):

diff --git a/bot/helper/ext_utils/bot_utils.py b/bot/helper/ext_utils/bot_utils.py
@@ -61,12 +61,8 @@ async def initiate_help_messages():
         telegraph.create_page(
             title="Mirror-Leech Command Usage", content=MIRROR_HELP_MESSAGE
         ),
-        telegraph.create_page(
-            title="YTDLP Command Usage", content=YT_HELP_MESSAGE
-        ),
-        telegraph.create_page(
-            title="Clone Command Usage", content=CLONE_HELP_MESSAGE
-        ),
+        telegraph.create_page(title="YTDLP Command Usage", content=YT_HELP_MESSAGE),
+        telegraph.create_page(title="Clone Command Usage", content=CLONE_HELP_MESSAGE),
     )
     buttons = ButtonMaker()
     buttons.ubutton("Usage Guide", f"https://telegra.ph/{mirror['path']}")

diff --git a/bot/helper/ext_utils/media_utils.py b/bot/helper/ext_utils/media_utils.py
@@ -26,12 +26,14 @@ def getSplitSizeBytes(size):
     return size
 
 
-async def createThumb(msg):
+async def createThumb(msg, _id=""):
+    if not _id:
+        _id = msg.id
     path = "Thumbnails/"
     if not await aiopath.isdir(path):
         await mkdir(path)
     photo_dir = await msg.download()
-    des_dir = f"{path}{msg.id}.jpg"
+    des_dir = f"{path}{_id}.jpg"
     await sync_to_async(Image.open(photo_dir).convert("RGB").save, des_dir, "JPEG")
     await aioremove(photo_dir)
     return des_dir

diff --git a/bot/helper/mirror_utils/rclone_utils/transfer.py b/bot/helper/mirror_utils/rclone_utils/transfer.py
@@ -220,7 +220,9 @@ async def _get_gdrive_link(self, config_path, remote, rc_path, mime_type):
 
         if code == 0:
             result = loads(res)
-            fid = next((r["ID"] for r in result if r["Path"] == self._listener.name), "err")
+            fid = next(
+                (r["ID"] for r in result if r["Path"] == self._listener.name), "err"
+            )
             link = (
                 f"https://drive.google.com/drive/folders/{fid}"
                 if mime_type == "Folder"
@@ -419,7 +421,9 @@ async def clone(self, config_path, src_remote, src_path, mime_type):
                 return (None, None) if self._is_cancelled else (link, destination)
             else:
                 if mime_type != "Folder":
-                    destination += f"/{self._listener.name}" if dst_path else self._listener.name
+                    destination += (
+                        f"/{self._listener.name}" if dst_path else self._listener.name
+                    )
 
                 cmd = ["rclone", "link", "--config", config_path, destination]
                 res, err, code = await cmd_exec(cmd)