Update bot.py

bot.py

@@ -5,13 +5,11 @@ from threading import Thread
 from http.server import BaseHTTPRequestHandler, HTTPServer
 from pyrogram import Client, filters, idle
 from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaDocument, InputMediaAudio
-from pyrogram.errors import FloodWait
+from pyrogram.errors import FloodWait, RPCError
 
-# --- 1. إعداد مساحات التخزين (تم العودة للقرص الدائم لحماية الملفات وقت الحظر) ---
+# --- 1. إعداد مساحات التخزين الدائمة ---
 DATA_DIR = "/data"
 DB_PATH = os.path.join(DATA_DIR, 'bot_memory.db')
-
-# استخدام التخزين الدائم لضمان عدم ضياع أي مقطع إذا انطفأ السيرفر أثناء حظر تيليجرام
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
@@ -38,7 +36,7 @@ def mark_album_processed(album_id):
     cursor.execute("INSERT OR IGNORE INTO processed_albums VALUES (?)", (album_id,))
     conn.commit()
 
-# --- 3. سيرفر الويب الوهمي (متوافق 100% مع UptimeRobot) ---
+# --- 3. سيرفر الويب الوهمي للحفاظ على العمل 24/7 ---
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         body = b"Ultimate Speed Archiver is Running 24/7!"
@@ -65,7 +63,7 @@ def start_web_server():
 
 Thread(target=start_web_server, daemon=True).start()
 
-# --- 4. إعداد البوت مع خيارات السرعة المتوازية ---
+# --- 4. إعدادات المتغيرات وتجهيز العميل ---
 API_ID = int(os.environ.get("API_ID", 0))
 API_HASH = os.environ.get("API_HASH", "")
 SESSION_STRING = os.environ.get("SESSION_STRING", "")
@@ -73,6 +71,9 @@ SESSION_STRING = os.environ.get("SESSION_STRING", "")
 target_env = os.environ.get("TARGET_CHAT", "")
 TARGET_CHAT = int(target_env) if target_env.lstrip('-').isdigit() else target_env
 
+source_env = os.environ.get("SOURCE_CHAT", "")
+SOURCE_CHAT = int(source_env) if source_env.lstrip('-').isdigit() else source_env
+
 app = Client(
     "my_userbot",
     api_id=API_ID,
@@ -86,26 +87,48 @@ app = Client(
 download_queue = asyncio.Queue()
 upload_queue = asyncio.Queue()
 
-# --- 5. الرادار الذكي: التقاط الملفات والألبومات بدون تفويت ---
+# --- 5. الرادار المباشر + تأخير الألبومات ---
 @app.on_message(filters.incoming & ~filters.me & (filters.channel | filters.group) & filters.media)
 async def catch_new_messages(client, message):
     if message.media_group_id:
         if not is_album_processed(str(message.media_group_id)):
             mark_album_processed(str(message.media_group_id))
-            print(f"🚨 [رصد ألبوم] تجميع الأجزاء... (ننتظر 3 ثوانٍ لضمان وصول كل المقاطع)")
-            # تشغيل مهمة جانبية للانتظار حتى لا يتعطل الرادار عن باقي الرسائل
             asyncio.create_task(queue_album_after_delay(message))
         return
 
     if not is_processed(message.chat.id, message.id):
-        print(f"🚨 [رصد مفرد] تم التقاط ملف من: {message.chat.title}")
         await download_queue.put({'type': 'single', 'message': message})
 
 async def queue_album_after_delay(message):
-    await asyncio.sleep(3) # التحديث الأهم: تأخير مقصود لتكتمل حزمة الألبوم في سيرفر تيليجرام
+    await asyncio.sleep(4)  # مهلة لتجميع كافة أجزاء الألبوم من سيرفرات تيليجرام
     await download_queue.put({'type': 'album', 'message': message})
 
-# --- 6. كتيبة التحميل (حفظ بالقرص الدائم) ---
+# --- 6. حاصد الأرشيف المفقود (Backfill History Scanner) ---
+async def scan_old_history():
+    if not SOURCE_CHAT:
+        print("ℹ️ لم يتم تحديد SOURCE_CHAT في المتغيرات، سيتم الاعتماد على الرصد المباشر فقط.")
+        return
+
+    print(f"🔎 [جيمي] بدء مسح الأرشيف القديم للقناة المصدر: {SOURCE_CHAT}...")
+    count = 0
+    try:
+        # جلب تاريخ القناة من الأقدم إلى الأحدث
+        async for message in app.get_chat_history(SOURCE_CHAT):
+            if message.media:
+                if message.media_group_id:
+                    if not is_album_processed(str(message.media_group_id)):
+                        mark_album_processed(str(message.media_group_id))
+                        await download_queue.put({'type': 'album', 'message': message})
+                        count += 1
+                else:
+                    if not is_processed(message.chat.id, message.id):
+                        await download_queue.put({'type': 'single', 'message': message})
+                        count += 1
+        print(f"✅ اكتمل مسح الأرشيف! تم إدراج {count} عنصر قديم في طابور التحميل.")
+    except Exception as e:
+        print(f"�� خطأ أثناء مسح الأرشيف: {e}")
+
+# --- 7. كتيبة التحميل إلى التخزين الدائم ---
 async def downloader_worker(worker_id):
     while True:
         task = await download_queue.get()
@@ -121,7 +144,6 @@ async def downloader_worker(worker_id):
                     file_path = await msg.download(file_name=f"{DOWNLOAD_DIR}/")
                     if file_path:
                         downloaded_media.append({'path': file_path, 'msg': msg})
-                        mark_processed(msg.chat.id, msg.id)
                 
                 if downloaded_media:
                     await upload_queue.put({'type': 'album', 'items': downloaded_media})
@@ -131,7 +153,7 @@ async def downloader_worker(worker_id):
                     download_queue.task_done()
                     continue
                     
-                print(f"⚡ [محمل {worker_id}] جارٍ تحميل الملف الدائم (ID: {message.id})...")
+                print(f"⚡ [محمل {worker_id}] جارٍ تحميل الملف (ID: {message.id})...")
                 file_path = await message.download(file_name=f"{DOWNLOAD_DIR}/")
                 
                 if file_path:
@@ -142,7 +164,23 @@ async def downloader_worker(worker_id):
         finally:
             download_queue.task_done()
 
-# --- 7. عامل الرفع الصبور (يحترم الحظر ويحفظ الملفات) ---
+# --- 8. عامل الرفع مع آلية التفكيك الاحتياطية عند الفشل ---
+async def send_single_item(file_path, message):
+    caption = message.caption or ""
+    entities = message.caption_entities
+    if message.photo:
+        await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption, caption_entities=entities)
+    elif message.video:
+        await app.send_video(TARGET_CHAT, video=file_path, caption=caption, caption_entities=entities, supports_streaming=True)
+    elif message.animation:
+        await app.send_animation(TARGET_CHAT, animation=file_path, caption=caption, caption_entities=entities)
+    elif message.audio:
+        await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption, caption_entities=entities)
+    elif message.voice:
+        await app.send_voice(TARGET_CHAT, voice=file_path, caption=caption, caption_entities=entities)
+    else:
+        await app.send_document(TARGET_CHAT, document=file_path, caption=caption, caption_entities=entities)
+
 async def uploader_worker():
     while True:
         task = await upload_queue.get()
