d4rktg 0.8.1__tar.gz → 0.8.2__tar.gz

{d4rktg-0.8.1 → d4rktg-0.8.2}/PKG-INFO

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

d4rktg-0.8.2/VERSION.txt

@@ -0,0 +1 @@
+0.8.2

{d4rktg-0.8.1 → d4rktg-0.8.2}/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()
@@ -52,11 +89,9 @@ def command(command: Union[str, list], description: str,Custom_filter=None):
             "handler": func
         })
         logger.info(f"Registered command: {command} - {description}")
-        if Custom_filter:
-            filter = cmd(command, prefixes="/") & 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 → d4rktg-0.8.2}/d4rktg.egg-info/PKG-INFO

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

d4rktg-0.8.1/VERSION.txt

@@ -1 +0,0 @@
-0.8.1