Update bot.py

bot.py

@@ -1,23 +1,19 @@
 import os
 import asyncio
 import sqlite3
-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, RPCError
+from pyrogram.errors import FloodWait
+from pyrogram.enums import ChatType
 
-# --- 1. إعداد مساحات التخزين الدائمة ---
+# --- 1. إعداد مساحات التخزين وقاعدة البيانات ---
 DATA_DIR = "/data"
-DB_PATH = os.path.join(DATA_DIR, 'bot_memory.db')
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
+DB_PATH = os.path.join(DATA_DIR, 'bot_memory.db')
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
-# --- 2. إعداد قاعدة البيانات ---
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 cursor.execute('''CREATE TABLE IF NOT EXISTS processed (chat_id INTEGER, msg_id INTEGER, UNIQUE(chat_id, msg_id))''')
-cursor.execute('''CREATE TABLE IF NOT EXISTS processed_albums (album_id TEXT UNIQUE)''')
 conn.commit()
 
 def is_processed(chat_id, msg_id):
@@ -28,265 +24,118 @@ def mark_processed(chat_id, msg_id):
     cursor.execute("INSERT OR IGNORE INTO processed VALUES (?, ?)", (chat_id, msg_id))
     conn.commit()
 
-def is_album_processed(album_id):
-    cursor.execute("SELECT 1 FROM processed_albums WHERE album_id=?", (album_id,))
-    return cursor.fetchone() is not None
-
-def mark_album_processed(album_id):
-    cursor.execute("INSERT OR IGNORE INTO processed_albums VALUES (?)", (album_id,))
-    conn.commit()
-
-# --- 3. سيرفر الويب الوهمي للحفاظ على العمل 24/7 ---
-class RequestHandler(BaseHTTPRequestHandler):
-    def do_GET(self):
-        body = b"Ultimate Speed Archiver is Running 24/7!"
-        self.send_response(200)
-        self.send_header("Content-type", "text/html; charset=utf-8")
-        self.send_header("Content-Length", str(len(body)))
-        self.end_headers()
-        self.wfile.write(body)
-
-    def do_HEAD(self):
-        body = b"Ultimate Speed Archiver is Running 24/7!"
-        self.send_response(200)
-        self.send_header("Content-type", "text/html; charset=utf-8")
-        self.send_header("Content-Length", str(len(body)))
-        self.end_headers()
-
-    def log_message(self, format, *args): pass
-
-def start_web_server():
-    try:
-        HTTPServer(('0.0.0.0', 7860), RequestHandler).serve_forever()
-    except Exception as e:
-        print(f"⚠️ تجاوز خطأ السيرفر: {e}")
-
-Thread(target=start_web_server, daemon=True).start()
-
-# --- 4. إعدادات المتغيرات وتجهيز العميل ---
+# --- 2. إعدادات الحساب ---
 API_ID = int(os.environ.get("API_ID", 0))
 API_HASH = os.environ.get("API_HASH", "")
 SESSION_STRING = os.environ.get("SESSION_STRING", "")
+TARGET_CHAT = int(os.environ.get("TARGET_CHAT", 0))
 
-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,
-    api_hash=API_HASH,
+    "restricted_bypass_bot", 
+    api_id=API_ID, 
+    api_hash=API_HASH, 
     session_string=SESSION_STRING,
-    sleep_threshold=300,
-    max_concurrent_transmissions=5,
-    workers=10
+    max_concurrent_transmissions=2 
 )
 
-download_queue = asyncio.Queue()
-upload_queue = asyncio.Queue()
+process_queue = asyncio.Queue()
 
-# --- 5. الرادار المباشر + تأخير الألبومات ---
+# --- 3. صائد الرسائل الجديدة ---
 @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))
-            asyncio.create_task(queue_album_after_delay(message))
-        return
-
     if not is_processed(message.chat.id, message.id):
