Update main.py

main.py

@@ -698,204 +698,112 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 # ==============================================================================
-# 🟢 پارت 11: آپلودر صبور، هوشمند و صنعتی (با ظرفیت 50 کارگر همزمان طبق درخواست)
+# 🟢 پارت 11: آپلودر با استفاده انحصاری از کتابخانه رسمی روب‌پای (Rubpy)
 # ==============================================================================
 import asyncio
 import os
-import aiohttp
 
-# 🟢 ۱. مدیریت صف آپلود اصلی (ارتقا به ۵۰ کارگر)
+# 🟢 ۱. مدیریت صف آپلود
 _upload_lock = None
-_shared_session = None
 
 def get_upload_lock():
     global _upload_lock
     if _upload_lock is None:
-        # ظرفیت روی 50 تنظیم شد.
-        _upload_lock = asyncio.Semaphore(2) 
+        # برای اینکه کتابخانه کرش نکند، فایل‌ها را 2 تا 2 تا به کتابخانه می‌سپاریم
+        _upload_lock = asyncio.Semaphore(10) 
     return _upload_lock
 
-async def get_shared_session():
-    global _shared_session
-    if _shared_session is None or _shared_session.closed:
-        connector = aiohttp.TCPConnector(limit=100, keepalive_timeout=60)
-        _shared_session = aiohttp.ClientSession(connector=connector)
-    return _shared_session
-
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
     if not os.path.exists(abs_path):
         return "File Not Found"
 
-    try:
-        file_bytes = await asyncio.to_thread(sync_read_file, abs_path)
-    except Exception as e:
-        return f"File Read Error: {e}"
-
     error_logs = []
-    bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
     
+    # تشخیص نوع فایل
     api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else file_type
     if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'): 
         api_file_type = "Music"
     if api_file_type == "File" and abs_path.lower().endswith('.mp3'): 
         api_file_type = "Music"
-        
+
     async with get_upload_lock():
-        session = await get_shared_session()
+        backoff_time = 10 # زمان صبر اولیه در صورت برخورد به محدودیت
         
-        # سیستم خواب تصاعدی: اگر روبیکا بلاک کرد، چقدر صبر کنیم؟
-        backoff_time = 10 
-        
-        # 🚀 فاز اصلی: استفاده از API پرسرعت روبیکا
-        for attempt in range(12): 
+        # ربات ۵ بار تلاش می‌کند تا کتابخانه کار را انجام دهد
+        for attempt in range(5): 
             try:
-                # --- مرحله ۱: گرفتن لینک آپلود ---
-                req_resp = await session.post(
-                    f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile", 
-                    json={"type": api_file_type},
-                    timeout=aiohttp.ClientTimeout(total=20)
-                )
-                if req_resp.status != 200:
-                    error_logs.append(f"reqSend HTTP {req_resp.status}")
-                    await asyncio.sleep(3)
-                    continue
-                    
-                req_data = await req_resp.json()
-                if req_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                    error_logs.append(f"TOO_REQ (گرفتن لینک - صبر {backoff_time}s)")
-                    await asyncio.sleep(backoff_time)
-                    backoff_time += 5 
-                    continue
-
-                upload_url = req_data.get("data", {}).get("upload_url")
-                if not upload_url: 
-                    error_logs.append("لینک آپلود دریافت نشد")
-                    await asyncio.sleep(3)
-                    continue
-
-                # --- مرحله ۲: آپلود فیزیکی فایل ---
-                form = aiohttp.FormData()
-                form.add_field('file', file_bytes, filename=os.path.basename(abs_path))
-                
-                up_resp = await session.post(
-                    upload_url, 
-                    data=form, 
-                    timeout=aiohttp.ClientTimeout(total=120)
-                )
-                if up_resp.status != 200:
-                    error_logs.append(f"Upload HTTP {up_resp.status}")
-                    await asyncio.sleep(3)
-                    continue
-
-                up_data = await up_resp.json()
-                if up_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                    error_logs.append(f"TOO_REQ (آپلود - صبر {backoff_time}s)")
-                    await asyncio.sleep(backoff_time)
-                    backoff_time += 5
-                    continue
-
-                final_file_id = None
-                if up_data.get("status") in ["OK", "OK_DET"]:
-                    if "data" in up_data and isinstance(up_data["data"], dict):
-                        final_file_id = up_data["data"].get("file_id") or up_data["data"].get("id")
-                    else:
-                        final_file_id = up_data.get("file_id")
-                else:
-                    error_logs.append(f"Up Error: {up_data.get('status')}")
-                            
-                if final_file_id:
-                    # --- مرحله ۳: ارسال فایل به کاربر ---
-                    send_payload = {
-                        "chat_id": str(chat_id),
-                        "file_id": str(final_file_id),
-                        "text": caption,
-                        "chat_keypad_type": "New",
-                        "chat_keypad": MAIN_KEYPAD_DICT
-                    }
-                    send_resp = await session.post(
-                        f"https://botapi.rubika.ir/v3/{bot_token}/sendFile", 
-                        json=send_payload,
-                        timeout=aiohttp.ClientTimeout(total=20)
-                    )
-                    s_data = await send_resp.json()
-                    
-                    if s_data.get("status") == "OK": 
-                        await asyncio.sleep(1)
+                if api_file_type == "Image":
+                    if hasattr(client, "send_photo"):
+                        try:
+                            await client.send_photo(chat_id, photo=abs_path, caption=caption)
+                        except TypeError:
+                            await client.send_photo(chat_id, abs_path, caption=caption)
                         return True