@@ -171,103 +209,84 @@ async def uploader_worker():
                         media_group.append(InputMediaDocument(media=file_path, caption=caption, caption_entities=entities))
                 
                 if media_group:
-                    print("🚀 رفع ألبوم متكامل...")
-                    # رفع الألبوم (تيليجرام قد يفصل المستندات عن الصور، Pyrogram يتعامل معها)
-                    await app.send_media_group(TARGET_CHAT, media=media_group)
-                    print("✅ الألبوم ارتفع بنجاح!")
+                    try:
+                        print("🚀 جارٍ رفع الألبوم كدفعة واحدة...")
+                        await app.send_media_group(TARGET_CHAT, media=media_group)
+                        print("✅ تم رفع الألبوم بنجاح!")
+                    except RPCError as e:
+                        print(f"⚠️ فشل رفع الألبوم كمجموعة واحدة ({e}). جارٍ التفكيك والرفع منفصلاً لمنع الفقدان...")
+                        for item in task['items']:
+                            await send_single_item(item['path'], item['msg'])
+                            await asyncio.sleep(2)
                 
-                # لا نحذف الملفات إلا بعد نجاح الرفع 100%
-                for fp in files_to_remove:
-                    if os.path.exists(fp): os.remove(fp)
+                for item in task['items']:
+                    mark_processed(item['msg'].chat.id, item['msg'].id)
+                    if os.path.exists(item['path']): os.remove(item['path'])
 
             elif task_type == 'single':
                 file_path = task['file_path']
                 message = task['message']
-                caption = message.caption or ""
-                entities = message.caption_entities
                 
