Update bot.py

bot.py

@@ -1,6 +1,7 @@
 import os
 import asyncio
 import sqlite3
+import time
 from threading import Thread
 from http.server import BaseHTTPRequestHandler, HTTPServer
 from pyrogram import Client, filters, idle
@@ -8,16 +9,21 @@ from pyrogram.errors import FloodWait
 from pyrogram.enums import ChatType
 from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaDocument, InputMediaAudio
 
-# --- 1. إعداد مساحات التخزين وقاعدة البيانات المستدامة ---
+# --- 1. إعداد مساحات التخزين والتنظيف الأولية ---
 DATA_DIR = "./data"
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
 DB_PATH = os.path.join(DATA_DIR, 'bot_archiver.db')
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
+# تنظيف الملفات المؤقتة والعالقة من التشغيل السابق
+for f in os.listdir(DOWNLOAD_DIR):
+    try: os.remove(os.path.join(DOWNLOAD_DIR, f))
+    except: pass
+
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 
-# جدول إدارة الطابور وحفظ الألبومات
+# إنشاء الجداول بحقول المحاذاة الزمنية
 cursor.execute('''
 CREATE TABLE IF NOT EXISTS queue (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -28,18 +34,19 @@ CREATE TABLE IF NOT EXISTS queue (
     caption TEXT,
     file_path TEXT,
     status TEXT DEFAULT 'DOWNLOADED',
-    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    timestamp REAL,
     UNIQUE(chat_id, msg_id)
 )
 ''')
+cursor.execute('''CREATE TABLE IF NOT EXISTS scan_progress (chat_id INTEGER PRIMARY KEY, last_msg_id INTEGER)''')
 conn.commit()
 
-# --- 2. سيرفر الويب لمنع إغلاق منصة Hugging Face ---
+# --- 2. سيرفر الويب الوهمي ---
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         self.send_response(200)
         self.end_headers()
-        self.wfile.write(b"Dual-Engine Telegram Archiver Active!")
+        self.wfile.write(b"Super-Engine Archiver Active!")
     def log_message(self, format, *args): pass
 
 Thread(target=lambda: HTTPServer(('0.0.0.0', 7860), RequestHandler).serve_forever(), daemon=True).start()
@@ -56,22 +63,21 @@ app = Client(
     api_hash=API_HASH,
     session_string=SESSION_STRING,
     in_memory=True,
-    max_concurrent_transmissions=3 # تسريع التحميل المتوازي
+    max_concurrent_transmissions=2
 )
 
-# --- 4. دالة التنزيل الآمن مع إعادة المحاولة لكسر خطأ 0 بايت ---
+# --- 4. دالة التنزيل الآمن مع كسر الـ 0 بايت وحفظ التوقيت ---
 async def download_media_safe(message):
     cursor.execute("SELECT status FROM queue WHERE chat_id=? AND msg_id=?", (message.chat.id, message.id))
     if cursor.fetchone():
-        return None # تم تنزيله سابقاً
+        return None
 
     unique_name = f"{DOWNLOAD_DIR}/{message.chat.id}_{message.id}"
     
-    for attempt in range(3): # 3 محاولات تنزيل عند حدوث خلل شبكي
+    for attempt in range(3):
         try:
             file_path = await message.download(file_name=unique_name)
             if file_path and os.path.exists(file_path) and os.path.getsize(file_path) > 0:
-                # تحديد نوع الوسائط
                 m_type = "doc"
                 if message.photo: m_type = "photo"
                 elif message.video: m_type = "video"
@@ -81,64 +87,73 @@ async def download_media_safe(message):
                 caption = message.caption or ""
 
                 cursor.execute(
-                    "INSERT OR IGNORE INTO queue (chat_id, msg_id, media_group_id, media_type, caption, file_path) VALUES (?, ?, ?, ?, ?, ?)",
-                    (message.chat.id, message.id, group_id, m_type, caption, file_path)
+                    "INSERT OR IGNORE INTO queue (chat_id, msg_id, media_group_id, media_type, caption, file_path, timestamp) VALUES (?, ?, ?, ?, ?, ?, ?)",
+                    (message.chat.id, message.id, group_id, m_type, caption, file_path, time.time())
                 )
                 conn.commit()
-                print(f"⚡ [صائد لحظي] تم سحب وحفظ الملف: ID {message.id}")
+                print(f"⚡ [صائد لحظي] تم تنزيل الملف بنجاح: ID {message.id}")
                 return file_path
             else:
