Update devgagan/__init__.py

devgagan/__init__.py

@@ -1,6 +1,3 @@
-#devggn
-
-
 import asyncio
 import logging
 from pyromod import listen
@@ -8,6 +5,14 @@ from pyrogram import Client
 from config import API_ID, API_HASH, BOT_TOKEN
 from telethon.sync import TelegramClient
 
+# --- Proxy Configuration ---
+PROXY_CONFIG = {
+    "scheme": "http", # Or "socks5"
+    "hostname": "154.196.33.209",
+    "port": 63452,
+    "username": "L4SzFXFkA",
+    "password": "EC43VV3LU"
+}
 
 loop = asyncio.get_event_loop()
 
@@ -16,32 +21,34 @@ logging.basicConfig(
     level=logging.INFO,
 )
 
-sex = TelegramClient('sexrepo', API_ID, API_HASH).start(bot_token=BOT_TOKEN)
+# Telethon Proxy Setup
+sex = TelegramClient(
+    'sexrepo', 
+    API_ID, 
+    API_HASH, 
+    proxy=(PROXY_CONFIG['scheme'], PROXY_CONFIG['hostname'], PROXY_CONFIG['port'], True, PROXY_CONFIG['username'], PROXY_CONFIG['password'])
+).start(bot_token=BOT_TOKEN)
 
+# Pyrogram Proxy Setup
 app = Client(
     ":RestrictBot:",
     api_id=API_ID,
     api_hash=API_HASH,
     bot_token=BOT_TOKEN,
-    workers=10,
+    workers=20,
     sleep_threshold=20,
-    max_concurrent_transmissions=5
+    max_concurrent_transmissions=5,
+    proxy=PROXY_CONFIG # Pyrogram accepts a dictionary
 )
 
-
-
 async def restrict_bot():
     global BOT_ID, BOT_NAME, BOT_USERNAME
-    await app.start()
-    getme = await app.get_me()
-    BOT_ID = getme.id
-    BOT_USERNAME = getme.username
-    if getme.last_name:
-        BOT_NAME = getme.first_name + " " + getme.last_name
-    else:
-        BOT_NAME = getme.first_name
-
-
-loop.run_until_complete(restrict_bot())
-
-
+    async with app: # Using context manager is safer
+        getme = await app.get_me()
+        BOT_ID = getme.id
+        BOT_USERNAME = getme.username
+        BOT_NAME = f"{getme.first_name} {getme.last_name}" if getme.last_name else getme.first_name
+        print(f"Bot started as {BOT_NAME}")
+
+if __name__ == "__main__":
+    loop.run_until_complete(restrict_bot())