File size: 2,095 Bytes
157e137 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import asyncio
import logging
from os import environ
from ..config import Telegram
from pyrogram import Client
from . import multi_clients, work_loads, FileStream
async def initialize_clients():
all_tokens = dict(
(c + 1, t)
for c, (_, t) in enumerate(
filter(
lambda n: n[0].startswith("MULTI_TOKEN"), sorted(environ.items())
)
)
)
if not all_tokens:
multi_clients[0] = FileStream
work_loads[0] = 0
print("No additional clients found, using default client")
return
async def start_client(client_id, token):
try:
if len(token) >= 100:
session_string=token
bot_token=None
print(f'Starting Client - {client_id} Using Session String')
else:
session_string=None
bot_token=token
print(f'Starting Client - {client_id} Using Bot Token')
if client_id == len(all_tokens):
await asyncio.sleep(2)
print("This will take some time, please wait...")
client = await Client(
name=str(client_id),
api_id=Telegram.API_ID,
api_hash=Telegram.API_HASH,
bot_token=bot_token,
sleep_threshold=Telegram.SLEEP_THRESHOLD,
no_updates=True,
session_string=session_string,
in_memory=True,
).start()
client.id = (await client.get_me()).id
work_loads[client_id] = 0
return client_id, client
except Exception:
logging.error(f"Failed starting Client - {client_id} Error:", exc_info=True)
clients = await asyncio.gather(*[start_client(i, token) for i, token in all_tokens.items()])
multi_clients.update(dict(clients))
if len(multi_clients) != 1:
Telegram.MULTI_CLIENT = True
print("Multi-Client Mode Enabled")
else:
print("No additional clients were initialized, using default client")
|