-                if file_path and os.path.exists(file_path):
-                    os.remove(file_path)
+                if file_path and os.path.exists(file_path): os.remove(file_path)
         except Exception as e:
-            print(f"⚠️ محاولة تنزيل فاشلة ({attempt+1}/3) للرسالة {message.id}: {e}")
             await asyncio.sleep(2)
             
     return None
 
-# --- 5. صائد الرسائل اللحظي (التقاط فوري قبل الحذف) ---
+# --- 5. صائد الرسائل اللحظي ---
 @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:
         await download_media_safe(message)
 
-# --- 6. حاصد السجل القديم ---
+# --- 6. حاصد الأرشيف الذكي (يستكمل من حيث توقف) ---
 async def scan_history_worker():
-    print("🔎 بدء مسح السجل القديم للمجموعات والقنوات...")
+    print("🔎 بدء مسح الأرشيف القديم...")
     async for dialog in app.get_dialogs():
         if dialog.chat.type in [ChatType.CHANNEL, ChatType.SUPERGROUP, ChatType.GROUP] and dialog.chat.id != TARGET_CHAT:
             try:
+                cursor.execute("SELECT last_msg_id FROM scan_progress WHERE chat_id=?", (dialog.chat.id,))
+                res = cursor.fetchone()
+                last_id = res[0] if res else 0
+
                 async for message in app.get_chat_history(dialog.chat.id):
+                    if message.id <= last_id:
+                        break # الوصول إلى ما تم فاعلياً سابقاً
                     if message.media:
                         await download_media_safe(message)
+                    
+                    cursor.execute("INSERT OR REPLACE INTO scan_progress (chat_id, last_msg_id) VALUES (?, ?)", (dialog.chat.id, message.id))
+                    conn.commit()
+
             except Exception as e:
-                print(f"تجاوز القناة {dialog.chat.title} بسبب: {e}")
-            await asyncio.sleep(5)
-    print("✅ اكتمل مسح كافة الأرشيفات القديمة.")
+                print(f"تجاوز {dialog.chat.title}: {e}")
+            await asyncio.sleep(4)
+    print("✅ تم استكمال مسح جميع الأرشيفات.")
 
-# --- 7. محرك الرفع الذكي وتجميع الألبومات (Uploader & Album Engine) ---
+# --- 7. محرك الرفع الذكي (مع مؤقت تجميع الألبومات) ---
 async def upload_worker():
     while True:
         try:
-            # البحث عن أقدم عنصر نزل ولم يرفع بعد
-            cursor.execute("SELECT id, chat_id, msg_id, media_group_id, media_type, caption, file_path FROM queue WHERE status='DOWNLOADED' ORDER BY id ASC LIMIT 1")
+            cursor.execute("SELECT id, chat_id, msg_id, media_group_id, media_type, caption, file_path, timestamp FROM queue WHERE status='DOWNLOADED' ORDER BY id ASC LIMIT 1")
             row = cursor.fetchone()
 
             if not row:
                 await asyncio.sleep(5)
                 continue
 
-            q_id, chat_id, msg_id, group_id, m_type, caption, file_path = row
+            q_id, chat_id, msg_id, group_id, m_type, caption, file_path, ts = row
 
-            # التأكد من وجود الملف على القرص
             if not os.path.exists(file_path):
                 cursor.execute("UPDATE queue SET status='FAILED' WHERE id=?", (q_id,))
                 conn.commit()
                 continue
 
-            # معالجة الألبومات (Media Groups)
+            # معالجة الألبومات الذكية بمهلة انتظار (15 ثانية للتأكد من وصول كافة أجزاء الألبوم)
             if group_id:
