Update bot.py

bot.py

@@ -19,7 +19,7 @@ os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
 HEALTHCHECK_URL = os.environ.get("HEALTHCHECK_URL", "")
 MAX_FILE_SIZE = 2000 * 1024 * 1024 # 2GB
-DYNAMIC_UPLOAD_DELAY = 3 # تم تقليل وقت الانتظار الافتراضي
+DYNAMIC_UPLOAD_DELAY = 3 
 download_queue = asyncio.Queue()
 
 # --- 2. سيرفر الحفاظ على التشغيل ---
@@ -32,7 +32,7 @@ class RequestHandler(BaseHTTPRequestHandler):
 
 Thread(target=lambda: HTTPServer(('0.0.0.0', 7860), RequestHandler).serve_forever(), daemon=True).start()
 
-# --- 3. إعدادات الحساب (محاكاة هاتف حقيقي) ---
+# --- 3. إعدادات الحساب ---
 API_ID = int(os.environ.get("API_ID", 0))
 API_HASH = os.environ.get("API_HASH", "")
 SESSION_STRING = os.environ.get("SESSION_STRING", "")
@@ -44,7 +44,6 @@ app = Client(
     api_hash=API_HASH,
     session_string=SESSION_STRING,
     in_memory=True,
-    # التخفي كأنه هاتف حقيقي لتجنب الحظر
     device_model="Samsung Galaxy S23 Ultra",
     system_version="Android 14.0",
     app_version="10.14.1"
@@ -91,7 +90,7 @@ def get_media_details(message):
     file_size = getattr(media, "file_size", 0)
     return unique_id, m_type, file_size
 
-# --- 5. محرك التنزيل المتوازي ---
+# --- 5. محرك التنزيل المتوازي (تم تحديثه لحل مشكلة الامتداد) ---
 async def download_worker_task(worker_id):
     while True:
         message = await download_queue.get()
@@ -107,12 +106,13 @@ async def download_worker_task(worker_id):
                 async with db.execute("SELECT id FROM queue WHERE file_unique_id=?", (unique_id,)) as cursor:
                     if await cursor.fetchone():
                         continue
-
-            unique_name = f"{DOWNLOAD_DIR}/{message.chat.id}_{message.id}"
+            
+            # التعديل هنا: إضافة ( _ ) في النهاية تجعل بايروجرام يضيف امتداد الملف الصحيح تلقائياً (.jpg, .mp4)
+            file_prefix = f"{DOWNLOAD_DIR}/{message.chat.id}_{message.id}_"
             
             for attempt in range(3):
                 try:
-                    file_path = await asyncio.wait_for(message.download(file_name=unique_name), timeout=300)
+                    file_path = await asyncio.wait_for(message.download(file_name=file_prefix), timeout=300)
                     if file_path and os.path.exists(file_path) and os.path.getsize(file_path) > 0:
                         group_id = str(message.media_group_id) if message.media_group_id else None
                         src_title = message.chat.title or "المصدر"
@@ -136,7 +136,7 @@ async def download_worker_task(worker_id):
         finally:
             download_queue.task_done()
 
-# --- 6. صائد الرسائل الحية (الرادار) ---
+# --- 6. صائد الرسائل الحية ---
 @app.on_message(filters.incoming & ~filters.me & (filters.channel | filters.group) & filters.media)
 async def catch_incoming(client, message):
     if message.chat.id != TARGET_CHAT:
@@ -163,7 +163,7 @@ async def status_command(client, message):
     )
     await message.reply_text(report)
 
-# --- 8. محرك الرفع التسلسلي الصارم ---
+# --- 8. محرك الرفع (تم تحديثه لمنع الدوامة اللانهائية) ---
 async def upload_worker():
     global DYNAMIC_UPLOAD_DELAY
     while True:
@@ -248,8 +248,13 @@ async def upload_worker():
             DYNAMIC_UPLOAD_DELAY += 2
             await asyncio.sleep(e.value + 5)
         except Exception as e:
-            print(f"⚠️ خطأ أثناء الرفع: {e}")
-            await asyncio.sleep(5)
+            # التعديل هنا: إذا حدث خطأ غير طبيعي، نتجاوز الملف حتى لا يعلق السكربت
+            print(f"⚠️ خطأ أثناء الرفع، سيتم تخطي الملف: {e}")
+            async with aiosqlite.connect(DB_PATH) as db:
+                await db.execute("UPDATE queue SET status='FAILED' WHERE id=?", (q_id,))
+                await db.commit()
+            if os.path.exists(file_path): os.remove(file_path)
+            await asyncio.sleep(2)
 
 async def healthcheck_ping():
     while True:
@@ -258,7 +263,6 @@ async def healthcheck_ping():
             try: urllib.request.urlopen(HEALTHCHECK_URL, timeout=10)
             except: pass
 
-# --- التحديث هنا: دالة main لتعريف القنوات وتجنب الأخطاء ---
 async def main():
     await init_db()
     await app.start()