Some minor fixes

Signed-off-by: anasty17 <[email protected]>

bot/__init__.py

@@ -24,7 +24,6 @@
 from time import time
 from tzlocal import get_localzone
 from uvloop import install
-from asyncio import run as aiorun
 
 # from faulthandler import enable as faulthandler_enable
 # faulthandler_enable()

bot/helper/ext_utils/db_handler.py

@@ -37,7 +37,7 @@ async def db_load(self):
             return
         # Save bot settings
         try:
-            res = await self._db.settings.config.replace_one(
+            await self._db.settings.config.replace_one(
                 {"_id": bot_id}, config_dict, upsert=True
             )
         except Exception as e:

bot/helper/ext_utils/media_utils.py

@@ -202,9 +202,8 @@ async def get_document_type(path):
                 path,
             ]
         )
-        if res := result[1]:
-            if mime_type.startswith("video"):
-                is_video = True
+        if result[1] and mime_type.startswith("video"):
+            is_video = True
     except Exception as e:
         LOGGER.error(f"Get Document Type: {e}. Mostly File not found! - File: {path}")
         if mime_type.startswith("video"):

bot/helper/listeners/aria2_listener.py

@@ -37,7 +37,7 @@ async def _onDownloadStarted(api, gid):
                     if download.is_removed or download.followed_by_ids:
                         await deleteMessage(meta)
                         break
-                    download = download.live
+                    download = await sync_to_async(download.live)
         return
     else:
         LOGGER.info(f"onDownloadStarted: {download.name} - Gid: {gid}")
@@ -46,7 +46,7 @@ async def _onDownloadStarted(api, gid):
     if task := await getTaskByGid(gid):
         download = await sync_to_async(api.get_download, gid)
         await sleep(2)
-        download = download.live
+        download = await sync_to_async(download.live)
         task.listener.name = download.name
         msg, button = await stop_duplicate_check(task.listener)
         if msg:
@@ -127,7 +127,7 @@ async def _onBtDownloadComplete(api, gid):
         await task.listener.onDownloadComplete()
         if Intervals["stopAll"]:
             return
-        download = download.live
+        download = await sync_to_async(download.live)
         if task.listener.seed:
             if download.is_complete:
                 if task := await getTaskByGid(gid):

bot/helper/mirror_leech_utils/download_utils/direct_link_generator.py

@@ -1445,13 +1445,13 @@ def filelions_and_streamwish(url):
         if quality == version["name"]:
             return version["url"]
         elif version["name"] == "l":
-            error += f"\nLow"
+            error += "\nLow"
         elif version["name"] == "n":
-            error += f"\nNormal"
+            error += "\nNormal"
         elif version["name"] == "o":
-            error += f"\nOriginal"
+            error += "\nOriginal"
         elif version["name"] == "h":
-            error += f"\nHD"
+            error += "\nHD"
         error += f" <code>{url}_{version['name']}</code>"
     raise DirectDownloadLinkException(f"ERROR: {error}")
 

bot/helper/mirror_leech_utils/rclone_utils/transfer.py

@@ -401,8 +401,9 @@ async def clone(self, config_path, src_remote, src_path, mime_type, method):
             return None, None
         elif return_code != 0:
             error = (
-                await self._proc.stderr.read()
-            ).decode().strip() or "Use <code>/shell cat rlog.txt</code> to see more information"
+                (await self._proc.stderr.read()).decode().strip()
+                or "Use <code>/shell cat rlog.txt</code> to see more information"
+            )
             LOGGER.error(error)
             await self._listener.onUploadError(error[:4000])
             return None, None

bot/modules/exec.py

@@ -88,7 +88,7 @@ async def do(func, message):
             func_return = (
                 await sync_to_async(rfunc) if func == "exec" else await rfunc()
             )
-    except Exception as e:
+    except:
         value = stdout.getvalue()
         return f"{value}{format_exc()}"
     else:

bot/modules/ytdlp.py

@@ -390,7 +390,7 @@ async def newEvent(self):
             return
 
         if "mdisk.me" in self.link:
-            name, self.link = await _mdisk(self.link, name)
+            self.name, self.link = await _mdisk(self.link, self.name)
 
         try:
             await self.beforeStart()