-                    elif s_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                        error_logs.append(f"TOO_REQ (ارسال نهایی - صبر {backoff_time}s)")
-                        await asyncio.sleep(backoff_time)
-                        backoff_time += 5
-                        continue
-                    else:
-                        error_logs.append(f"Send Error: {s_data.get('status')}")
+                        
+                elif api_file_type in ["Music", "Voice"]:
+                    if hasattr(client, "send_music"):
+                        try:
+                            await client.send_music(chat_id, abs_path, caption=caption)
+                        except TypeError:
+                            await client.send_music(chat_id, music=abs_path, caption=caption)
+                        return True
+                        
                 else:
-                    if not final_file_id and "INVALID_INPUT" in str(up_data):
-                        api_file_type = "File"
-                        error_logs.append("تغییر اجباری نوع به File")
-
+                    if hasattr(client, "send_document"):
+                        try:
+                            await client.send_document(chat_id, document=abs_path, caption=caption)
+                        except TypeError:
+                            await client.send_document(chat_id, abs_path, caption=caption)
+                        return True
+                    elif hasattr(client, "send_file"):
+                        try:
+                            await client.send_file(chat_id, file=abs_path)
+                        except TypeError:
+                            await client.send_file(chat_id, abs_path)
+                        if caption:
+                            await send_with_keyboard(client, chat_id, caption, True)
+                        return True
+                        
+                error_logs.append("متد مناسب در کتابخانه یافت نشد.")
+                break # اگر متد کلاً وجود نداشت، از حلقه خارج می‌شویم
+                
             except Exception as e:
-                error_logs.append(f"تلاش {attempt+1} خطا: {str(e)[:25]}")
-                await asyncio.sleep(3)
-
-        # 🛡️ فاز دوم (Fallback امن)
-        try:
-            sent_success = False
-            if api_file_type == "Image" and hasattr(client, "send_photo"):
-                try:
-                    await client.send_photo(chat_id, photo=abs_path, caption=caption)
-                    sent_success = True
-                except Exception as e:
-                    if "TOO_REQ" in str(e).upper():
-                        await asyncio.sleep(15)
-                        await client.send_photo(chat_id, photo=abs_path, caption=caption)
-                        sent_success = True
-            
-            elif api_file_type in ["Music", "Voice"] and hasattr(client, "send_music"):
-                try:
-                    await client.send_music(chat_id, abs_path, caption=caption)
-                    sent_success = True
-                except Exception as e:
-                    if "TOO_REQ" in str(e).upper():
-                        await asyncio.sleep(15)
-                        await client.send_music(chat_id, abs_path, caption=caption)
-                        sent_success = True
-            
-            if not sent_success:
-                if hasattr(client, "send_document"):
-                    await client.send_document(chat_id, document=abs_path, caption=caption)
-                    sent_success = True
-                elif hasattr(client, "send_file"):
-                    await client.send_file(chat_id, abs_path)
-                    sent_success = True
-            
-            if sent_success:
-                return True
+                err_str = str(e)
+                error_logs.append(f"تلاش {attempt+1} با کتابخانه: {err_str[:40]}")
                 
-        except Exception as e:
-            error_logs.append(f"Fallback Fail: {str(e)[:40]}")
+                # اگر کتابخانه ارور TOO_REQUESTS داد، صبر می‌کنیم
+                if "TOO_REQ" in err_str.upper() or "429" in err_str:
+                    await asyncio.sleep(backoff_time)
+                    backoff_time += 15 # دفعه بعد بیشتر صبر می‌کند
+                else:
+                    await asyncio.sleep(4)
+
+    return "❌ خطا در آپلود با کتابخانه روب‌پای:\n" + "\n".join(error_logs[-3:])
 
-    return "❌ خطا در آپلود:\n" + "\n".join(error_logs[-5:])
 
 # ==============================================================================
-# 🟢 ۲. تعریف کارگرهای پس‌زمینه عمومی (ارتقا به 50 کارگر اختصاصی)
+# 🟢 ۲. تعریف کارگرهای پس‌زمینه عمومی (مشترک برای همه بخش‌ها)
 # ==============================================================================
-_general_upload_workers = asyncio.Semaphore(50) # ظرفیت 50 فایل همزمان در پس‌زمینه
+_general_upload_workers = asyncio.Semaphore(10) # ظرفیت پردازش پس‌زمینه
 
 async def background_general_uploader(client, chat_id, file_name, file_type, caption_text, str_chat_id, credit_key=None):
     async with _general_upload_workers:
         try:
             res = await helper_upload_file(client, chat_id, file_name, file_type, caption_text)
             if res is not True:
+                # عودت اعتبار در صورت شکست قطعی آپلود
                 creds = get_user_credits(str_chat_id)
                 if not creds.get("is_premium") and credit_key:
                     user_credits_db[str_chat_id][credit_key] += 1
                     save_db(user_credits_db)
                 await send_with_keyboard(client, chat_id, f"❌ سرور ارسال گزارش داد:\nآپلود {file_type} انجام نشد و اعتبار شما برگشت داده شد.\n`{str(res)[:500]}`", True)
-        except: pass
+        except Exception as e: 
+            pass
         finally:
-            if os.path.exists(file_name): os.remove(file_name)
+            if os.path.exists(file_name): 
+                os.remove(file_name)
+