Update main.py

main.py

@@ -1,4 +1,4 @@
-# filename: main.py
+# filename: main.py (Updated with DNS Resolver in Startup)
 
 import logging
 import asyncio
@@ -11,6 +11,7 @@ load_dotenv()
 from fastapi import FastAPI
 from fastapi.responses import FileResponse, PlainTextResponse
 import uvicorn
+import dns.asyncresolver # <-- NEW IMPORT
 
 # Now, import our application modules AFTER loading the environment
 import config
@@ -21,15 +22,13 @@ from database import manager as db_manager
 from handlers import start, admin, user, links
 
 # --- Logging Configuration ---
-# Creates the logs directory if it doesn't exist
 os.makedirs("logs", exist_ok=True)
-
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     handlers=[
-        logging.StreamHandler(), # To show logs in the console
-        logging.FileHandler("logs/bot.log", mode='a', encoding='utf-8') # To save logs to a file
+        logging.StreamHandler(),
+        logging.FileHandler("logs/bot.log", mode='a', encoding='utf-8')
     ]
 )
 logger = logging.getLogger(__name__)
@@ -55,6 +54,17 @@ async def startup_event():
     # 1. Initialize the database and create indexes for fast queries
     await db_manager.setup_database_indexes()
     
+    # --- DNS RESOLVER CODE ADDED HERE, AS REQUESTED ---
+    try:
+        logger.info("Attempting to configure custom DNS resolvers...")
+        async_resolver = dns.asyncresolver.Resolver(configure=False)
+        async_resolver.nameservers = ["8.8.8.8", "8.8.4.4", "1.1.1.1"]
+        dns.asyncresolver.default_resolver = async_resolver
+        logger.info("Successfully configured custom DNS resolvers [Google, Cloudflare].")
+    except Exception as e:
+        logger.warning(f"Could not configure custom DNS resolver: {e}")
+    # --- END OF DNS CODE ---
+
     # 2. Start the Telethon client
     logger.info("Starting Telegram client...")
     await bot.start(bot_token=config.BOT_TOKEN)
@@ -79,12 +89,9 @@ async def shutdown_event():
 
 
 # --- FastAPI Routes (Web Endpoints) ---
+# ... (The rest of the file is unchanged) ...
 @app.get("/", response_class=FileResponse)
 async def read_index():
-    """
-    Serves the main web interface if index.html exists.
-    Otherwise, returns a simple status message.
-    """
     index_path = "index.html"
     if os.path.exists(index_path):
         return FileResponse(index_path)
@@ -93,7 +100,6 @@ async def read_index():
 
 @app.get("/health", response_class=PlainTextResponse)
 async def health_check():
-    """A simple endpoint for health checks and monitoring."""
     if bot.is_connected():
         return PlainTextResponse("✅ OK: Bot service is running and connected to Telegram.")
     else:
@@ -103,11 +109,9 @@ async def health_check():
 # --- Main Execution Block ---
 if __name__ == "__main__":
     logger.info("Starting Uvicorn server...")
-    # Uvicorn runs the FastAPI application.
-    # 'reload=True' is for development; you might remove it in production.
     uvicorn.run(
         "main:app",
         host="0.0.0.0",
-        port=8080, # Common port for web services
-        reload=True # Automatically restarts the server when you save a file
+        port=8080,
+        reload=True
     )