some fixes

Signed-off-by: anasty17 <e.anastayyar@gmail.com>

Author: anasty17

bot/core/startup.py

@@ -1,7 +1,7 @@
 from aiofiles.os import path as aiopath, remove, makedirs
 from aiofiles import open as aiopen
 from aioshutil import rmtree
-from asyncio import create_subprocess_exec, create_subprocess_shell
+from asyncio import create_subprocess_exec, create_subprocess_shell, sleep
 from importlib import import_module
 
 from .. import (
@@ -26,6 +26,7 @@
 
 
 async def update_qb_options():
+    LOGGER.info("Get qBittorrent options from server")
     if not qbit_options:
         opt = await TorrentManager.qbittorrent.app.preferences()
         qbit_options.update(opt)
@@ -42,6 +43,7 @@ async def update_qb_options():
 
 
 async def update_aria2_options():
+    LOGGER.info("Get aria2 options from server")
     if not aria2_options:
         op = await TorrentManager.aria2.getGlobalOption()
         aria2_options.update(op)
@@ -50,8 +52,15 @@ async def update_aria2_options():
 
 
 async def update_nzb_options():
-    no = (await sabnzbd_client.get_config())["config"]["misc"]
-    nzb_options.update(no)
+    LOGGER.info("Get SABnzbd options from server")
+    while True:
+        try:
+            no = (await sabnzbd_client.get_config())["config"]["misc"]
+            nzb_options.update(no)
+        except:
+            await sleep(0.5)
+            continue
+        break
 
 
 async def load_settings():

bot/helper/ext_utils/bot_utils.py

@@ -144,7 +144,11 @@ def arg_parser(items, arg_base):
             else:
                 sub_list = []
                 for j in range(i + 1, total):
-                    if items[j] in arg_base:
+                    if (
+                        items[j] in arg_base
+                        and not arg_base[items[j]]
+                        and items[j] != part
+                    ):
                         if part in bool_arg_set and not sub_list:
                             arg_base[part] = True
                             break

bot/helper/ext_utils/media_utils.py

@@ -138,8 +138,6 @@ async def take_ss(video_file, ss_nb) -> bool:
                 "1",
                 "-frames:v",
                 "1",
-                "-threads",
-                f"{max(1, cpu_no // 2)}",
                 output,
             ]
             cap_time += interval
@@ -178,8 +176,6 @@ async def get_audio_thumbnail(audio_file):
         "-an",
         "-vcodec",
         "copy",
-        "-threads",
-        f"{max(1, cpu_no // 2)}",
         output,
     ]
     try:
@@ -221,8 +217,6 @@ async def get_video_thumbnail(video_file, duration):
         "1",
         "-frames:v",
         "1",
-        "-threads",
-        f"{max(1, cpu_no // 2)}",
         output,
     ]
     try:
@@ -266,8 +260,6 @@ async def get_multiple_frames_thumbnail(video_file, layout, keep_screenshots):
         "1",
         "-f",
         "mjpeg",
-        "-threads",
-        f"{max(1, cpu_no // 2)}",
         output,
     ]
     try:
@@ -446,8 +438,6 @@ async def convert_video(self, video_file, ext, retry=False):
                 "libx264",
                 "-c:a",
                 "aac",
-                "-threads",
-                f"{max(1, cpu_no // 2)}",
                 output,
             ]
             if ext == "mp4":
@@ -470,8 +460,6 @@ async def convert_video(self, video_file, ext, retry=False):
                 "0",
                 "-c",
                 "copy",
-                "-threads",
-                f"{max(1, cpu_no // 2)}",
                 output,
             ]
         if self._listener.is_cancelled:
@@ -517,8 +505,6 @@ async def convert_audio(self, audio_file, ext):
             "pipe:1",
             "-i",
             audio_file,
-            "-threads",
-            f"{max(1, cpu_no // 2)}",
             output,
         ]
         if self._listener.is_cancelled:
@@ -597,8 +583,6 @@ async def sample_video(self, video_file, sample_duration, part_duration):
             "libx264",
             "-c:a",
             "aac",
-            "-threads",
-            f"{max(1, cpu_no // 2)}",
             output_file,
         ]
 
@@ -662,8 +646,6 @@ async def split(self, f_path, file_, parts, split_size):
                 "-2",
                 "-c",
                 "copy",
-                "-threads",
-                f"{max(1, cpu_no // 2)}",
                 out_path,
             ]
             if not multi_streams:

bot/helper/mirror_leech_utils/status_utils/nzb_status.py

@@ -10,6 +10,8 @@
 
 
 async def get_download(nzo_id, old_info=None):
+    if old_info is None:
+        old_info = {}
     try:
         queue = await sabnzbd_client.get_downloads(nzo_ids=nzo_id)
         if res := queue["queue"]["slots"]:
@@ -59,49 +61,51 @@ def __init__(self, listener, gid, queued=False):
         self.queued = queued
         self.listener = listener
         self._gid = gid
-        self._info = None
+        self._info = {}
         self.tool = "sabnzbd"
 
     async def update(self):
         self._info = await get_download(self._gid, self._info)
 
     def progress(self):
-        return f"{self._info['percentage']}%"
+        return f"{self._info.get('percentage', "0")}%"
 
     def processed_raw(self):
-        return (float(self._info["mb"]) - float(self._info["mbleft"])) * 1048576
+        return (
+            float(self._info.get("mb", "0")) - float(self._info.get("mbleft", "0"))
+        ) * 1048576
 
     def processed_bytes(self):
         return get_readable_file_size(self.processed_raw())
 
     def speed_raw(self):
-        if self._info["mb"] == self._info["mbleft"]:
+        if self._info.get("mb", "0") == self._info.get("mbleft", "0"):
             return 0
         try:
-            return int(float(self._info["mbleft"]) * 1048576) / self.eta_raw()
+            return int(float(self._info.get("mbleft", "0")) * 1048576) / self.eta_raw()
         except:
             return 0
 
     def speed(self):
         return f"{get_readable_file_size(self.speed_raw())}/s"
 
     def name(self):
-        return self._info["filename"]
+        return self._info.get("filename", "")
 
     def size(self):
-        return self._info["size"]
+        return self._info.get("size", 0)
 
     def eta_raw(self):
-        return int(time_to_seconds(self._info["timeleft"]))
+        return int(time_to_seconds(self._info.get("timeleft", "0")))
 
     def eta(self):
         return get_readable_time(self.eta_raw())
 
     async def status(self):
         await self.update()
-        if self._info["mb"] == self._info["mbleft"]:
+        if self._info.get("mb", "0") == self._info.get("mbleft", "0"):
             return MirrorStatus.STATUS_QUEUEDL
-        state = self._info["status"]
+        state = self._info.get("status")
         if state == "Paused" and self.queued:
             return MirrorStatus.STATUS_QUEUEDL
         elif state in [