-                await asyncio.sleep(4) # مهلة زمنية للتأكد من اكتمال تنزيل جميع عناصر الألبوم
+                time_passed = time.time() - ts
+                if time_passed < 15:
+                    await asyncio.sleep(15 - time_passed)
+
                 cursor.execute(
                     "SELECT id, media_type, caption, file_path FROM queue WHERE chat_id=? AND media_group_id=? AND status='DOWNLOADED' ORDER BY msg_id ASC",
                     (chat_id, group_id)
@@ -150,28 +165,22 @@ async def upload_worker():
                     ids_to_update = []
                     files_to_delete = []
 
-                    for idx, g_item in enumerate(group_rows[:10]): # تيليجرام يدعم حتى 10 وسائط في الألبوم الواحد
+                    for idx, g_item in enumerate(group_rows[:10]):
                         g_id, g_type, g_cap, g_path = g_item
-                        # وضع النص الوصفي على العنصر الأول فقط في الألبوم
                         item_caption = g_cap if idx == 0 else ""
 
-                        if g_type == "photo":
-                            media_list.append(InputMediaPhoto(media=g_path, caption=item_caption))
-                        elif g_type == "video":
-                            media_list.append(InputMediaVideo(media=g_path, caption=item_caption, supports_streaming=True))
-                        elif g_type == "doc":
-                            media_list.append(InputMediaDocument(media=g_path, caption=item_caption))
-                        elif g_type == "audio":
-                            media_list.append(InputMediaAudio(media=g_path, caption=item_caption))
+                        if g_type == "photo": media_list.append(InputMediaPhoto(media=g_path, caption=item_caption))
+                        elif g_type == "video": media_list.append(InputMediaVideo(media=g_path, caption=item_caption, supports_streaming=True))
+                        elif g_type == "doc": media_list.append(InputMediaDocument(media=g_path, caption=item_caption))
+                        elif g_type == "audio": media_list.append(InputMediaAudio(media=g_path, caption=item_caption))
 
                         ids_to_update.append(g_id)
                         files_to_delete.append(g_path)
 
                     if media_list:
-                        print(f"📦 جاري رفع ألبوم يحتوي على {len(media_list)} عناصر...")
+                        print(f"📦 جاري رفع ألبوم كاشمل لـ {len(media_list)} عناصر...")
                         await app.send_media_group(TARGET_CHAT, media=media_list)
                         
-                        # تحديث الحالة وحذف الملفات بعد نجاح الرفع
                         for item_id in ids_to_update:
                             cursor.execute("UPDATE queue SET status='UPLOADED' WHERE id=?", (item_id,))
                         conn.commit()
@@ -179,39 +188,35 @@ async def upload_worker():
                         for f_path in files_to_delete:
                             if os.path.exists(f_path): os.remove(f_path)
                         
-                        print("✅ تم رفع الألبوم بنجاح!")
-                        await asyncio.sleep(15) # فاصل حماية بعد رفع الألبوم
+                        print("✅ تم رفع الألبوم ككتلة واحدة بنجاح!")
+                        await asyncio.sleep(12)
                         continue
 
-            # معالجة الرسائل المنفردة
+            # رفع الملفات الفردية
             print(f"⬆️ جاري رفع ملف منفرد (ID: {msg_id})...")
-            if m_type == "photo":
-                await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption)
-            elif m_type == "video":
-                await app.send_video(TARGET_CHAT, video=file_path, caption=caption, supports_streaming=True)
-            elif m_type == "doc":
-                await app.send_document(TARGET_CHAT, document=file_path, caption=caption)
-            elif m_type == "audio":
-                await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption)
+            if m_type == "photo": await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption)
+            elif m_type == "video": await app.send_video(TARGET_CHAT, video=file_path, caption=caption, supports_streaming=True)
+            elif m_type == "doc": await app.send_document(TARGET_CHAT, document=file_path, caption=caption)
+            elif m_type == "audio": await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption)
 
             cursor.execute("UPDATE queue SET status='UPLOADED' WHERE id=?", (q_id,))
             conn.commit()
 
             if os.path.exists(file_path): os.remove(file_path)
-            print(f"✅ تم رفع الملف بنجاح (ID: {msg_id})!")
-            await asyncio.sleep(12)
+            print(f"✅ تم الرفع بنجاح (ID: {msg_id})!")
+            await asyncio.sleep(10)
 
         except FloodWait as e:
-            print(f"🛑 [FloodWait] طلب تيليجرام التوقف لـ {e.value} ثانية. سيتم الانتظار دون فقدان البيانات.")
+            print(f"🛑 [FloodWait] انتظار إجباري لـ {e.value} ثانية...")
             await asyncio.sleep(e.value + 5)
         except Exception as e:
             print(f"❌ خطأ أثناء الرفع: {e}")
-            await asyncio.sleep(10)
+            await asyncio.sleep(8)
 
 # --- 8. نقطة الانطلاق ---
 async def main():
     await app.start()
-    print("🚀 تم تشغيل نظام الأرشيف المدرع (سحب فوري + رفع ألبومات)...")
+    print("🚀 تشغيل النسخة الفائقة والمعدلة كلياً...")
     
     asyncio.create_task(upload_worker())
     asyncio.create_task(scan_history_worker())