Update bot.py

bot.py

@@ -19,14 +19,12 @@ logger = logging.getLogger(__name__)
 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_CHAT_RAW = os.environ.get("TARGET_CHAT", "me")
 
 DATA_DIR = "./data"
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
-logger.info(f"🎯 القناة الهدف: {TARGET_CHAT}")
-
 # ============ العميل ============
 app = Client(
     "archiver",
@@ -40,6 +38,9 @@ app = Client(
     lang_code="ar"
 )
 
+# متغير عام للقناة الهدف الفعلية (قد تتغير بعد التحقق)
+TARGET_CHAT = None
+
 # ============ معرفة نوع الملف ============
 def get_media_info(message):
     if message.photo:
@@ -58,7 +59,7 @@ def get_media_info(message):
         return "animation", message.animation.file_size or 0
     return None, 0
 
-# ============ اسم المصدر (يعمل مع كل الأنواع) ============
+# ============ اسم المصدر ============
 def get_source_name(message):
     chat = message.chat
     if chat.type in ["channel", "supergroup", "group"]:
@@ -153,19 +154,17 @@ async def flush_media_group(chat_id, mg_id):
             except:
                 pass
 
-# ============ المعالج الرئيسي — يستقبل من كل شيء ============
+# ============ المعالج الرئيسي ============
 @app.on_message(filters.incoming & ~filters.me & filters.media)
 async def on_media(client, message):
     chat_id = message.chat.id
     
-    # لا تنقل من القناة الهدف نفسها
-    if chat_id == TARGET_CHAT:
+    if str(chat_id) == str(TARGET_CHAT):
         return
     
     src_name = get_source_name(message)
-    logger.info(f"📥 جديد من {chat_id} | {src_name} | نوع الدردشة: {message.chat.type} | رسالة {message.id} | نوع: {message.media}")
+    logger.info(f"📥 جديد من {chat_id} | {src_name} | نوع: {message.chat.type} | رسالة {message.id} | نوع: {message.media}")
     
-    # إذا ألبوم
     if message.media_group_id:
         key = (chat_id, message.media_group_id)
         async with mg_lock:
@@ -175,7 +174,6 @@ async def on_media(client, message):
             media_groups[key].append(message)
         return
     
-    # رسالة منفردة
     m_type, fsize = get_media_info(message)
     if not m_type:
         return
@@ -222,17 +220,36 @@ async def on_media(client, message):
             except:
                 pass
 
-# ============ مراقبة عامة — يستقبل من كل شيء ============
+# ============ مراقبة عامة ============
 @app.on_message(filters.incoming & ~filters.me)
 async def monitor_all(client, message):
-    if message.chat.id == TARGET_CHAT:
+    if str(message.chat.id) == str(TARGET_CHAT):
         return
     src_name = get_source_name(message)
-    logger.info(f"👁️ رسالة واردة من {message.chat.id} | نوع: {message.chat.type} | اسم: {src_name} | نص: {message.text or '[ملف]'}")
-    if message.media:
-        logger.info(f"   └─ تحتوي على media: {message.media}")
+    logger.info(f"👁️ رسالة من {message.chat.id} | نوع: {message.chat.type} | اسم: {src_name}")
+
+# ============ التحقق من القناة الهدف ============
+async def resolve_target_chat():
+    global TARGET_CHAT
+    raw = TARGET_CHAT_RAW
+    
+    if raw.lower() == "me":
+        TARGET_CHAT = "me"
+        logger.info("🎯 الهدف: Saved Messages (me)")
+        return
+    
+    try:
+        # جرب كـ integer أولاً
+        chat_id_int = int(raw)
+        chat = await app.get_chat(chat_id_int)
+        TARGET_CHAT = chat_id_int
+        logger.info(f"🎯 تم التحقق من القناة: {chat.title} | id: {chat_id_int}")
+    except Exception as e:
+        logger.error(f"❌ فشل التحقق من القناة {raw}: {e}")
+        logger.info("⚠️ سيتم استخدام Saved Messages (me) كاحتياطي")
+        TARGET_CHAT = "me"
 
-# ============ تشغيل Gradio في thread منفصلة ============
+# ============ تشغيل Gradio ============
 def run_gradio():
     with gr.Blocks(title="Telegram Archiver") as demo:
         gr.Markdown("# 📡 Telegram Archiver")
@@ -240,22 +257,22 @@ def run_gradio():
         
         gr.Textbox(
             label="الحالة",
-            value="✅ نشط - يستقبل من جميع القنوات والمجموعات والخاص",
+            value="✅ نشط - يستقبل من الكل",
             interactive=False
         )
         
         gr.Markdown("""
-        ### 🔧 المتغيرات المطلوبة:
+        ### 🔧 المتغيرات:
         
-        | المتغير | الوصف |
-        |---------|-------|
-        | `API_ID` | من my.telegram.org |
-        | `API_HASH` | من my.telegram.org |
-        | `SESSION_STRING` | سلسلة الجلسة |
-        | `TARGET_CHAT` | معر�� قناتك الخاصة |
+        | المتغير | القيمة | مثال |
+        |---------|--------|------|
+        | `API_ID` | رقم | `12345678` |
+        | `API_HASH` | نص | `abc123...` |
+        | `SESSION_STRING` | نص | `BQAD...` |
+        | `TARGET_CHAT` | معرف القناة الخاصة | `-1001234567890` |
         
-        ### 📝 ملاحظة:
-        البوت يستقبل تلقائياً من **الكل**: قنوات، مجموعات، ودردشات خاصة.
+        ### ⚠️ ملاحظة مهمة للقنوات الخاصة:
+        الحساب لازم يكون **عضو** في القناة الخاصة. إذا فشل التحقق، يرسل إلى Saved Messages.
         """)
     
     demo.launch(server_name="0.0.0.0", server_port=7860)
@@ -264,6 +281,10 @@ def run_gradio():
 async def main():
     await app.start()
     me = await app.get_me()
+    
+    # 🔑 التحقق من القناة الهدف
+    await resolve_target_chat()
+    
     logger.info("=" * 50)
     logger.info(f"🚀 البوت يعمل!")
     logger.info(f"👤 الحساب: {me.first_name} (@{me.username})")