Update app.py

app.py

@@ -20,9 +20,10 @@ import uvicorn
 import random
 import docx  # لإضافة دعم .docx
 
-# ========== (الحل) إضافة المكتبات اللازمة لحل مشكلة DNS ==========
+# ========== (الحل) إضافة المكتبات الصحيحة لحل مشكلة DNS ==========
 import httpx
-import httpx_dnspython
+import dns.asyncresolver
+from httpx import AsyncClient, AsyncHTTPTransport, Request, Response
 # ==========================================================
 
 # ========== تكوين السجلات ==========
@@ -49,16 +50,58 @@ REPO_ID = "Riy777/Study"
 # ========== حالات المحادثة ==========
 SELECTING_SUBJECT, SELECTING_ACTION, WAITING_FOR_QUESTION = range(3)
 
+
+# ========== (الحل) فئة الناقل المخصص للـ DNS ==========
+class CustomDNSTransport(AsyncHTTPTransport):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.resolver = dns.asyncresolver.Resolver()
+        self.resolver.nameservers = ['1.1.1.1', '8.8.8.8']
+        logger.info("🔧 CustomDNSTransport initialized with 1.1.1.1 and 8.8.8.8")
+
+    async def handle_async_request(self, request: Request) -> Response:
+        try:
+            host = request.url.host
+            
+            # التحقق إذا كان Host هو أصلاً IP
+            is_ip = True
+            try:
+                parts = host.split('.')
+                if len(parts) != 4 or not all(p.isdigit() for p in parts):
+                    is_ip = False
+            except Exception:
+                is_ip = False
+
+            if is_ip:
+                logger.info(f"Host {host} is already an IP. Skipping resolution.")
+            else:
+                logger.info(f"🔧 Resolving host: {host}")
+                result = await self.resolver.resolve(host, 'A')
+                ip = result[0].address
+                logger.info(f"✅ Resolved {host} to {ip}")
+                
+                # تعيين SNI (مهم جداً لـ SSL/HTTPS)
+                request.extensions["sni_hostname"] = host 
+                # تحديث الـ URL لاستخدام الـ IP
+                request.url = request.url.copy_with(host=ip)
+
+        except Exception as e:
+            logger.error(f"❌ DNS resolution failed for {host}: {e}. Trying request with original host...")
+            # إذا فشل الحل، اسمح للطلب بالاستمرار مع اسم المضيف الأصلي
+            pass
+
+        # استدعاء المعالج الأصلي (الأب) لإكمال الطلب
+        return await super().handle_async_request(request)
+# ==========================================================
+
+
 # ========== تطبيق FastAPI ==========
-# تعديل: استخدام lifespan بدلاً من on_event
 @asyncio.contextmanager
 async def lifespan(app: FastAPI):
-    # الكود الذي كان في on_startup
     logger.info("🚀 بدء تشغيل FastAPI... بدء مهمة تهيئة البوت في الخلفية.")
     asyncio.create_task(bot.initialize_application())
     logger.info("✅ خادم FastAPI يعمل. التهيئة (الاتصال بـ Telegram) جارية في الخلفية.")
     yield
-    # كود إيقاف التشغيل (on_shutdown) (إن وجد)
     logger.info("🛑 إيقاف تشغيل خادم FastAPI.")
 
 app = FastAPI(title="Medical Lab Bot", version="1.0.0", lifespan=lifespan)
@@ -83,12 +126,9 @@ class MedicalLabBot:
             logger.info("🔄 جاري تهيئة تطبيق التليجرام...")
             
             # ========== !! الحل الجذري (DNS) !! ==========
-            logger.info("🔧 إعداد عميل HTTP مخصص مع DNS (1.1.1.1)...")
-            transport = httpx_dnspython.AsyncDNSTransport(
-                resolver=httpx_dnspython.SyncResolver(nameservers=["1.1.1.1", "8.8.8.8"])
-            )
-            # نستخدم httpx.AsyncClient مع الناقل المخصص
-            custom_client = httpx.AsyncClient(transport=transport)
+            logger.info("🔧 إعداد عميل HTTP مخصص مع CustomDNSTransport...")
+            custom_transport = CustomDNSTransport()
+            custom_client = httpx.AsyncClient(transport=custom_transport)
             # ============================================
 
             self.application = (
@@ -99,8 +139,8 @@ class MedicalLabBot:
             )
             await self.setup_handlers()
             
-            max_retries = 3 # تقليل المحاولات لأن المشكلة يفترض أن تُحل
-            retry_delay = 5 # تقليل التأخير
+            max_retries = 3 
+            retry_delay = 5
             
             for attempt in range(max_retries):
                 try:
@@ -545,7 +585,7 @@ class MedicalLabBot:
             keyboard.append([InlineKeyboardButton(display_name, callback_data=f"subject_{subject}")])
         
         keyboard.append([InlineKeyboardButton("🔄 تحديث قائمة المواد", callback_data="refresh_materials")])
-        keyboard.append([InlineKeyboardButton("❓ مساعدة", callback_Ddata="general_help")])
+        keyboard.append([InlineKeyboardButton("❓ مساعدة", callback_data="general_help")])
         return keyboard
     
     async def handle_subject_selection(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
@@ -809,15 +849,6 @@ bot = MedicalLabBot()
 
 # ========== دوال FastAPI ==========
 
-# تم نقل @app.on_event("startup") إلى تعريف "lifespan"
-# @app.on_event("startup")
-# async def on_startup():
-#     """بدء تشغيل التطبيق وبدء تهيئة البوت في الخلفية"""
-#     logger.info("🚀 بدء تشغيل FastAPI... بدء مهمة تهيئة البوت في الخلفية.")
-#     # لا ننتظر (no await) - سيبدأ الخادم فوراً، وستعمل هذه المهمة في الخلفية
-#     asyncio.create_task(bot.initialize_application())
-#     logger.info("✅ خادم FastAPI يعمل. التهيئة (الاتصال بـ Telegram) جارية في الخلفية.")
-
 @app.get("/", response_class=HTMLResponse)
 async def root():
     """الصفحة الرئيسية"""