Update bot.py

bot.py

@@ -14,8 +14,9 @@ from hydrogram.types import (
 from fastapi import FastAPI
 
 # ============ CRITICAL FIX: Monkey-patch FloodPremiumWait ============
-# Older Hydrogram versions don't have FloodPremiumWait. Telegram sends it → AttributeError → infinite crash loop.
+import hydrogram
 import hydrogram.errors.rpc_error as _rpc_error
+
 if "FLOOD_PREMIUM_WAIT" not in [k for d in _rpc_error.exceptions.values() for k in d.keys()]:
     class FloodPremiumWait(FloodWait):
         CODE = 420
@@ -61,7 +62,6 @@ TARGET_CHAT = None
 
 # ============ Global Flood Control ============
 class GlobalFloodControl:
-    """Pause ALL workers when ANY worker hits a FloodWait."""
     def __init__(self):
         self.until = 0
         self._lock = asyncio.Lock()
@@ -84,7 +84,7 @@ flood_control = GlobalFloodControl()
 
 # ============ Throttle Controller ============
 class ThrottleController:
-    def __init__(self, rate=3.0):  # slightly slower to avoid bans
+    def __init__(self, rate=3.0):
         self.rate = rate
         self.last_send = {}
         self.lock = asyncio.Lock()
@@ -102,7 +102,7 @@ throttle = ThrottleController(rate=3.0)
 
 # ============ Locks ============
 download_lock = asyncio.Lock()
-send_sem = asyncio.Semaphore(1)  # reduce to 1 to be gentler on API
+send_sem = asyncio.Semaphore(1)
 
 # ============ قاعدة البيانات ============
 async def init_db():
@@ -167,7 +167,6 @@ async def add_job(message):
         file_id = message.animation.file_id
         file_size = message.animation.file_size or 0
     
-    # CRITICAL FIX: Explicit typing to prevent "unsupported type" SQLite errors
     chat_id = int(message.chat.id)
     message_id = int(message.id)
     file_size = int(file_size) if file_size else 0
@@ -194,7 +193,6 @@ async def add_job(message):
             logger.error(f"Add job error: {e}")
 
 async def update_job_status(chat_id, message_id, status, file_path=None, error=None):
-    # CRITICAL FIX: Explicit int typing
     chat_id = int(chat_id)
     message_id = int(message_id)
     
@@ -208,7 +206,7 @@ async def update_job_status(chat_id, message_id, status, file_path=None, error=N
             await db.execute("""
                 UPDATE jobs SET status=?, error=?, attempts=attempts+1, updated_at=CURRENT_TIMESTAMP 
                 WHERE chat_id=? AND message_id=?
-            """, (status, str(error)[:500], chat_id, message_id))  # cap error length
+            """, (status, str(error)[:500], chat_id, message_id))
         else:
             await db.execute("""
                 UPDATE jobs SET status=?, updated_at=CURRENT_TIMESTAMP 
@@ -306,7 +304,6 @@ async def verify_download(file_path, expected_size):
     return True
 
 async def download_with_message_object(chat_id, message_id, file_path):
-    """إعادة جلب الرسالة وتحميلها (الأكثر موثوقية)"""
     try:
         logger.info(f"Re-fetching message {chat_id}/{message_id}")
         message = await app.get_messages(int(chat_id), int(message_id))
@@ -323,8 +320,6 @@ async def download_with_message_object(chat_id, message_id, file_path):
         return None
 
 async def download_job(job):
-    """3 محاولات: file_id → message object → final retry"""
-    # CRITICAL FIX: Explicit int typing to prevent SQLite binding errors
     chat_id = int(job['chat_id'])
     message_id = int(job['message_id'])
     expected_size = int(job['file_size'] or 0)
@@ -344,14 +339,12 @@ async def download_job(job):
         except:
             pass
     
-    # Respect global flood waits before starting
     await flood_control.wait()
     
     try:
         await update_job_status(chat_id, message_id, 'downloading')
         downloaded = None
         
-        # المحاولة 1: file_id
         if file_id:
             logger.info(f"Attempt 1 (file_id): {message_id}")
             try:
@@ -362,12 +355,10 @@ async def download_job(job):
             except Exception as e:
                 logger.warning(f"file_id failed: {e}")
         
-        # المحاولة 2: message object
         if not await verify_download(downloaded, expected_size):
             logger.info(f"Attempt 2 (message): {message_id}")
             downloaded = await download_with_message_object(chat_id, message_id, file_path)
         
-        # المحاولة 3: إعادة أخيرة
         if not await verify_download(downloaded, expected_size):
             logger.info(f"Attempt 3 (retry): {message_id}")
             await asyncio.sleep(5)
@@ -402,7 +393,6 @@ async def download_job(job):
 
 # ============ الإرسال ============
 async def safe_send(send_func):
-    """Manual retry that RESPECTS FloodWait from Telegram."""
     for attempt in range(5):
         try:
             return await send_func()
@@ -432,7 +422,6 @@ async def send_job(job):
         await update_job_status(chat_id, message_id, 'pending', error="File not found")
         return False
     
-    # Respect global flood waits
     await flood_control.wait()
     
     try:
@@ -685,7 +674,6 @@ def run_gradio():
 
 # ============ Main ============
 async def main():
-    # 🔑 تأكد من عدم وجود اتصال سابق
     try:
         if app.is_connected:
             await app.stop()