Fix: Add safety checks in get_client() and improve error handling

utils/clients.py

@@ -3,8 +3,6 @@ from pathlib import Path
 from pyrogram import Client
 from utils.directoryHandler import backup_drive_data, loadDriveData
 from utils.logger import Logger
-import os
-import signal
 
 logger = Logger(__name__)
 
@@ -84,9 +82,7 @@ async def initialize_clients():
     )
     if len(multi_clients) == 0:
         logger.error("No Clients Were Initialized")
-
-        # Forcefully terminates the program immediately
-        os.kill(os.getpid(), signal.SIGKILL)
+        raise RuntimeError("No Clients Were Initialized - Check BOT_TOKENS configuration")
 
     if len(premium_clients) == 0:
         logger.info("No Premium Clients Were Initialized")
@@ -104,10 +100,14 @@ def get_client(premium_required=False) -> Client:
     global multi_clients, work_loads, premium_clients, premium_work_loads
 
     if premium_required:
+        if not premium_work_loads:
+            raise RuntimeError("No Premium Clients Available")
         index = min(premium_work_loads, key=premium_work_loads.get)
         premium_work_loads[index] += 1
         return premium_clients[index]
 
+    if not work_loads:
+        raise RuntimeError("No Clients Available - Check BOT_TOKENS configuration")
     index = min(work_loads, key=work_loads.get)
     work_loads[index] += 1
     return multi_clients[index]