-        await download_queue.put({'type': 'single', 'message': message})
-
-async def queue_album_after_delay(message):
-    await asyncio.sleep(4)  # مهلة لتجميع كافة أجزاء الألبوم من سيرفرات تيليجرام
-    await download_queue.put({'type': 'album', 'message': message})
+        await process_queue.put(message)
 
-# --- 6. حاصد الأرشيف المفقود (Backfill History Scanner) ---
+# --- 4. حاصد الأرشيف القديم (لكل القنوات المشترك بها) ---
 async def scan_old_history():
-    if not SOURCE_CHAT:
-        print("ℹ️ لم يتم تحديد SOURCE_CHAT في المتغيرات، سيتم الاعتماد على الرصد المباشر فقط.")
-        return
-
-    print(f"🔎 [جيمي] بدء مسح الأرشيف القديم للقناة المصدر: {SOURCE_CHAT}...")
-    count = 0
+    print("🔎 [جيمي] بدء مسح أرشيف القنوات والمجموعات...")
     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} عنصر قديم في طابور التحميل.")
+        async for dialog in app.get_dialogs():
+            if dialog.chat.type in [ChatType.CHANNEL, ChatType.SUPERGROUP, ChatType.GROUP] and dialog.chat.id != TARGET_CHAT:
+                print(f"📂 فحص: {dialog.chat.title}...")
+                try:
+                    async for message in app.get_chat_history(dialog.chat.id):
+                        if message.media and not is_processed(message.chat.id, message.id):
+                            await process_queue.put(message)
+                except Exception as e:
+                    print(f"⚠️ تجاوز {dialog.chat.title} بسبب خطأ: {e}")
+                
+                await asyncio.sleep(3) # فاصل بين فحص كل قناة وأخرى
+        print("✅ اكتمل سحب كافة الرسائل المفقودة إلى الطابور!")
     except Exception as e:
-        print(f"❌ خطأ أثناء مسح الأرشيف: {e}")
+        print(f"❌ خطأ عام بالمسح: {e}")
 
-# --- 7. كتيبة التحميل إلى التخزين الدائم ---
-async def downloader_worker(worker_id):
+# --- 5. محرك الكسر والرفع (العملية المتسلسلة) ---
+async def restricted_bypass_worker():
     while True:
-        task = await download_queue.get()
-        task_type = task['type']
-        message = task['message']
+        message = await process_queue.get()
         
+        # تجنب معالجة ما تم إنجازه
+        if is_processed(message.chat.id, message.id):
+            process_queue.task_done()
+            continue
+            
+        file_path = None
         try:
-            if task_type == 'album':
-                album_msgs = await app.get_media_group(message.chat.id, message.id)
-                downloaded_media = []
-                
-                for msg in album_msgs:
-                    file_path = await msg.download(file_name=f"{DOWNLOAD_DIR}/")
-                    if file_path:
-                        downloaded_media.append({'path': file_path, 'msg': msg})
-                
-                if downloaded_media:
-                    await upload_queue.put({'type': 'album', 'items': downloaded_media})
-                    
-            elif task_type == 'single':
-                if is_processed(message.chat.id, message.id):
-                    download_queue.task_done()
-                    continue
-                    
-                print(f"⚡ [محمل {worker_id}] جارٍ تحميل الملف (ID: {message.id})...")
-                file_path = await message.download(file_name=f"{DOWNLOAD_DIR}/")
-                
-                if file_path:
-                    await upload_queue.put({'type': 'single', 'file_path': file_path, 'message': message})
-                    
-        except Exception as e:
-            print(f"❌ خطأ بالتحميل: {e}")
-        finally:
-            download_queue.task_done()
-
-# --- 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)
+            print(f"⬇️ جاري سحب الملف (ID: {message.id}) من قن��ة مقفلة...")
+            
+            # خطوة 1: التحميل الفعلي
+            file_path = await message.download(file_name=f"{DOWNLOAD_DIR}/")
+            
+            if not file_path:
+                print("⚠️ فشل التحميل، سيتم التخطي.")
+                continue
 
