d4rktg 0.4.9__py3-none-any.whl → 0.5.1__py3-none-any.whl

d4rk/Utils/_round.py

@@ -1,39 +1,54 @@
 from functools import wraps
 from pyrogram.types import Message
+import asyncio
 
 # Track last bot index per chat
 last_index_per_chat = {}
-# Track all bot IDs dynamically
-active_bot_ids = []
+# Track bot order per chat
+bot_order_per_chat = {}
+# Track messages already responded to: chat_id -> set of message_ids
+responded_messages = {}
+# Lock per chat to prevent race conditions
+chat_locks = {}
 
 def round_robin():
     def decorator(func):
         @wraps(func)
         async def wrapper(client, message: 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)
 
-            # Add bot ID to active list dynamically
-            if client.me.id not in active_bot_ids:
-                active_bot_ids.append(client.me.id)
-
-            # Initialize last_index for this chat
-            if chat_id not in last_index_per_chat:
+            # 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
-
-            current_index = last_index_per_chat[chat_id]
-            selected_bot_id = active_bot_ids[current_index]
-
-            # Only the selected bot responds
-            if client.me.id == selected_bot_id:
-                result = await func(client, message, *args, **kwargs)
-
-                # Rotate to next bot index
-                last_index_per_chat[chat_id] = (current_index + 1) % len(active_bot_ids)
-                return result
+                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
+    return decorator

{d4rktg-0.4.9.dist-info → d4rktg-0.5.1.dist-info}/METADATA

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

{d4rktg-0.4.9.dist-info → d4rktg-0.5.1.dist-info}/RECORD

@@ -16,9 +16,9 @@ d4rk/Utils/_fonts.py,sha256=CQsDqPgvp27t3f75Cxod0EmZogEssLwIpyesH-YY5KM,7518
 d4rk/Utils/_ip.py,sha256=KJJW2QSngshIVWCO5YPXF1wj4IPQzVN5oFofpfzlU5w,559
 d4rk/Utils/_movie_parser.py,sha256=QEPd3z04p4pk2vxZ-2fYlRxHmQbmwQd2mCIlBsBM5AY,7039
 d4rk/Utils/_ractions.py,sha256=wOVPyoFnbDuMgoP6NF_gLO1DYcfhERC0trdAK1jWSE8,2170
-d4rk/Utils/_round.py,sha256=PSvQGViHXiWX7Ly7MX-p0HlxKdIswngqWeowFTgWQcU,1356
+d4rk/Utils/_round.py,sha256=jMFTG0eLcL3RVQZrgdSpKhLLus0Nc1WZFgJCET5Fdaw,2100
 d4rk/Utils/_terminal.py,sha256=Anu4OcffY3v6LMOrCskP1cHrJIliomo1Hjownbhh2sQ,125
-d4rktg-0.4.9.dist-info/METADATA,sha256=OgP5aTWAzsyOJbUudUI_ylg3iXz2QB3DolusW_cNdb0,652
-d4rktg-0.4.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-d4rktg-0.4.9.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
-d4rktg-0.4.9.dist-info/RECORD,,
+d4rktg-0.5.1.dist-info/METADATA,sha256=aSUzke8cuqx_rrCsIEPw30LDd4xKFnLgWJ62L8htUz0,652
+d4rktg-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+d4rktg-0.5.1.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
+d4rktg-0.5.1.dist-info/RECORD,,