-                print(f"🚀 رفع ملف مفرد...")
-                if message.photo: 
-                    await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption, caption_entities=entities)
-                elif message.video: 
-                    await app.send_video(
-                        TARGET_CHAT, 
-                        video=file_path, 
-                        caption=caption, 
-                        caption_entities=entities,
-                        duration=message.video.duration,
-                        width=message.video.width,
-                        height=message.video.height,
-                        supports_streaming=True
-                    )
-                elif message.animation: 
-                    await app.send_animation(TARGET_CHAT, animation=file_path, caption=caption, caption_entities=entities)
-                elif message.audio: 
-                    await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption, caption_entities=entities)
-                elif message.voice: 
-                    await app.send_voice(TARGET_CHAT, voice=file_path, caption=caption, caption_entities=entities)
-                else: 
-                    await app.send_document(TARGET_CHAT, document=file_path, caption=caption, caption_entities=entities)
+                print(f"🚀 رفع ملف منفرد (ID: {message.id})...")
+                await send_single_item(file_path, message)
                 
                 mark_processed(message.chat.id, message.id)
                 print(f"✅ تم الرفع بنجاح!")
                 if os.path.exists(file_path): os.remove(file_path)
                 
         except FloodWait as e:
-            # الخدعة الذهبية: نعيد المهمة للطابور، وننام، والملف بأمان في الهارد دسك!
-            print(f"⏳ [حظر إرسال] تيليجرام مقفل مؤقتاً. سننتظر {e.value} ثانية. الملفات مخزنة بأمان ولن تضيع!")
+            print(f"⏳ [حظر إرسال] انتظار {e.value} ثانية... الملفات محفوظة ولن تضيع.")
             await asyncio.sleep(e.value + 5)
-            await upload_queue.put(task) 
+            await upload_queue.put(task)
             
         except Exception as e:
-            print(f"❌ خطأ غير متوقع بالرفع: {e}")
-            # في حال وجود خطأ آخر غير الحظر، نحتفظ بالملف ليتم استرجاعه عند إعادة التشغيل
+            print(f"❌ خطأ غير متوقع أثناء الرفع: {e}")
             
         finally:
             upload_queue.task_done()
-            await asyncio.sleep(2)
+            await asyncio.sleep(2.5)
 
-# --- 8. المنقذ: استرجاع ملفات ما قبل الانطفاء/الحظر ---
+# --- 9. استرجاع الملفات المعلقة عند إعادة التشغيل ---
 async def recover_orphaned_files():
     if not os.path.exists(DOWNLOAD_DIR): return
     files = [f for f in os.listdir(DOWNLOAD_DIR) if os.path.isfile(os.path.join(DOWNLOAD_DIR, f))]
     if files:
-        print(f"🔍 [عملية إنقاذ] تم العثور على {len(files)} ملفات مخزنة من الجلسة السابقة بسبب حظر أو انطفاء!")
+        print(f"🔍 [استرجاع] العثور على {len(files)} ملف معلق في القرص، جاري إنقاذها...")
         for filename in files:
             file_path = os.path.join(DOWNLOAD_DIR, filename)
             try:
-                # نرفع الملفات المعلقة كمستندات لضمان عدم ضياعها
-                await app.send_document(TARGET_CHAT, document=file_path, caption="♻️ ملف مسترجع من الذاكرة الاحتياطية (كان معلقاً)")
+                await app.send_document(TARGET_CHAT, document=file_path, caption="♻️ ملف مسترجع تلقائياً")
                 os.remove(file_path)
-                await asyncio.sleep(3) # تجنب الحظر أثناء الاسترجاع
+                await asyncio.sleep(3)
             except FloodWait as e:
-                print(f"⏳ لا يزال الحظر مستمراً، ننتظر {e.value} ثانية...")
                 await asyncio.sleep(e.value + 5)
             except Exception as e:
-                print(f"❌ فشل إنقاذ الملف {filename}: {e}")
+                print(f"❌ تعذر إنقاذ {filename}: {e}")
 
-# --- 9. التشغيل المركزي ---
+# --- 10. التشغيل الرئيسي ---
 async def main():
     await app.start()
     
-    print("🔄 تنشيط ذاكرة الحساب لجلب الأكسس هاش...")
+    print("🔄 جلب بيانات القنوات والاتصال...")
     try:
         async for dialog in app.get_dialogs():
             if dialog.chat.id == TARGET_CHAT:
-                print(f"✅ تم التعرف على القناة الهدف: {dialog.chat.title}")
+                print(f"✅ القناة الهدف جاهزة: {dialog.chat.title}")
                 break
-    except Exception as e:
+    except Exception:
         pass
 
-    # فور التشغيل، ابحث عن أي ملفات كانت معلقة بسبب حظر وارفعها
     await recover_orphaned_files()
     
+    # تشغيل العمال
     for i in range(1, 4):
         asyncio.create_task(downloader_worker(i))
-        
     asyncio.create_task(uploader_worker())
+
+    # تشغيل مسح الأرشيف القديم لضمان جلب الفيديوهات الـ 116 السابقة
+    asyncio.create_task(scan_old_history())
     
-    print("🤖 جي��ي: صائد الملفات المدرّع يعمل الآن! لا ضياع بعد اليوم...")
+    print("🤖 [جيمي] البوت بآلية الفحص الشاملة يعمل الآن بكفاءة!")
     await idle()
     await app.stop()