add error handling for magnet uri being none

core/network/aria2_integration.py

@@ -4,6 +4,7 @@ from core.utils.data.state import state
 from core.utils.general.logs import update_download_completed_by_hash
 from core.utils.general.logs import consoleLog
 from plyer import notification
+import socket
 import time
 import sys
 
@@ -42,6 +43,21 @@ def aria2server():
         stderr=subprocess.DEVNULL,
         creationflags=subprocess.CREATE_NO_WINDOW if sys.platform == 'win32' else 0
     )
+
+    max_checks = 50
+    for check in range(max_checks):
+        try:
+            sock = socket.socket()
+            sock.settimeout(0.1)
+            sock.connect(("localhost", 6800))
+            sock.close()
+            break
+        except(ConnectionRefusedError, socket.timeout):
+            time.sleep(0.1)
+    else:
+        raise RuntimeError("Aria2 Server failed to start")
+
+    consoleLog("Aria2 Server started")
     return aria2server
 
 def dlprogress():

core/network/aria2_wrapper.py

@@ -8,8 +8,11 @@ def start_client():
     consoleLog("Started Aria2p")
 
 def add_magnet(uri):
-    aria2.add_magnet(uri)
-    consoleLog("Magnet URI added to Aria2")
+    if uri is not None and uri.startswith("magnet:?"):  
+        aria2.add_magnet(uri)
+        consoleLog("Magnet URI added to Aria2")
+    else:
+        consoleLog(f"Invalid Magnet Link: {uri}")
 
 
 

core/utils/general/loghandler.py

@@ -1,4 +1,3 @@
-from core.utils.data.state import state
 from core.utils.general.logs import consoleLog
 from core.utils.network.download import run_download_direct