Update bot.py

bot.py

@@ -4,14 +4,14 @@ import logging
 import time
 import threading
 import shutil
-from typing import Optional, List, Dict, Any, Set
+from typing import Optional, List, Dict, Any
 import aiosqlite
 import gradio as gr
-from hydrogram import Client, filters
+from hydrogram import Client, filters, enums
 from hydrogram.errors import FloodWait, RPCError
 from hydrogram.types import (
     InputMediaPhoto, InputMediaVideo, InputMediaDocument, 
-    InputMediaAudio, Message, Chat
+    InputMediaAudio, Message
 )
 from fastapi import FastAPI
 import uvicorn
@@ -44,7 +44,7 @@ logging.basicConfig(
     format='%(asctime)s | %(levelname)-7s | %(message)s',
     datefmt='%Y-%m-%d %H:%M:%S'
 )
-logger = logging.getLogger("AutoArchiverPro")
+logger = logging.getLogger("UniversalArchiver")
 
 # ============ الإعدادات والمتغيرات ============
 API_ID = int(os.environ.get("API_ID", 0))
@@ -55,12 +55,12 @@ AUTO_SCAN_ALL = os.environ.get("AUTO_SCAN_ALL", "1") == "1"
 
 DATA_DIR = "/data" if os.path.exists("/data") else "/tmp/data"
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
-DB_PATH = os.path.join(DATA_DIR, "auto_archiver_v4.db")
+DB_PATH = os.path.join(DATA_DIR, "ultra_archiver_v5.db")
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
 # ============ إنشاء العميل ============
 app = Client(
-    "auto_archiver",
+    "universal_archiver",
     api_id=API_ID,
     api_hash=API_HASH,
     session_string=SESSION_STRING,
@@ -69,7 +69,7 @@ app = Client(
     max_concurrent_transmissions=4,
     device_model="Universal Archiver Server",
     system_version="Linux 64bit",
-    app_version="4.0.0",
+    app_version="5.0.0",
     lang_code="ar"
 )
 
@@ -88,9 +88,9 @@ stats = {
     "status": "Starting..."
 }
 
-# ============ إدارة التحكم بالسرعة وتفادي الحظر ============
+# ============ إدارة التحكم بالسرعة ومكافحة الحظر ============
 class SmartFloodManager:
-    def __init__(self, normal_delay: float = 1.6):
+    def __init__(self, normal_delay: float = 1.5):
         self.normal_delay = normal_delay
         self.last_call = 0.0
         self.lock = asyncio.Lock()
@@ -113,9 +113,9 @@ class SmartFloodManager:
     def handle_flood(self, seconds: int):
         wait_seconds = max(seconds, 5) + 2
         self.flood_until = time.time() + wait_seconds
-        logger.warning(f"🚨 استلام FloodWait من تيليجرام: {seconds} ثانية. تهدئة حتى {wait_seconds} ثانية.")
+        logger.warning(f"🚨 استلام FloodWait: {seconds} ثانية. تهدئة العمليات حتى {wait_seconds} ثانية.")
 
-flood_manager = SmartFloodManager(normal_delay=1.8)
+flood_manager = SmartFloodManager(normal_delay=1.6)
 
 # ============ قاعدة البيانات ذاتية الإصلاح والتحديث ============
 def cleanup_corrupted_files():
@@ -175,7 +175,7 @@ async def init_db():
                 chat_type TEXT,
                 last_scraped_id INTEGER DEFAULT 0,
                 total_queued INTEGER DEFAULT 0,
-                status TEXT DEFAULT 'idle', -- idle, in_progress, completed
+                status TEXT DEFAULT 'idle',
                 updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
             )
         """)
@@ -381,27 +381,38 @@ async def queue_worker_loop():
             logger.error(f"خطأ في محرك الإرسال: {e}")
             await asyncio.sleep(4)
 
-# ============ تسخين ذاكرة المحادثات وحل خطأ PEER_ID_INVALID نهائياً ============
+# ============ فحص وتسخين الذاكرة لجميع المحادثات ============
 async def warm_up_dialogs_cache() -> List[Dict[str, Any]]:
     logger.info("🔄 جاري فحص وتسخين ذاكرة القنوات والمجموعات المنضم إليها الحساب...")
     discovered = []
     try:
         async for dialog in app.get_dialogs():
             chat = dialog.chat
+            if not chat:
+                continue
+
             if TARGET_CHAT_ID and chat.id == TARGET_CHAT_ID:
                 continue
-            if chat.type in ["channel", "supergroup", "group"]:
+            if str(chat.id) == str(TARGET_CHAT):
+                continue
+
+            type_str = str(chat.type).lower()
+            is_group_or_channel = any(k in type_str for k in ["channel", "supergroup", "group"])
+
+            if is_group_or_channel:
+                title = chat.title or chat.first_name or f"Chat_{chat.id}"
                 chat_info = {
                     "id": chat.id,
-                    "title": chat.title or f"Chat_{chat.id}",
+                    "title": title,
                     "username": chat.username or "",
-                    "type": chat.type
+                    "type": type_str
                 }
                 KNOWN_CHATS[chat.id] = chat_info
                 discovered.append(chat_info)
+                logger.info(f"📌 تم اكتشاف: [{title}] | الآيدي: ({chat.id}) | النوع: ({type_str})")
 
         stats["discovered_chats"] = len(discovered)
-        logger.info(f"✅ تم اكتشاف وتأمين {len(discovered)} قناة ومجموعة بنجاح!")
+        logger.info(f"✅ اكتمل الفحص: تم تأمين وتسجيل {len(discovered)} قناة ومجموعة بنجاح!")
         return discovered
     except Exception as e:
         logger.error(f"خطأ أثناء جلب المحادثات: {e}")
@@ -483,7 +494,7 @@ async def scrape_all_joined_chats(limit_per_chat: int = 0) -> str:
     try:
         chats = await warm_up_dialogs_cache()
         if not chats:
-            return "لم يتم العثور على قنوات أو مجموعات في الحساب."
+            return "لم يتم العثور على قنوات أو مجموعات في الحساب. تأكد من أن الحساب منضم إلى قنوات."
 
         logger.info(f"🚀 بدء عملية الفحص الشامل لـ {len(chats)} قناة ومجموعة...")
         total_all = 0
@@ -514,12 +525,13 @@ async def realtime_all_chats_handler(client, message: Message):
         if str(message.chat.id) == str(TARGET_CHAT):
             return
 
-        if message.chat.id not in KNOWN_CHATS and message.chat.type in ["channel", "supergroup", "group"]:
+        type_str = str(message.chat.type).lower()
+        if message.chat.id not in KNOWN_CHATS and any(k in type_str for k in ["channel", "supergroup", "group"]):
             KNOWN_CHATS[message.chat.id] = {
                 "id": message.chat.id,
                 "title": message.chat.title or f"Chat_{message.chat.id}",
                 "username": message.chat.username or "",
-                "type": message.chat.type
+                "type": type_str
             }
 
         if message.media_group_id:
@@ -570,7 +582,7 @@ def create_dashboard():
         gr.Markdown("""
         # 🌐 Universal Telegram Archiver Pro (النسخة الشاملة التلقائية)
         **يسحب من جميع القنوات والكروبات المنضم إليها تلقائياً بدون استثناء وبدون ضياع ملف واحد:**
