Enhance client management with DC tracking

Added client_dc_map to track data center IDs for clients and updated logging for client connections.

Backend/pyrofork/clients.py

@@ -2,7 +2,7 @@ from asyncio import gather, create_task
 from pyrogram import Client
 from Backend.logger import LOGGER
 from Backend.config import Telegram
-from Backend.pyrofork.bot import multi_clients, work_loads, StreamBot
+from Backend.pyrofork.bot import multi_clients, work_loads, StreamBot, client_dc_map
 from os import environ
 
 class TokenParser:
@@ -12,7 +12,7 @@ class TokenParser:
             c + 1: t
             for c, (_, t) in enumerate(
                 filter(
-                    lambda n: n[0].startswith("MULTI_TOKEN"), 
+                    lambda n: n[0].startswith("MULTI_TOKEN"),
                     sorted(environ.items())
                 )
             )
@@ -31,6 +31,15 @@ async def start_client(client_id, token):
             no_updates=True,
             in_memory=True
         ).start()
+        
+        try:
+            client_dc = await client.storage.dc_id()
+            client_dc_map[client_id] = client_dc
+            LOGGER.info(f"Client {client_id} connected to DC {client_dc}")
+        except Exception as e:
+            LOGGER.warning(f"Could not get DC for Client {client_id}: {e}")
+            client_dc_map[client_id] = None
+        
         work_loads[client_id] = 0
         return client_id, client
     except Exception as e:
@@ -39,6 +48,15 @@ async def start_client(client_id, token):
 
 async def initialize_clients():
     multi_clients[0], work_loads[0] = StreamBot, 0
+    
+    try:
+        main_dc = await StreamBot.storage.dc_id()
+        client_dc_map[0] = main_dc
+        LOGGER.info(f"Main StreamBot connected to DC {main_dc}")
+    except Exception as e:
+        LOGGER.warning(f"Could not get DC for StreamBot: {e}")
+        client_dc_map[0] = None
+    
     all_tokens = TokenParser.parse_from_env()
     if not all_tokens:
         LOGGER.info("No additional Bot Clients found, Using default client")
@@ -46,11 +64,11 @@ async def initialize_clients():
 
     tasks = [create_task(start_client(i, token)) for i, token in all_tokens.items()]
     clients = await gather(*tasks)
-    clients = {client_id: client for client_id, client in clients if client} 
+    clients = {client_id: client for client_id, client in clients if client}
     multi_clients.update(clients)
-    
+
     if len(multi_clients) != 1:
         LOGGER.info(f"Multi-Client Mode Enabled with {len(multi_clients)} clients")
+        LOGGER.info(f"DC Distribution: {client_dc_map}")
     else:
         LOGGER.info("No additional clients were initialized, using default client")
-