Update bot.py

bot.py

@@ -1,4 +1,5 @@
 import os
+import gc
 import asyncio
 import sqlite3
 import time
@@ -10,12 +11,13 @@ from pyrogram.errors import FloodWait
 from pyrogram.enums import ChatType
 from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaDocument, InputMediaAudio
 
-# --- 1. إعداد مساحات التخزين والتنظيف الأولية ---
+# --- 1. إعداد مساحات التخزين وتحسين SQLite للمزامنة السريعة ---
 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
@@ -23,6 +25,8 @@ for f in os.listdir(DOWNLOAD_DIR):
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 
+# تفعيل وضع WAL لمنع تضارب الكتابة والقراءة تكسيراً للأخطاء
+cursor.execute("PRAGMA journal_mode=WAL;")
 cursor.execute('''
 CREATE TABLE IF NOT EXISTS queue (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -40,12 +44,15 @@ CREATE TABLE IF NOT EXISTS queue (
 cursor.execute('''CREATE TABLE IF NOT EXISTS scan_progress (chat_id INTEGER PRIMARY KEY, last_msg_id INTEGER)''')
 conn.commit()
 
-# --- 2. سيرفر الويب الوهمي ---
+# متغيرة ديناميكية للتحكم بالفواصل الزمنية للحماية من الحظر
+DYNAMIC_UPLOAD_DELAY = 10
+
+# --- 2. سيرفر الويب الوهمي للحفاظ على التشغيل ---
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         self.send_response(200)
         self.end_headers()
-        self.wfile.write(b"Industrial Grade Archiver Active!")
+        self.wfile.write(b"Nuclear Grade Telegram Archiver Active!")
     def log_message(self, format, *args): pass
 
 Thread(target=lambda: HTTPServer(('0.0.0.0', 7860), RequestHandler).serve_forever(), daemon=True).start()
@@ -65,15 +72,14 @@ app = Client(
     max_concurrent_transmissions=2
 )
 
-# --- 4. فحص مساحة القرص ---
+# --- 4. فحص مساحة القرص والذاكرة ---
 def check_disk_space():
     total, used, free = shutil.disk_usage(DATA_DIR)
-    return (free / (1024 * 1024)) > 500  # إرجاع True إذا كان المتبقي أكبر من 500 ميجابايت
+    return (free / (1024 * 1024)) > 500 # 500MB حد أدنى للأمان
 
-# --- 5. دالة التنزيل الآمن المتقدمة ---
+# --- 5. دالة التنزيل الفولاذية ---
 async def download_media_safe(message):
     while not check_disk_space():
-        print("⚠️ المساحة المتبقية منخفضة! انتظار محرك الرفع لتنظيف القرص...")
         await asyncio.sleep(15)
 
     cursor.execute("SELECT status FROM queue WHERE chat_id=? AND msg_id=?", (message.chat.id, message.id))
@@ -84,7 +90,6 @@ async def download_media_safe(message):
     
     for attempt in range(3):
         try:
-            # تنزيل مع مهلة 5 دقائق للملفات الكبيرة
             file_path = await asyncio.wait_for(message.download(file_name=unique_name), timeout=300)
             
             if file_path and os.path.exists(file_path) and os.path.getsize(file_path) > 0:
@@ -104,24 +109,44 @@ async def download_media_safe(message):
                     (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)
         except Exception:
-            await asyncio.sleep(3)
+            await asyncio.sleep(2)
             
     return None
 
-# --- 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:
         await download_media_safe(message)
 
-# --- 7. حاصد السجل التراكمي ---
+# --- 7. أوامر الأدمن والمراقبة الحية ---
+@app.on_message(filters.me & filters.command("status"))
+async def status_command(client, message):
+    cursor.execute("SELECT COUNT(*) FROM queue WHERE status='DOWNLOADED'")
+    pending = cursor.fetchone()[0]
+    cursor.execute("SELECT COUNT(*) FROM queue WHERE status='UPLOADED'")
+    uploaded = cursor.fetchone()[0]
+    
+    total, used, free = shutil.disk_usage(DATA_DIR)
+    free_mb = round(free / (1024 * 1024), 2)
+    
+    report = (
+        f"📊 **حالة البوت الحالية:**\n\n"
+        f"⏳ مقاطع في طابور الرفع: `{pending}`\n"
+        f"✅ مقاطع تم رفعها بنجاح: `{uploaded}`\n"
+        f"💾 المساحة المتبقية: `{free_mb} MB`\n"
+        f"⏱️ التأخير الديناميكي الحالي: `{DYNAMIC_UPLOAD_DELAY} ثانية`"
+    )
+    await message.reply_text(report)
+
+# --- 8. حاصد السجل التراكمي ---
 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:
@@ -140,22 +165,22 @@ async def scan_history_worker():
 
             except Exception:
                 pass
-            await asyncio.sleep(4)
-    print("✅ اكتمل مسح جميع السجلات المتاحة.")
+            await asyncio.sleep(3)
+    print("✅ اكتمل سحب كافة الأرشيفات القديمة.")
 
-# --- 8. النسخ الاحتياطي السحابي لقاعدة البيانات ---
+# --- 9. النسخ الاحتياطي التلقائي ---
 async def db_backup_worker():
     while True:
         await asyncio.sleep(43200) # كل 12 ساعة
         try:
             if os.path.exists(DB_PATH):
                 await app.send_document(TARGET_CHAT, document=DB_PATH, caption="📦 نسخة احتياطية لسجل الأرشيف")
-                print("💾 تم إرسال نسخة احتياطية من قاعدة البيانات إلى القناة.")
-        except Exception as e:
-            print(f"❌ خطأ بالنسخ الاحتياطي: {e}")
+        except Exception:
+            pass
 
-# --- 9. محرك الرفع الشامل والألبومات ---
+# --- 10. محرك الرفع الذكي المقاوم للحظر وقاطع التسريبات ---
 async def upload_worker():
+    global DYNAMIC_UPLOAD_DELAY
     while True:
         try:
             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")
@@ -172,7 +197,7 @@ async def upload_worker():
                 conn.commit()
                 continue
 
-            # تجميع الألبومات وتقسيمها إذا زادت عن 10
+            # معالجة وتقسيم الألبومات
             if group_id:
                 time_passed = time.time() - ts
                 if time_passed < 15:
@@ -185,7 +210,6 @@ async def upload_worker():
                 group_rows = cursor.fetchall()
 
                 if group_rows:
-                    # تقسيم الألبوم كل 10 عناصر
                     for i in range(0, len(group_rows), 10):
                         chunk = group_rows[i:i+10]
                         media_list = []
@@ -205,7 +229,7 @@ async def upload_worker():
                             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:
@@ -215,11 +239,12 @@ async def upload_worker():
                             for f_path in files_to_delete:
                                 if os.path.exists(f_path): os.remove(f_path)
                             
-                            await asyncio.sleep(10)
+                            gc.collect() # تفريغ الذاكرة المؤقتة فوراً
+                            await asyncio.sleep(DYNAMIC_UPLOAD_DELAY)
                     continue
 
-            # رفع الوسائط المنفردة حسب نوعها
-            print(f"⬆️ رفع ملف منفرد (ID: {msg_id}) نوع: {m_type}...")
+            # رفع الوسائط المنفردة
+            print(f"⬆️ رفع (ID: {msg_id}) | نوع: {m_type}...")
             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)
@@ -232,20 +257,28 @@ async def upload_worker():
             conn.commit()
 
             if os.path.exists(file_path): os.remove(file_path)
+            
+            # تقليل التأخير تدريجياً إذا كان النظام مستقراً
+            if DYNAMIC_UPLOAD_DELAY > 8:
+                DYNAMIC_UPLOAD_DELAY -= 0.5
+
+            gc.collect() # تفريغ الذاكرة الرام
             print(f"✅ تم الرفع والتنظيف (ID: {msg_id})!")
-            await asyncio.sleep(10)
+            await asyncio.sleep(DYNAMIC_UPLOAD_DELAY)
 
         except FloodWait as e:
-            print(f"🛑 [FloodWait] انتظار إجباري {e.value} ثانية...")
+            # زيادة التأخير التلقائي عند مواجهة أي حظر زمني
+            DYNAMIC_UPLOAD_DELAY += 5
+            print(f"🛑 [FloodWait] انتظار {e.value} ثانية. رفع التأخير الديناميكي إلى {DYNAMIC_UPLOAD_DELAY} ثانية.")
             await asyncio.sleep(e.value + 5)
         except Exception as e:
-            print(f"❌ خطأ الرفع: {e}")
+            print(f"❌ خطأ بالرفع: {e}")
             await asyncio.sleep(8)
 
-# --- 10. نقطة التشغيل ---
+# --- 11. نقطة التشغيل الرئيسية ---
 async def main():
     await app.start()
-    print("🚀 تشغيل نظام الأرشيف الصناعي المتكامل (النسخة الفائقة)...")
+    print("🚀 تم إطلاق النسخة النووية الكاملة بنجاح!")
     
     asyncio.create_task(upload_worker())
     asyncio.create_task(scan_history_worker())