Testing Changes

FileStream/bot/__init__.py

@@ -26,7 +26,7 @@ ACTIVE_CLIENTS = {}
 async def req_client(PUBLIC_IP:str)-> dict:
   index = min(WORK_LOADS, key=WORK_LOADS.get)
   faster_client = MULTI_CLIENTS[index]
-  print("\n \nFaster Client",faster_client,type(faster_client),"\nWORK_LOADS",WORK_LOADS,type(WORK_LOADS),"\nMULTI_CLIENTS",MULTI_CLIENTS,type(MULTI_CLIENTS),"\n")
+  pprint("\n \nFaster Client",faster_client,type(faster_client),"\nWORK_LOADS",WORK_LOADS,type(WORK_LOADS),"\nMULTI_CLIENTS",MULTI_CLIENTS,type(MULTI_CLIENTS),"\n")
   #ACTIVE_CLIENTS.update(faster_client)
   #response = dict(index=index, client=faster_client)
   return dict(index=index,public_ip=PUBLIC_IP, client=faster_client)

FileStream/server/Functions/downloader.py

@@ -22,23 +22,24 @@ from FileStream.utils.FileProcessors.custom_ul import  TeleUploader
 async def media_streamer(request: web.Request, db_id: str, speed: str):
     # Get the Range header from the request, default to 0 if not present
     range_header = request.headers.get("Range", 0)
-    client = await req_client(request.remote)
+    Worker = await req_client(request.remote)
+    client=Worker["client"]
     # Log client info if multi-client mode
 
     if Telegram.MULTI_CLIENT:
-        logging.info(f"Client {client['index']} is now serving {request.headers.get('X-FORWARDED-FOR', request.remote)}")
+        logging.info(f"Client {Worker['index']} is now serving {request.headers.get('X-FORWARDED-FOR', request.remote)}")
 
     # Use an existing ByteStreamer or create a new one
     #tg_connect = ACTIVE_CLIENTS.get(client['client'], None)
     
-    if client["client"] in ACTIVE_CLIENTS:
-        logging.info(f"Creating new ByteStreamer object for client {client['index']}")
-        tg_connect = ACTIVE_CLIENTS(client["client"])
+    if client in ACTIVE_CLIENTS:
+        logging.info(f"Creating new ByteStreamer object for client {Worker['index']}")
+        tg_connect = ACTIVE_CLIENTS(client)
 
     else:
-        tg_connect = utils.ByteStreamer(client["client"])
-        ACTIVE_CLIENTS[client["client"]] = tg_connect
-        logging.info(f"Using cached ByteStreamer object for client {client['index']}")
+        tg_connect = utils.ByteStreamer(client)
+        ACTIVE_CLIENTS[client] = tg_connect
+        logging.info(f"Using cached ByteStreamer object for client {Worker['index']}")
 
     tg_connect.update_last_activity()
 
@@ -71,7 +72,7 @@ async def media_streamer(request: web.Request, db_id: str, speed: str):
 
         # Request the file chunks
         body = tg_connect.yield_file(
-            file_id, client['index'], offset, first_part_cut, last_part_cut, part_count, chunk_size
+            file_id, Worker['index'], offset, first_part_cut, last_part_cut, part_count, chunk_size
         )
 
         # Determine MIME type and filename