Update bot.py

bot.py

@@ -1,5 +1,6 @@
 import os
 import gc
+import sys
 import asyncio
 import sqlite3
 import time
@@ -11,7 +12,7 @@ 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')
@@ -24,6 +25,7 @@ for f in os.listdir(DOWNLOAD_DIR):
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 cursor.execute("PRAGMA journal_mode=WAL;")
+cursor.execute("PRAGMA auto_vacuum = FULL;")
 
 cursor.execute('''
 CREATE TABLE IF NOT EXISTS queue (
@@ -44,49 +46,46 @@ 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()
 
-DYNAMIC_UPLOAD_DELAY = 8
+DYNAMIC_UPLOAD_DELAY = 6
 
-# --- 2. سيرفر الحفاظ على التشغيل ---
+# --- 2. سيرفر الويب والبقاء حياً ---
 class RequestHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         self.send_response(200)
         self.end_headers()
-        self.wfile.write(b"Ultimate Enterprise Archiver Running!")
+        self.wfile.write(b"God-Mode Telegram Archiver Engine Active!")
     def log_message(self, format, *args): pass
 
 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", "")
 TARGET_CHAT = int(os.environ.get("TARGET_CHAT", 0))
 
+# يمكنك إضافة بروكسي هنا إذا كنت تستخدم بيئة محظورة IP
 app = Client(
     "memory_session",
     api_id=API_ID,
     api_hash=API_HASH,
     session_string=SESSION_STRING,
     in_memory=True,
-    max_concurrent_transmissions=2
+    max_concurrent_transmissions=3
 )
 
-# --- 4. أدوات التنبيه والحماية ---
+# --- 4. الأدوات المساعدة وحماية القرص ---
 async def send_admin_alert(text):
-    try:
-        await app.send_message("me", f"⚠️ **تنبيه النظام:**\n{text}")
-    except Exception:
-        pass
+    try: await app.send_message("me", f"🚨 **تنبيه النظام:**\n{text}")
+    except: pass
 
 def check_disk_space():
     total, used, free = shutil.disk_usage(DATA_DIR)
     return (free / (1024 * 1024)) > 500
 
-# استخراج البصمة الفريدة ونوع الوسيط
 def get_media_details(message):
     media = message.photo or message.video or message.document or message.audio or message.voice or message.video_note or message.animation
-    if not media:
-        return None, None, 0
+    if not media: return None, None, 0
     
     m_type = "doc"
     if message.photo: m_type = "photo"
@@ -100,20 +99,16 @@ def get_media_details(message):
     file_size = getattr(media, "file_size", 0)
     return unique_id, m_type, file_size
 
-# --- 5. محرك التنزيل المحصن من التكرار ---
+# --- 5. محرك التنزيل المحصن ---
 async def download_media_safe(message):
     while not check_disk_space():
-        await send_admin_alert("المساحة المتبقية أقل من 500MB! إيقاف التنزيل مؤقتاً.")
-        await asyncio.sleep(30)
+        await asyncio.sleep(20)
 
     unique_id, m_type, file_size = get_media_details(message)
-    if not unique_id:
-        return None
+    if not unique_id: return None
 
-    # فحص منع التكرار بناءً على البصمة الفريدة
     cursor.execute("SELECT id FROM queue WHERE file_unique_id=?", (unique_id,))
-    if cursor.fetchone():
-        return None
+    if cursor.fetchone(): return None
 
     unique_name = f"{DOWNLOAD_DIR}/{message.chat.id}_{message.id}"
     
@@ -123,8 +118,6 @@ async def download_media_safe(message):
             
             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 "المصدر"
                 raw_cap = message.caption or ""
                 final_caption = f"{raw_cap}\n\n📌 *المصدر:* {src_title}".strip()
@@ -134,22 +127,21 @@ async def download_media_safe(message):
                     (message.chat.id, message.id, unique_id, group_id, m_type, final_caption, file_path, file_size, 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(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'")
@@ -161,17 +153,16 @@ async def status_command(client, message):
     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}s`"
+        f"👑 **حالة المحرك النووي المطلق:**\n\n"
+        f"📥 بالطابور: `{pending}` ملف\n"
+        f"📤 مكتمل الرفع: `{uploaded}` ملف\n"
+        f"💾 المساحة الحرة: `{free_mb} MB`\n"
+        f"⏱️ التأخير التكيفي: `{DYNAMIC_UPLOAD_DELAY}s`"
     )
     await message.reply_text(report)
 
-# --- 8. مسح السجل القديم ---
+# --- 8. مسح الأرشيف القديم ---
 async def scan_history_worker():
-    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:
@@ -180,29 +171,24 @@ async def scan_history_worker():
                 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)
+                    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: pass
+            await asyncio.sleep(2)
 
-            except Exception:
-                pass
-            await asyncio.sleep(3)
-
-# --- 9. محرك الرفع الذكي بركوب الأولوية للأحجام الصغيرة ---
+# --- 9. محرك الرفع والتنظيف التلقائي ---
 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 file_size ASC, id ASC LIMIT 1")
             row = cursor.fetchone()
 
             if not row:
-                await asyncio.sleep(5)
+                await asyncio.sleep(4)
                 continue
 
             q_id, chat_id, msg_id, group_id, m_type, caption, file_path, ts = row
@@ -215,8 +201,7 @@ async def upload_worker():
             # تجميع الألبومات
             if group_id:
                 time_passed = time.time() - ts
-                if time_passed < 15:
-                    await asyncio.sleep(15 - time_passed)
+                if time_passed < 12: await asyncio.sleep(12 - 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",
@@ -227,9 +212,7 @@ async def upload_worker():
                 if group_rows:
                     for i in range(0, len(group_rows), 10):
                         chunk = group_rows[i:i+10]
-                        media_list = []
-                        ids_to_update = []
-                        files_to_delete = []
+                        media_list, ids_to_update, files_to_delete = [], [], []
 
                         for idx, g_item in enumerate(chunk):
                             g_id, g_type, g_cap, g_path = g_item
@@ -245,8 +228,7 @@ async def upload_worker():
 
                         if 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,))
+                            for item_id in ids_to_update: cursor.execute("UPDATE queue SET status='UPLOADED' WHERE id=?", (item_id,))
                             conn.commit()
 
                             for f_path in files_to_delete:
@@ -256,7 +238,7 @@ async def upload_worker():
                             await asyncio.sleep(DYNAMIC_UPLOAD_DELAY)
                     continue
 
-            # رفع الفردي
+            # رفع منفرد
             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)
@@ -269,37 +251,42 @@ async def upload_worker():
             conn.commit()
 
             if os.path.exists(file_path): os.remove(file_path)
-            
-            if DYNAMIC_UPLOAD_DELAY > 6:
-                DYNAMIC_UPLOAD_DELAY -= 0.2
-
+            if DYNAMIC_UPLOAD_DELAY > 5: DYNAMIC_UPLOAD_DELAY -= 0.1
             gc.collect()
             await asyncio.sleep(DYNAMIC_UPLOAD_DELAY)
 
         except FloodWait as e:
-            DYNAMIC_UPLOAD_DELAY += 5
+            DYNAMIC_UPLOAD_DELAY += 4
             await asyncio.sleep(e.value + 5)
-        except Exception as e:
-            await asyncio.sleep(6)
+        except Exception:
+            await asyncio.sleep(5)
 
-# --- 10. حارس المهمات الخلفية (Task Supervisor) ---
-async def supervisor():
+# --- 10. صيانة قاعدة البيانات والتنظيف الدوري ---
+async def db_maintenance():
     while True:
-        await asyncio.sleep(60)
-        # التأكد من استمرار عمل المهام وإعادة تشغيل أي جزء ينقطع
-        gc.collect()
+        await asyncio.sleep(86400) # كل 24 ساعة
+        try:
+            cursor.execute("VACUUM;")
+            conn.commit()
+            print("🧹 تم تنظيف وضغط قاعدة البيانات بنجاح.")
+        except: pass
 
-# --- 11. التشغيل الرئيسي ---
+# --- 11. التشغيل الذاتي الحامي (Self-Healing Launcher) ---
 async def main():
     await app.start()
-    print("🚀 تم تشغيل أقصى نظام أرشفة ممكن برمجياً!")
+    print("🚀 إطلاق الهيكلية المطلقة المحصنة بالكامل!")
     
     asyncio.create_task(upload_worker())
     asyncio.create_task(scan_history_worker())
-    asyncio.create_task(supervisor())
+    asyncio.create_task(db_maintenance())
     
     await idle()
     await app.stop()
 
 if __name__ == "__main__":
-    app.run(main())
+    while True:
+        try:
+            app.run(main())
+        except Exception as crash_error:
+            print(f"⚠️ انهيار غير متوقع: {crash_error}. إعادة التشغيل الذاتي خلال 10 ثوانٍ...")
+            time.sleep(10)