-        * ✅ حل مشكلة `PEER_ID_INVALID` بشكل جذري بتسخين الذاكرة التلقائي.
+        * ✅ حل مشكلة `PEER_ID_INVALID` بشكل جذري.
         * ✅ فحص أرشيف كل القنوات والمجموعات المنضم إليها دفعة واحدة تلقائياً.
         * ✅ رصد فوري لحظي لأي ملف أو ألبوم ينزل في أي قناة أو كروب.
         * ✅ طابور ذكي دائم مع كتابة المصدر ورقم المقطع ونظام ضد الحظر.
@@ -591,6 +603,13 @@ def create_dashboard():
 
         scrape_status = gr.Textbox(label="حالة الفحص والعمليات", interactive=False)
 
+        gr.Markdown("---")
+        gr.Markdown("### 🎯 فحص قناة أو كروب محدد (اختياري):")
+        with gr.Row():
+            single_ch_input = gr.Textbox(label="معرف أو رابط القناة", placeholder="@my_channel أو https://t.me/...")
+            single_lim_input = gr.Number(label="الحد الأقصى (0 = الكل)", value=0)
+            single_scan_btn = gr.Button("فحص هذه القناة فقط 📥")
+            
         def trigger_scan_all(lim):
             if MAIN_LOOP and MAIN_LOOP.is_running():
                 asyncio.run_coroutine_threadsafe(
@@ -600,20 +619,40 @@ def create_dashboard():
                 return "🚀 تم بدء عملية الفحص الشامل لجميع القنوات والكروبات في الخلفية بنجاح! راقب السجلات والطابور."
             return "❌ تعذر إطلاق المهمة، تأكد من اتصال البوت."
 
+        def trigger_single_scan(ch, lim):
+            if not ch or not ch.strip():
+                return "⚠️ يرجى إدخال معرف القناة أولاً."
+            if MAIN_LOOP and MAIN_LOOP.is_running():
+                async def do_single():
+                    try:
+                        c = await app.get_chat(ch.strip())
+                        await scrape_single_chat(c.id, c.title or ch, int(lim))
+                    except Exception as e:
+                        logger.error(f"Single scan error: {e}")
+                asyncio.run_coroutine_threadsafe(do_single(), MAIN_LOOP)
+                return f"🚀 تم بدء فحص القناة ({ch}) بنجاح!"
+            return "❌ تعذر إطلاق المهمة."
+
         scan_all_btn.click(fn=trigger_scan_all, inputs=[limit_input], outputs=[scrape_status])
+        single_scan_btn.click(fn=trigger_single_scan, inputs=[single_ch_input, single_lim_input], outputs=[scrape_status])
 
     return demo
 
 # ============ تشغيل الويب ============
 def run_web_server():
     fastapi_app = FastAPI()
-    demo = create_dashboard()
-    gr.mount_gradio_app(fastapi_app, demo, path="/")
     
     @fastapi_app.get("/health")
     async def health():
         return {"status": "ok", "stats": stats}
 
+    @fastapi_app.get("/ping")
+    async def ping():
+        return {"status": "alive", "stats": stats}
+
+    demo = create_dashboard()
+    gr.mount_gradio_app(fastapi_app, demo, path="/")
+    
     uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)
 
 # ============ نقطة الدخول ============