d4rktg 0.8.1__py3-none-any.whl → 0.8.3__py3-none-any.whl

d4rk/Models/_commands.py

@@ -1,30 +1,67 @@
 import re
 import requests
+import asyncio
 
 from typing import Union
 from functools import wraps
 from pyrogram import Client, filters
+from pyrogram.types import Message
 
 from d4rk.Logs import setup_logger
 from d4rk.Utils import clear_terminal
 
 logger = setup_logger(__name__)
 
+
 command_registry = []
 
 
-def cmd(commands, prefixes="/"):
-    # Normalize: accept str or list
-    if isinstance(commands, str):
-        commands = [commands]
-    async def func(flt, client, message):
-        text = message.text or message.caption or ""
-        for cmd in commands:
-            if text.startswith(f"{prefixes}{cmd}"):
-                return True
-        return False
-    return filters.create(func)
+last_index_per_chat = {}
+bot_order_per_chat = {}
+responded_messages = {}
+chat_locks = {}
+
+def round_robin():
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(client, message, *args, **kwargs):
+            chat_id = message.chat.id
+            msg_id = message.id
+
+            # Private chats: all bots respond
+            if message.chat.type.name.lower() == "private":
+                return await func(client, message, *args, **kwargs)
+
+            # Initialize tracking for this chat
+            if chat_id not in bot_order_per_chat:
+                bot_order_per_chat[chat_id] = [client.me.id]
+                last_index_per_chat[chat_id] = 0
+                responded_messages[chat_id] = set()
+                chat_locks[chat_id] = asyncio.Lock()
 
+            # Add new bot if not in the chat
+            if client.me.id not in bot_order_per_chat[chat_id]:
+                bot_order_per_chat[chat_id].append(client.me.id)
+
+            async with chat_locks[chat_id]:
+                # Skip if message already responded
+                if msg_id in responded_messages[chat_id]:
+                    return
+
+                # Decide which bot should respond
+                current_index = last_index_per_chat[chat_id]
+                selected_bot_id = bot_order_per_chat[chat_id][current_index]
+
+                if client.me.id == selected_bot_id:
+                    result = await func(client, message, *args, **kwargs)
+                    # Mark message as responded
+                    responded_messages[chat_id].add(msg_id)
+                    # Rotate for next message
+                    last_index_per_chat[chat_id] = (current_index + 1) % len(bot_order_per_chat[chat_id])
+                    return result
+
+        return wrapper
+    return decorator
 
 def get_priority(description: str) -> int:
     desc_lower = description.lower()
@@ -53,10 +90,11 @@ def command(command: Union[str, list], description: str,Custom_filter=None):
         })
         logger.info(f"Registered command: {command} - {description}")
         if Custom_filter:
-            filter = cmd(command, prefixes="/") & Custom_filter
+            filter = filters.command(command) & Custom_filter
         else:
-            filter = cmd(command, prefixes="/")
+            filter = filters.command(command)
         @Client.on_message(filter)
+        @round_robin()
         @wraps(func)
         async def wrapper(client, message):
             return await func(client, message)

{d4rktg-0.8.1.dist-info → d4rktg-0.8.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: d4rktg
-Version: 0.8.1
+Version: 0.8.3
 Summary: A module for create with easy and fast
 Author: D4rkShell
 Author-email: premiumqtrst@gmail.com

{d4rktg-0.8.1.dist-info → d4rktg-0.8.3.dist-info}/RECORD

@@ -7,7 +7,7 @@ d4rk/Handlers/_scheduler.py,sha256=AyqexO4nxZlIzRfU9vWTfJtTWQVQmP4de7GRPg-3JkA,1
 d4rk/Logs/__init__.py,sha256=mXWD5jXnyH3_AvS7K_ki3iw5BpoEAvrDFbmr-iEFNnY,22
 d4rk/Logs/_logger.py,sha256=lqfVvCO0vZ_IaGOdIE4HA2KAUQZh7yW2iAHZcBz7F4o,4120
 d4rk/Models/__init__.py,sha256=7bYM2c5Uk_VEg7NxIGH2EcEXGcAguwxvH23-Gy1orMM,100
-d4rk/Models/_commands.py,sha256=mPJ--4uZKxXQlewGT-KNVEXWxddu1-Y55jkc149A2HE,3380
+d4rk/Models/_commands.py,sha256=_1ageKjA7xNgw7cQsDIjcIuJ96h6DENAZ0Wx1Hm2970,4884
 d4rk/Models/_movie_title.py,sha256=Xp8eafBP-kD3MxAdeNcb7EyiAojn5fDQ18la7vxrg10,812
 d4rk/Utils/__init__.py,sha256=nWnDrvTJ8BgbqEE9aiVOcT8JoxK6P4XvP9_rgUwEHXs,333
 d4rk/Utils/_decorators.py,sha256=xplcr3qUSewGKET0Y7TnW2ddL3ZY0IeHS5y3WewB-tc,2672
@@ -18,7 +18,7 @@ d4rk/Utils/_movie_parser.py,sha256=QEPd3z04p4pk2vxZ-2fYlRxHmQbmwQd2mCIlBsBM5AY,7
 d4rk/Utils/_ractions.py,sha256=wOVPyoFnbDuMgoP6NF_gLO1DYcfhERC0trdAK1jWSE8,2170
 d4rk/Utils/_round.py,sha256=reQ5CqAL6lSaCtvIkcTekjYVdoUXTc6ej8VZYXabqtk,4474
 d4rk/Utils/_terminal.py,sha256=Anu4OcffY3v6LMOrCskP1cHrJIliomo1Hjownbhh2sQ,125
-d4rktg-0.8.1.dist-info/METADATA,sha256=XQz88byerWZz7lGUxTc-7xTaREXuBdcwjI1YhwQd3lc,652
-d4rktg-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-d4rktg-0.8.1.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
-d4rktg-0.8.1.dist-info/RECORD,,
+d4rktg-0.8.3.dist-info/METADATA,sha256=q-AuBeq0VCaCCeT3UsCzutF9cqrAyVCMJ-hm3UVv-Y8,652
+d4rktg-0.8.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+d4rktg-0.8.3.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
+d4rktg-0.8.3.dist-info/RECORD,,