-async def uploader_worker():
-    while True:
-        task = await upload_queue.get()
-        task_type = task['type']
-        
-        try:
-            if task_type == 'album':
-                media_group = []
-                files_to_remove = []
-                
-                for item in task['items']:
-                    file_path = item['path']
-                    msg = item['msg']
-                    files_to_remove.append(file_path)
-                    
-                    caption = msg.caption or ""
-                    entities = msg.caption_entities
-                    
-                    if msg.photo:
-                        media_group.append(InputMediaPhoto(media=file_path, caption=caption, caption_entities=entities))
-                    elif msg.video:
-                        media_group.append(InputMediaVideo(media=file_path, caption=caption, caption_entities=entities, supports_streaming=True))
-                    elif msg.audio:
-                        media_group.append(InputMediaAudio(media=file_path, caption=caption, caption_entities=entities))
-                    elif msg.document:
-                        media_group.append(InputMediaDocument(media=file_path, caption=caption, caption_entities=entities))
-                
-                if media_group:
-                    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)
+            print(f"⬆️ جاري الرفع إلى قناتك...")
+            
+            # خطوة 2: الرفع الفعلي
+            caption = message.caption or ""
+            if message.photo:
+                await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption)
+            elif message.video:
+                await app.send_video(TARGET_CHAT, video=file_path, caption=caption, supports_streaming=True)
+            elif message.document:
+                await app.send_document(TARGET_CHAT, document=file_path, caption=caption)
+            else:
+                await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption) # للأنواع الأخرى
                 
-                for item in task['items']:
-                    mark_processed(item['msg'].chat.id, item['msg'].id)
-                    if os.path.exists(item['path']): os.remove(item['path'])
+            # خطوة 3: التوثيق في قاعدة البيانات
+            mark_processed(message.chat.id, message.id)
+            print(f"✅ تم النقل بنجاح!")
+            
+            # خطوة 4: الفاصل الزمني الإجباري (لا تقم بتعديله، فهو درع الحماية من الحظر)
+            print("⏳ استراحة أمنية لمدة 12 ثانية...")
+            await asyncio.sleep(12) 
 
-            elif task_type == 'single':
-                file_path = task['file_path']
-                message = task['message']
-                
-                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 process_queue.put(message) # إعادة إدراج الرسالة كي لا تضيع
             
         except Exception as e:
-            print(f"❌ خطأ غير متوقع أثناء الرفع: {e}")
+            print(f"❌ خطأ أثناء معالجة الملف: {e}")
             
         finally:
-            upload_queue.task_done()
-            await asyncio.sleep(2.5)
-
-# --- 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)} ملف معلق في القرص، جاري إنقاذها...")
-        for filename in files:
-            file_path = os.path.join(DOWNLOAD_DIR, filename)
-            try:
-                await app.send_document(TARGET_CHAT, document=file_path, caption="♻️ ملف مسترجع تلقائياً")
-                os.remove(file_path)
-                await asyncio.sleep(3)
-            except FloodWait as e:
-                await asyncio.sleep(e.value + 5)
-            except Exception as e:
-                print(f"❌ تعذر إنقاذ {filename}: {e}")
-
-# --- 10. التشغيل الرئيسي ---
+            # خطوة 5: حذف الملف فوراً لتجنب امتلاء السيرفر
+            if file_path and os.path.exists(file_path):
+                try:
+                    os.remove(file_path)
+                except:
+                    pass
+            process_queue.task_done()
+
+# --- 6. التشغيل ---
 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}")
-                break
-    except Exception:
-        pass
-
-    await recover_orphaned_files()
+    # عامل رفع واحد فقط لضمان الترتيب وعدم الضغط على سيرفر تيليجرام
+    asyncio.create_task(restricted_bypass_worker())
     
-    # تشغيل العمال
-    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("🤖 [جيمي] البوت بآلية الفحص الشاملة يعمل الآن بكفاءة!")
     await idle()
     await app.stop()