File size: 1,223 Bytes
886cc56
e566133
 
fa3f974
e566133
 
 
 
 
 
bf84dca
 
 
3dac60c
 
 
 
 
e566133
5a67dbb
 
 
 
 
 
 
 
 
 
 
e566133
0e49d61
 
 
 
e566133
fdbc27b
26741bb
0e49d61
81b9358
 
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
import pprint
from ..config import Telegram
from pyrogram import Client
from pyrogram.session import Session

if Telegram.SECONDARY:
  plugins = None
  no_updates = True
else:
  plugins = dict(root="FileStream/bot/plugins")
  no_updates = False
  
# Create a custom session object with the sqlite.cloud connection string
#session = Session(
#    api_id=Telegram.API_ID,
#    api_hash=Telegram.API_HASH,
#    db_uri="sqlitecloud://cgz1np12hz.g1.sqlite.cloud:8860/FileStream?apikey=NnfEaSHjnV1jpSUMlRMzm3agsn5jUb0dymlnCUXS2k8",  # Set the custom database URI
#)

FileStream = Client(
            name="FileStream",
            api_id=Telegram.API_ID,
            api_hash=Telegram.API_HASH,
            workdir="FileStream",
            plugins=plugins,
            bot_token=Telegram.BOT_TOKEN,
            sleep_threshold=Telegram.SLEEP_THRESHOLD,
            workers=Telegram.WORKERS,
            no_updates=no_updates,
          )


WORK_LOADS = {}
MULTI_CLIENTS = {}
ACTIVE_CLIENTS = {}

async def req_client(PUBLIC_IP:str)-> dict:
  index = min(WORK_LOADS, key=WORK_LOADS.get)
  faster_client = MULTI_CLIENTS[index]
  print("WorkLoads :",WORK_LOADS)
  return dict(index=index,public_ip=PUBLIC_IP, client=faster_client)