CliRemote 1.6.0__py3-none-any.whl → 1.6.2__py3-none-any.whl

{cliremote-1.6.0.dist-info → cliremote-1.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CliRemote
-Version: 1.6.0
+Version: 1.6.2
 Summary: Remote client framework for Telegram automation using Pyrogram
 Home-page: https://github.com/MohammadAhmadi-R/CliRemote
 Author: MrAhmadiRad

{cliremote-1.6.0.dist-info → cliremote-1.6.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cliremote-1.6.0.dist-info/licenses/LICENSE,sha256=O-0zMbcEi6wXz1DiSdVgzMlQjJcNqNe5KDv08uYzqR0,1055
+cliremote-1.6.2.dist-info/licenses/LICENSE,sha256=O-0zMbcEi6wXz1DiSdVgzMlQjJcNqNe5KDv08uYzqR0,1055
 remote/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 remote/account_manager.py,sha256=u8yqcyq7Vl8LZ1rNlrEfwlJEbJiBzoSAlL3z2hhTzyc,9829
 remote/account_viewer.py,sha256=MQoH5lOz24651EjhlzVwi6O5hVUAT7Q5fFU6ZHjBlsM,4243
@@ -8,8 +8,8 @@ remote/batch_manager.py,sha256=jVGhYVwHMKJd7f7JxcWjKlwr03dq0RaGD1KdkyYdb00,1051
 remote/block_manager.py,sha256=R7UaQigr-hTRtjxjG3OvJdKhvp0mDpLaESp3Of1AYhs,5692
 remote/caption_manager.py,sha256=ekgcZ_D1q8C24WP18TXxlM5eWTknJmw-KNXDfqlsnEw,966
 remote/cleaner.py,sha256=gvPtkosGsf7Eb3zmYyeC44xB4HtCxlzKH3e3qLuVos4,5742
-remote/client_manager.py,sha256=EP2pVasUOUwjr4h-vO-LWWf9_rMCGAsFrzVvnGXJcNI,8250
-remote/client_picker.py,sha256=3QYi1UQ8pZjqR5YU2oKgQlCYV7-1pSPWknCyRFMcFak,4428
+remote/client_manager.py,sha256=dI9ZKOVPDXsq6_9Hi5Y_E_zKWyhSaqOxjiyRlr0hG5Y,10158
+remote/client_picker.py,sha256=6WDIabYhcXSwcCAuRUfJfh6V1ZXP2EiFdAnutK2qLTk,139
 remote/config.py,sha256=VK0e96gEINRViKIq99CYYuYyaVZTLtlWlPKKkBd41Cg,2377
 remote/device_manager.py,sha256=SUCONe1qa5jMHOMqqS27ATtv3CaqAT8cN9jNi7AI_Go,5813
 remote/file_sender.py,sha256=5_3ptTkoFejhJhaSyzh-8y5l_k7frxFq9LS_WL5jsGc,3657
@@ -31,8 +31,9 @@ remote/speed_manager.py,sha256=fIWSQAP9qW8AHZtMZq0MrC4_nvxcTFU1SBU75kpRzB8,1115
 remote/stop_manager.py,sha256=UXzKJTblEyQqCjp7fenvQ51Q96Unx05WeOiuFMdj25M,1151
 remote/text_manager.py,sha256=C2wNSXPSCDu8NSD3RsfbKmUQMWOYd1B5N4tzy-Jsriw,2195
 remote/username_manager.py,sha256=nMNdke-2FIv86xR1Y6rR-43oUoQu_3Khw8wEo54noXI,3388
+remote/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 remote/utils/sqlite_utils.py,sha256=5i0oUXsBgKC_8qHZPJ-Gyhp9D1TwqKHVvuZRIhKpS6w,1260
-cliremote-1.6.0.dist-info/METADATA,sha256=TVNkkckVN52498m8gUoc5SFnEYyZ1cKFE6wDUZh6GTk,1202
-cliremote-1.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cliremote-1.6.0.dist-info/top_level.txt,sha256=yBZidJ6zCix_a2ubGlYaewvlzBFXWbckQt20dudxJ1E,7
-cliremote-1.6.0.dist-info/RECORD,,
+cliremote-1.6.2.dist-info/METADATA,sha256=pXO0OVFiYv3MzWlY2qLDRHs0AhT33RGUf2hF3FGguDg,1202
+cliremote-1.6.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cliremote-1.6.2.dist-info/top_level.txt,sha256=yBZidJ6zCix_a2ubGlYaewvlzBFXWbckQt20dudxJ1E,7
+cliremote-1.6.2.dist-info/RECORD,,

remote/client_manager.py

@@ -229,3 +229,53 @@ def accounts() -> List[str]:
 
 def get_active_accounts() -> Set[str]:
     return set(accounts())
+
+
+def remove_client_from_pool(phone_number: str) -> None:
+    """
+    کلاینت را از pool خارج و stop می‌کند (به‌صورت غیرمسدودکننده).
+    """
+    cli = client_pool.get(phone_number)
+    if cli is not None:
+        try:
+            asyncio.create_task(cli.stop())
+            logger.info("%s: scheduled stop()", phone_number)
+        except Exception as e:
+            logger.warning("%s: stop() scheduling error: %s: %s", phone_number, type(e).__name__, e)
+    client_pool.pop(phone_number, None)
+    client_locks.pop(phone_number, None)
+    logger.info("%s: removed from client_pool and client_locks", phone_number)
+
+async def get_any_client(message=None) -> Optional[object]:
+    """
+    یک کلاینت آماده برمی‌گرداند:
+      1) اگر کلاینت متصل در pool هست، همان را برمی‌گرداند.
+      2) وگرنه از بین active_accounts یکی را استارت می‌کند.
+    """
+    # Use any connected client
+    for phone, cli in list(client_pool.items()):
+        try:
+            if getattr(cli, "is_connected", False):
+                logger.debug("get_any_client: using connected client %s", phone)
+                return cli
+        except Exception:
+            pass
+
+    # Start one if needed
+    accs = list(get_active_accounts())
+    if not accs:
+        logger.warning("get_any_client: no active accounts")
+        return None
+
+    random.shuffle(accs)
+    for phone in accs:
+        try:
+            cli = await get_or_start_client(phone)
+            if cli and getattr(cli, "is_connected", False):
+                logger.info("get_any_client: started %s", phone)
+                return cli
+        except Exception as e:
+            logger.warning("get_any_client: failed start %s: %s: %s", phone, type(e).__name__, e)
+
+    logger.error("get_any_client: could not get any client")
+    return None

remote/client_picker.py

@@ -1,109 +1,4 @@
-# remote/client_picker.py
 import random
-import asyncio
 import logging
-from typing import Optional, Callable, Iterable
-
-logger = logging.getLogger(__name__)
-
-# ---- وابستگی‌های اصلی
-# تلاش می‌کنیم client_manager را حتماً داشته باشیم
-from . import client_manager
-
-# get_active() را به‌صورت ایمن تعیین می‌کنیم:
-def _resolve_get_active() -> Callable[[], Iterable[str]]:
-    """
-    سعی می‌کند منبع معتبر لیست اکانت‌های فعال را پیدا کند:
-      1) client_manager.get_active_accounts()
-      2) account_manager.get_active_accounts()
-      3) account_manager.accounts()  (fallback)
-      4) client_manager.accounts()   (fallback)
-    و در نهایت اگر هیچ‌کدام نبود، یک فانکشنِ خالی برمی‌گرداند.
-    """
-    # 1) client_manager.get_active_accounts
-    if hasattr(client_manager, "get_active_accounts"):
-        return client_manager.get_active_accounts
-
-    # 2) account_manager...
-    try:
-        from . import account_manager  # ممکن است وجود نداشته باشد
-        if hasattr(account_manager, "get_active_accounts"):
-            return account_manager.get_active_accounts
-        if hasattr(account_manager, "accounts"):
-            return lambda: set(account_manager.accounts())
-    except Exception:
-        pass
-
-    # 3) fallback به client_manager.accounts
-    if hasattr(client_manager, "accounts"):
-        return lambda: set(client_manager.accounts())
-
-    # 4) آخرین fallback: لیست خالی
-    return lambda: set()
-
-_get_active_accounts = _resolve_get_active()
-
-
-async def get_any_client(message=None, max_attempts: int = 3) -> Optional[object]:
-    """
-    تلاش برای گرفتن یک کلاینت فعال از بین اکانت‌ها.
-    - تا `max_attempts` بار با اکانت‌های تصادفی امتحان می‌کند.
-    - اگر موفق نشد، پیام خطا (در صورت وجود message) ارسال می‌کند،
-      سپس stop_all_clients() فراخوانی می‌شود و در نهایت None برمی‌گرداند.
-    """
-    try:
-        acc_iter = _get_active_accounts()
-        acc_list = list(acc_iter) if not isinstance(acc_iter, (list, set, tuple)) else list(acc_iter)
-    except Exception as e:
-        logger.error(f"❌ نتوانستم لیست اکانت‌ها را بگیرم: {type(e).__name__} - {e}")
-        acc_list = []
-
-    if not acc_list:
-        if message:
-            try:
-                await message.reply("⚠️ هیچ اکانت فعالی برای اتصال وجود ندارد.")
-            except Exception:
-                pass
-        logger.warning("⚠️ هیچ اکانت فعالی در دسترس نیست.")
-        return None
-
-    tried = set()
-
-    for attempt in range(1, max_attempts + 1):
-        if len(tried) == len(acc_list):
-            break
-
-        phone = random.choice([p for p in acc_list if p not in tried])
-        tried.add(phone)
-        logger.info(f"🔁 تلاش {attempt}/{max_attempts} برای اتصال با اکانت {phone}")
-
-        try:
-            cli = await client_manager.get_or_start_client(phone)
-            if cli and getattr(cli, "is_connected", True):
-                logger.info(f"✅ اتصال موفق با اکانت {phone}")
-                return cli
-            else:
-                logger.warning(f"⚠️ اکانت {phone} وصل نیست یا کلاینت معتبر برنگشته.")
-        except Exception as e:
-            logger.error(f"❌ خطا در اتصال {phone}: {type(e).__name__} - {e}")
-            try:
-                await asyncio.sleep(1)
-            except Exception:
-                pass
-
-    # شکست پس از تلاش‌ها
-    error_msg = f"❌ هیچ کلاینت فعالی پس از {max_attempts} تلاش یافت نشد. در حال ریست کامل کلاینت‌ها..."
-    if message:
-        try:
-            await message.reply(error_msg)
-        except Exception:
-            pass
-    logger.error(error_msg)
-
-    try:
-        await client_manager.stop_all_clients()
-        logger.warning("🔄 تمام کلاینت‌ها ریست شدند (stop_all_clients فراخوانی شد).")
-    except Exception as e:
-        logger.error(f"⚠️ خطا در ریست کلاینت‌ها: {type(e).__name__} - {e}")
-
-    return None
+from typing import Optional
+from .client_manager import client_pool, get_active_accounts, get_or_start_client